[None][infra] Recognize SM107 (Rubin) in build config and arch detection

cpp/cmake/modules/cuda_configuration.cmake

@@ -61,7 +61,7 @@ Functions
 
   ``all`` or unset
     Resolves to architectures TensorRT-LLM is optimized for and the
-    compiler supports (80, 86, 89, 90, 100, 103, 120 depending on CUDA version).
+    compiler supports (80, 86, 89, 90, 100, 103, 107, 120 depending on CUDA version).
 
   ``all-major``
     Unsupported. Results in a fatal error.
@@ -341,6 +341,9 @@ function(setup_cuda_architectures)
     if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "12.9")
       list(APPEND CMAKE_CUDA_ARCHITECTURES_RAW 103)
     endif()
+    if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "13.4")
+      list(APPEND CMAKE_CUDA_ARCHITECTURES_RAW 107)
+    endif()
   endif()
 
   # CMAKE_CUDA_ARCHITECTURES_ORIG contains all architectures enabled, without
@@ -431,6 +434,7 @@ function(setup_cuda_architectures)
       90
       100
       103
+      107
       120)
   foreach(CUDA_ARCH IN LISTS ARCHITECTURES_WITH_KERNELS)
     if(NOT ${CUDA_ARCH} IN_LIST CMAKE_CUDA_ARCHITECTURES_ORIG)

cpp/include/tensorrt_llm/common/cudaUtils.h

@@ -306,10 +306,12 @@ inline int getSMVersion(bool queryRealSmArch = false)
     return sm;
 }
 
-inline bool isSM100Family()
+inline bool isSM100Family(std::optional<int> sm = std::nullopt)
 {
-    int const sm = getSMVersion();
-    return sm == 100 || sm == 103; // To be continued...
+    // Not value_or(): its argument is evaluated eagerly, so an explicit sm
+    // would still trigger a device query (which throws with no device present).
+    int smVersion = sm.has_value() ? *sm : getSMVersion();
+    return smVersion >= 100 && smVersion < 110;
 }
 
 inline int getDevice()

tensorrt_llm/_utils.py

@@ -683,7 +683,7 @@ def get_sm_version():
 def is_sm_100f(sm_version=None):
     if sm_version is None:
         sm_version = get_sm_version()
-    return sm_version == 100 or sm_version == 103
+    return sm_version >= 100 and sm_version < 110
 
 
 @lru_cache(maxsize=1)

tests/integration/defs/conftest.py

@@ -1585,7 +1585,7 @@ def get_sm_version():
 def is_sm_100f(sm_version=None):
     if sm_version is None:
         sm_version = get_sm_version()
-    return sm_version == 100 or sm_version == 103
+    return sm_version >= 100 and sm_version < 110
 
 
 def get_gpu_device_list():
@@ -1641,11 +1641,21 @@ def is_ipc_nvls_supported():
     reason="This test is not supported in post-Blackwell architecture",
 )
 
+skip_no_rubin = pytest.mark.skipif(
+    get_sm_version() != 107,
+    reason="This test is only supported in Rubin architecture",
+)
+
 skip_post_blackwell_ultra = pytest.mark.skipif(
     get_sm_version() >= 103,
     reason="This test is not supported in post-Blackwell-Ultra architecture",
 )
 
+skip_pre_rubin = pytest.mark.skipif(
+    get_sm_version() < 107,
+    reason="This test is not supported in pre-Rubin architecture",
+)
+
 skip_device_contain_gb200 = pytest.mark.skipif(
     check_device_contain(["GB200"]),
     reason="This test is not supported on GB200 or GB100",