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

Fix sdd_lut (#368)

Browse Source
This commit is contained in:
daadaada
2021-11-09 00:25:05 +08:00
committed by GitHub
parent 5d54352164
commit 9a02dddf29
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -107,6 +107,7 @@ def sdd_matmul(a, b, trans_a, trans_b, trans_c, spdims, block, lut, widths, out
def sdd_lut(layout, block, device): def sdd_lut(layout, block, device):
lut = layout.nonzero(as_tuple=False).to(device).int() lut = layout.nonzero(as_tuple=False).to(device).int()
lut = lut.contiguous()
return lut, None return lut, None
# ----------------------------- # -----------------------------