summarylogtreecommitdiffstats
path: root/030-tensorrt-fix-gpu-archs-list.patch
blob: aadbc5e09da8435f6e4440bf90de7e8072c3d143 (plain)
1
2
3
4
5
6
7
8
9
10
11
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -177,7 +177,7 @@ if (DEFINED GPU_ARCHS)
   message(STATUS "GPU_ARCHS defined as ${GPU_ARCHS}. Generating CUDA code for SM ${GPU_ARCHS}")
   separate_arguments(GPU_ARCHS)
   foreach(SM IN LISTS GPU_ARCHS)
-    list(APPEND CMAKE_CUDA_ARCHITECTURES SM)
+    list(APPEND CMAKE_CUDA_ARCHITECTURES ${SM})
   endforeach()
 else()
   list(APPEND CMAKE_CUDA_ARCHITECTURES 72 75 80 86 87 89 90)