kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None][fix] Fix regex pattern for cubin filtering (#9914)

Browse Source 
Signed-off-by: Anthony Chang <27950904+rosenrodt@users.noreply.github.com>
This commit is contained in:
Anthony Chang 2025-12-15 10:02:48 +08:00 committed by GitHub
parent bf923a1074
commit 3be5f3abcf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cpp/tensorrt_llm/kernels/CMakeLists.txt
View File

@ -40,9 +40,7 @@ list(FILTER SRC_CU EXCLUDE REGEX "fusedLayernormKernels/.*")
function(filter_cuda_archs ARCH SOURCES_VAR)
if(NOT "${ARCH}" IN_LIST CMAKE_CUDA_ARCHITECTURES_ORIG)
set(FILTER_REGEX
".*_sm(_)?${ARCH}[.]cubin[.]cpp|^.*Sm(_)?${ARCH}.*cubin.cpp$|.*_sm(_)?${ARCH}[.]cu|^.*Sm(_)?${ARCH}.*cu$"
)
set(FILTER_REGEX ".*[Ss][Mm]_?${ARCH}(af)?.*(cubin\.cpp|\.cu)$")
list(APPEND SOURCES ${${SOURCES_VAR}})
list(APPEND SOURCES_FILTERED ${SOURCES})
list(FILTER SOURCES_FILTERED INCLUDE REGEX "${FILTER_REGEX}")