Skip to content

Commit 7e59bc5

Browse files
committed
add back float8
1 parent e629f5b commit 7e59bc5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/triton_kernels/tests/test_matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ def _make_tensor(shape, dtype, trans):
493493
(False, True),
494494
(False, True),
495495
(False, True),
496-
(torch.float16, torch.bfloat16),
496+
(torch.float16, torch.bfloat16, torch.float8_e5m2),
497497
):
498498
if device == "cuda" and (
499499
torch.cuda.get_device_capability()[0] < 10

0 commit comments

Comments
 (0)