From 9a02dddf29803d5229907d01ebc4e7b4edd179f9 Mon Sep 17 00:00:00 2001 From: daadaada Date: Tue, 9 Nov 2021 00:25:05 +0800 Subject: [PATCH] Fix sdd_lut (#368) --- python/triton/ops/blocksparse/matmul.py | 1 + 1 file changed, 1 insertion(+) 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 # -----------------------------