diff --git a/cpp/tensorrt_llm/deep_ep/CMakeLists.txt b/cpp/tensorrt_llm/deep_ep/CMakeLists.txt index e00815eed13e..1b2f69a2e8ac 100644 --- a/cpp/tensorrt_llm/deep_ep/CMakeLists.txt +++ b/cpp/tensorrt_llm/deep_ep/CMakeLists.txt @@ -21,7 +21,7 @@ foreach(CUDA_ARCH IN LISTS CMAKE_CUDA_ARCHITECTURES) if(${CUDA_ARCH_MAJOR} GREATER_EQUAL 9) # The FP4-related conversion instructions in DeepEP require SM100a, SM110a, # or SM120a. - if(${CUDA_ARCH_MAJOR} EQUAL 10 AND ${CUDA_ARCH_MINOR} EQUAL 0) + if(${CUDA_ARCH_MAJOR} EQUAL 10) if(CMAKE_VERSION VERSION_GREATER_EQUAL 3.31) list(APPEND DEEP_EP_CUDA_ARCHITECTURES "100f${CUDA_ARCH_POSTFIX}") else()