Fix : fix moe regression for sm120 (#5823)

peaceh-nv · web-flow · commit 52684d79f791 · 2025-07-09T21:25:11.000+08:00
Signed-off-by: peaceh &lt;103117813+peaceh-nv@users.noreply.github.com&gt;
diff --git a/cpp/tensorrt_llm/kernels/cutlass_kernels/CMakeLists.txt b/cpp/tensorrt_llm/kernels/cutlass_kernels/CMakeLists.txt
@@ -119,7 +119,7 @@ function(add_instantiations library base_dir)
 
       if(${ARCH} EQUAL 90)
         process_target(${TARGET_NAME} true false)
-      elseif(${ARCH} EQUAL 100)
+      elseif(${ARCH} GREATER_EQUAL 100)
         process_target(${TARGET_NAME} false true)
       endif()
     endif()
