Skip some tests on Blackwell (#9777)

Signed-off-by: Hao Lu <14827759+hlu1@users.noreply.github.com>
This commit is contained in:
hlu1
2025-08-28 20:00:32 -07:00
committed by GitHub
parent 38cd5fb1e0
commit a7d825fccc
3 changed files with 23 additions and 0 deletions

View File

@@ -0,0 +1,9 @@
import torch
def is_sm10x():
return torch.cuda.get_device_capability() >= (10, 0)
def is_hopper():
return torch.cuda.get_device_capability() == (9, 0)