Skip to content

Commit 2653139

Browse files
[ROCm][release/2.7] Bug fix TunableOp validator UT. (#2576)
This is a partial cherry-pick from upstream: pytorch#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.
1 parent 10cbfa3 commit 2653139

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_linalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5162,7 +5162,7 @@ def test_validator_tunableop_rocm(self, device, dtype):
51625162
# Check for rocBLAS and hipBLASLt
51635163
self.assertTrue("ROCBLAS_VERSION" in validators)
51645164
# format: [major].[minor].[patch].[tweak].[commit id]
5165-
self.assertTrue(re.match(r'^\d+.\d+.\d+.\d+.[a-z0-9]+$', validators["ROCBLAS_VERSION"]))
5165+
self.assertTrue(re.match(r'^\d+[a-z0-9.]+$', validators["ROCBLAS_VERSION"]))
51665166
self.assertTrue("HIPBLASLT_VERSION" in validators)
51675167
self.assertTrue(re.match(r'^\d+-[a-z0-9]+$', validators["HIPBLASLT_VERSION"]))
51685168

0 commit comments

Comments
 (0)