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

[PYTHON] Bugfix on FP32 blocksparse matmul

Browse Source
This commit is contained in:
Philippe Tillet
2021-02-24 14:43:08 -05:00
parent 567a1a3d17
commit f64b779b0d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -362,7 +362,7 @@ class matmul:
return self.lut_cache[key] return self.lut_cache[key]
# C look-up table # C look-up table
layout, block = self.layout, self.block layout, block = self.layout, self.block
step = 8 if dtype == torch.float32 else 16 step = 16
if self.mode == 'sdd': if self.mode == 'sdd':
c_lut, c_num_locks, c_width, c_packs = _matmul.make_sdd_lut(layout, block, dtype, device) c_lut, c_num_locks, c_width, c_packs = _matmul.make_sdd_lut(layout, block, dtype, device)
elif self.mode == 'dsd': elif self.mode == 'dsd':