Skip to content

Commit 5effac5

Browse files
authored
make float 8 error message test BC (#1284)
this test is going to fail on older versions of pytorch, i.e. https://github.com/pytorch/ao/actions/runs/11827805971/job/32956667984?pr=1282 we can make this handle both old/current by being slightly less specific with the error check
1 parent 06e69f6 commit 5effac5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/float8/test_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -632,7 +632,7 @@ def test_pad_inner_dim(self, base_dtype, use_fast_accum):
632632
with pytest.raises(
633633
RuntimeError,
634634
match=re.escape(
635-
"Expected trailing dimension of mat1 to be divisible by 16 but got mat1 shape: (16x41)."
635+
"Expected trailing dimension of mat1 to be divisible by 16 but got mat1 shape: (16x41"
636636
),
637637
):
638638
a_fp8 @ b_fp8

0 commit comments

Comments
 (0)