Fix failing internal codecache test (#141405)

When internal remote cache version was bumped to 11, this test started failing, I guess no one noticed it, and it got disabled.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141405
Approved by: https://github.com/aorenste
This commit is contained in:
Oguz Ulgen
2024-11-22 16:39:00 -08:00
committed by PyTorch MergeBot
parent 1aea642393
commit a8ab6b0938

View File

@ -1229,7 +1229,7 @@ class TestAutotuneCache(TestCase):
for k in global_stats.autotune_remote.cache.keys():
self.assertRegex(k, r"[0-9a-z]{52}\.py")
for k in global_stats.triton.cache.keys():
self.assertRegex(k, r"triton:[0-9a-f]{64}::[0-9a-f]{64}:c10")
self.assertRegex(k, r"triton:[0-9a-f]{64}::[0-9a-f]{64}:c11")
@unittest.skipIf(not HAS_CUDA, "Requires CUDA")
@unittest.skipIf(not SM80OrLater, "Requires SM80+")