Skip to content

Commit

Permalink
make float 8 error message test BC (#1284)
Browse files Browse the repository at this point in the history
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
  • Loading branch information
HDCharles authored Nov 14, 2024
1 parent 06e69f6 commit 5effac5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/float8/test_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -632,7 +632,7 @@ def test_pad_inner_dim(self, base_dtype, use_fast_accum):
with pytest.raises(
RuntimeError,
match=re.escape(
"Expected trailing dimension of mat1 to be divisible by 16 but got mat1 shape: (16x41)."
"Expected trailing dimension of mat1 to be divisible by 16 but got mat1 shape: (16x41"
),
):
a_fp8 @ b_fp8
Expand Down

0 comments on commit 5effac5

Please sign in to comment.