Skip to content

Commit f25a905

Browse files
fix linting
Signed-off-by: cliu-us <[email protected]>
1 parent eb9a25f commit f25a905

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/triton_kernels/test_triton_mm.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,8 @@ def test_triton_matmul_fp(mkn, dtype_to_test):
4747
cuda_cc = torch.cuda.get_device_capability()
4848
if cuda_cc[0] < 9 and cuda_cc != (8, 9):
4949
return
50-
else:
51-
# torch.matmul does not support fp8 x fp8 on cuda
52-
torch_mm_device = "cpu"
50+
# torch.matmul does not support fp8 x fp8 on cuda
51+
torch_mm_device = "cpu"
5352

5453
a = a.to(dtype_to_test)
5554
b = b.to(dtype_to_test)

0 commit comments

Comments
 (0)