diff --git a/test/test_linalg.py b/test/test_linalg.py index 6c4e6ccf04cd8..6fc46663a67e7 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -5162,7 +5162,7 @@ def test_validator_tunableop_rocm(self, device, dtype): # Check for rocBLAS and hipBLASLt self.assertTrue("ROCBLAS_VERSION" in validators) # format: [major].[minor].[patch].[tweak].[commit id] - self.assertTrue(re.match(r'^\d+.\d+.\d+.\d+.[a-z0-9]+$', validators["ROCBLAS_VERSION"])) + self.assertTrue(re.match(r'^\d+[a-z0-9.]+$', validators["ROCBLAS_VERSION"])) self.assertTrue("HIPBLASLT_VERSION" in validators) self.assertTrue(re.match(r'^\d+-[a-z0-9]+$', validators["HIPBLASLT_VERSION"]))