gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PYTHON] Bugfix in trans_c for dsd in blocksparse matmul (#131)

Browse Source
This commit is contained in:
Philippe Tillet
2021-07-12 23:57:44 -07:00
committed by Philippe Tillet
parent 8cea583109
commit 3169e4355c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/triton/ops/blocksparse/matmul.py
View File

@@ -512,8 +512,8 @@ class _matmul(torch.autograd.Function):
b.stride(2 if trans_b else 3), b.stride(2 if trans_b else 3),
c.stride(0), c.stride(0),
c.stride(1), c.stride(1),
c.stride(2), c.stride(3 if trans_c else 2),
c.stride(3), c.stride(2 if trans_c else 3),
BS3, BS3,
AS1, AS1,
0, 0,