fix: add SM110 (Jetson AGX Thor) to Blackwell capability check (#18787)
This commit is contained in:
@@ -246,7 +246,7 @@ is_hopper_with_cuda_12_3 = lru_cache(maxsize=1)(
|
||||
is_blackwell_supported = is_blackwell = lru_cache(maxsize=1)(
|
||||
partial(
|
||||
_check_cuda_device_version,
|
||||
device_capability_majors=[10, 12],
|
||||
device_capability_majors=[10, 11, 12],
|
||||
cuda_version=(12, 8),
|
||||
)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user