Skip to content

Commit a6b6697

Browse files
alugoreypruthvistony
authored andcommitted
Enable e5m2 x e4m3 test in test_float8_scale (#1419)
1 parent 8bd8f81 commit a6b6697

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/test_matmul_cuda.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -374,8 +374,7 @@ def test_float8_basics(self, device) -> None:
374374
def test_float8_scale(self, device) -> None:
375375
size = (16, 16)
376376
x = torch.full(size, .5, device=device, dtype=e4m3_type)
377-
# hipblaslt does not yet support mixed e4m3_type input
378-
y_type = e4m3_type if torch.version.hip else e5m2_type
377+
y_type = e5m2_type
379378
y = torch.full(size, .5, device=device, dtype=y_type).t()
380379
scale_a = torch.tensor(1.5, device=device)
381380
scale_b = torch.tensor(0.66, device=device)

0 commit comments

Comments
 (0)