diff options
| author | Feiteng <lifeiteng0422@gmail.com> | 2019-01-02 06:19:58 +0300 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-02 06:22:22 +0300 |
| commit | 4321c4d96a76d4735f0869ae1adb6fa2197cf377 (patch) | |
| tree | 86c01c48f590dd618e7983a723232d7ce19f802c /src/FbgemmFP16.cc | |
| parent | b86d64219a0cb8fc2f8e9595c91986faff17863c (diff) | |
Fix a bug in FbgemmFP16 (#52)
Summary:
https://github.com/pytorch/FBGEMM/blob/master/test/FP16Test.cc#L46 will fail if batch_size (e.g. m) is larger than 120.
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/52
Reviewed By: jspark1105
Differential Revision: D13554483
Pulled By: jianyuh
fbshipit-source-id: ed50d475f618b9ad4bec1062d19ebf9aa5a243d5
Diffstat (limited to 'src/FbgemmFP16.cc')
| -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]; |