From 2712a64aad119e32cd5bebdc72808e2be37eef47 Mon Sep 17 00:00:00 2001 From: "Nichols A. Romero" <165712832+naromero77amd@users.noreply.github.com> Date: Tue, 26 Aug 2025 11:33:47 -0500 Subject: [PATCH] [ROCm][release/2.7] Bug fix TunableOp validator UT. (#2576) This is a partial cherry-pick from upstream: https://github.com/pytorch/pytorch/pull/158887 The TunableOp UT appears to fail with different versions of ROCm, but the real issue is the that regex in the UT is incorrect. @jeffdaily discovered this when we upgraded the nightly wheels to ROCm 6.4.2 in upstream PyTorch. Will resolve SWDEV-548689. --- test/test_linalg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_linalg.py b/test/test_linalg.py index 1f5d4009cebba..297d75c3845ac 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -5217,7 +5217,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"]))