diff --git a/python/triton/ops/blocksparse/matmul.py b/python/triton/ops/blocksparse/matmul.py index 8a020e5c2..49497777a 100644 --- 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 # -----------------------------