Fix sdd_lut (#368)

2021-11-09 00:25:05 +08:00
parent 5d54352164
commit 9a02dddf29
1 changed files with 1 additions and 0 deletions
--- a/python/triton/ops/blocksparse/matmul.py
+++ b/python/triton/ops/blocksparse/matmul.py
@@ -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):
    lut = layout.nonzero(as_tuple=False).to(device).int()
+    lut = lut.contiguous()
    return lut, None

 # -----------------------------