diff options
| -rw-r--r-- | src/PackMatrix.cc | 11 |
1 files changed, 1 insertions, 10 deletions
diff --git a/src/PackMatrix.cc b/src/PackMatrix.cc index c9a68a6..33227fb 100644 --- a/src/PackMatrix.cc +++ b/src/PackMatrix.cc @@ -62,16 +62,7 @@ int PackMatrix<PT, inpType, accType>::packedBufferSize( } } - if (fbgemmHasAvx512Support()) { - if (isA()) { - return MCB * KCB; - } else { - int rowBlock = KCB; - int colBlock = NCB; - return (((rows + rowBlock - 1) / rowBlock) * rowBlock) * - (((cols + colBlock - 1) / colBlock) * colBlock); - } - } else if (fbgemmHasAvx2Support()) { + if (fbgemmHasAvx512Support() || fbgemmHasAvx2Support()) { if (isA()) { return MCB * KCB; } else { |