[fix] fix enable_pdl for blackwell (#9011)
This commit is contained in:
@@ -43,8 +43,8 @@ def _to_tensor_scalar_tuple(x):
|
||||
|
||||
|
||||
@functools.lru_cache(maxsize=1)
|
||||
def is_hopper_arch() -> bool:
|
||||
def is_arch_support_pdl() -> bool:
|
||||
# Hopper arch's compute capability == 9.0
|
||||
device = torch.cuda.current_device()
|
||||
major, minor = torch.cuda.get_device_capability(device)
|
||||
return major == 9
|
||||
return major >= 9
|
||||
|
||||
Reference in New Issue
Block a user