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

2026-02-16 00:26:58 +08:00
parent 57f7e06cb9
commit b2f74d660a
1 changed files with 1 additions and 1 deletions
@@ -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),
    )
 )