From a5a82df5a497d5a4fd7d50df334178a6e6e11ae9 Mon Sep 17 00:00:00 2001 From: HDCharles <39544797+HDCharles@users.noreply.github.com> Date: Wed, 13 Nov 2024 22:14:58 -0500 Subject: [PATCH] make float 8 error message test BC 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 --- test/float8/test_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/float8/test_base.py b/test/float8/test_base.py index 7bc5a3788..a0ea96baa 100644 --- a/test/float8/test_base.py +++ b/test/float8/test_base.py @@ -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