fix: add SM110 (Jetson AGX Thor) to Blackwell capability check (#18787)

This commit is contained in:
WiwilZ
2026-02-16 00:26:58 +08:00
committed by GitHub
parent 57f7e06cb9
commit b2f74d660a

View File

@@ -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),
)
)