diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/FbgemmFP16.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/FbgemmFP16.cc b/src/FbgemmFP16.cc index 2af1f89..69680bc 100644 --- a/src/FbgemmFP16.cc +++ b/src/FbgemmFP16.cc @@ -229,7 +229,7 @@ FBGEMM_API void cblas_gemm_compute( const int kb = std::min(Bp.blockRowSize(), Bp.numRows() - k_ind); - auto m1 = 0; + auto m1 = m0; for (auto c = 0; c < 2; c++) { auto kernel_nrows = KernelInfo::partition[mb][c][0]; auto nkernel_nrows = KernelInfo::partition[mb][c][1]; |