diff options
author | Jianyu Huang <jianyuhuang@fb.com> | 2019-02-13 23:58:01 +0300 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-02-14 00:31:12 +0300 |
commit | aa88eafebf8df24dfa8c60768b1103f6981f7322 (patch) | |
tree | 381e14f5073cd28b815ea0b3443a08965655f85a | |
parent | 64269506c9bab424cfd808d4fa398574ac38cd5f (diff) |
Fix PackBMatrix<T, accT>::printPackedMatrix issues
Summary: Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/74
Reviewed By: jspark1105
Differential Revision: D14063593
fbshipit-source-id: 4c4a23df21e2d66eb3b6d3bee7196c6ad1935362
-rw-r--r-- | src/PackBMatrix.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/PackBMatrix.cc b/src/PackBMatrix.cc index 1b38997..ee9be84 100644 --- a/src/PackBMatrix.cc +++ b/src/PackBMatrix.cc @@ -115,8 +115,8 @@ void PackBMatrix<T, accT>::printPackedMatrix(std::string name) { << "[" << BaseType::blockRowSize() << ", " << BaseType::blockColSize() << "]" << std::endl; - T* out = BaseType::getBuf() + - this->packedBufferSize(this->numPackedRows(), this->numPackedCols()); + T* out = BaseType::getBuf(); + for (auto nr = 0; nr < BaseType::blockRows(); ++nr) { auto rows = (nr == BaseType::blockRows() - 1) ? BaseType::lastBrow() : BaseType::blockRowSize(); |