Skip to content

Commit 990069a

Browse files
committed
Enable triton kernels matmul tests
Signed-off-by: Witold Dziurdz <[email protected]>
1 parent c0e155d commit 990069a

File tree

4 files changed

+1
-4
lines changed

4 files changed

+1
-4
lines changed

python/triton_kernels/triton_kernels/matmul_ogs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -554,7 +554,7 @@ def matmul_ogs(x, w, bias,
554554
out_matmul_scale = out_matmul_scale.data.view(torch.uint8)
555555
if has_scratchpad and "mx_out_scale" in memory["scratchpad"]:
556556
out_matmul_scale = memory["scratchpad"]["mx_out_scale"]
557-
out_matmul_has_mx = out_matmul_scale is not None and out_matmul.element_size() == 1
557+
out_matmul_has_mx = out_matmul_scale is not None and bitwidth(out_dtype) == 8
558558
# matrix multiplication
559559
flex = precision_config.flex_ctx
560560
bias_stride = None if bias is None else bias.stride(0)
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,2 @@
1-
tests/test_matmul.py::test_op
21
tests/test_matmul.py::test_fused_act
32
tests/test_matmul.py::test_zero_reduction_dim
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,2 @@
1-
tests/test_matmul.py::test_op
21
tests/test_matmul.py::test_fused_act
32
tests/test_matmul.py::test_zero_reduction_dim
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,2 @@
1-
tests/test_matmul.py::test_op
21
tests/test_matmul.py::test_fused_act
32
tests/test_matmul.py::test_zero_reduction_dim

0 commit comments

Comments
 (0)