frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI][CUDA][Blackwell] sm_\d\d no longer matches sm_100. (#145641)

Browse Source 
Therefore making it sm_\d+

Fixes this unit test failure: python test/test_cpp_extensions_jit.py -k TestCppExtensionJIT.test_jit_cuda_archflags

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145641
Approved by: https://github.com/eqy, https://github.com/malfet
This commit is contained in:
Wei Wang
2025-01-24 23:20:22 +00:00
committed by PyTorch MergeBot
parent 4cc5e880f9
commit 0741963e01
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_cpp_extensions_jit.py
View File

@ -159,7 +159,7 @@ class TestCppExtensionJIT(common.TestCase):
f"Output: {output} "
)
actual_arches = sorted(re.findall(r"sm_\d\d", output))
actual_arches = sorted(re.findall(r"sm_\d+", output))
expected_arches = sorted(["sm_" + xx for xx in expected_values])
self.assertEqual(
actual_arches,