diff options
author | Jianyu Huang <jianyuhuang@fb.com> | 2019-03-08 21:34:32 +0300 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-03-08 21:37:10 +0300 |
commit | 50b43162fd1742122d01f2704945c78f13e0d73e (patch) | |
tree | d5fee7d82429cd63aa1f8bee3628e822bd010436 /include/fbgemm | |
parent | 844dacc267391cd2a725d81c2495636f0765771b (diff) |
No need for PackA when m==1 (#83)
Summary:
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/83
When m = 1, PackA is actually not necessary: PackA operations for FP16 in these two libraries are both simply matrix transposition. In this case, we don’t need to do the transposition. We can just pass the pointer of the original A matrix buffer to the packed A buffer.
Reviewed By: zhengwy888
Differential Revision: D14299246
fbshipit-source-id: 78a62c5ff3a396b59afb15462efe38461cb71e15
Diffstat (limited to 'include/fbgemm')
0 files changed, 0 insertions, 0 deletions