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

[FP8] Add other Blackwell compute-capabiilities to expected fail test_honor_sm_carveout (#165159)

Browse Source 
CUTLASS SM hint also isn't working for other Blackwells, need green context for carveout

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165159
Approved by: https://github.com/Skylion007
This commit is contained in:
eqy
2025-10-16 18:35:01 +00:00
committed by PyTorch MergeBot
parent 7d87d7052e
commit a214371008
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_scaled_matmul_cuda.py
View File

@ -1211,7 +1211,7 @@ class TestFP8Matmul(TestCase):
self.assertEqual(no_carveout, no_carveout_again)
capability = torch.cuda.get_device_capability()
if capability == (10, 0):
if capability in {(10, 0), (10, 3), (12, 0), (12, 1)}:
# expected failure
# CUTLASS only supports SM carveout via green contexts on SM100
self.assertEqual(no_carveout, carveout_66)