FIX: fix cmakelists

miguelcarcamov · Nov 5, 2024 · c62d9d9 · c62d9d9
1 parent 2f15c3d
commit c62d9d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -75,7 +75,8 @@ if(Boost_FOUND)
   cuda_detect_installed_gpus(INSTALLED_GPU_CCS_1)
   string(STRIP "${INSTALLED_GPU_CCS_1}" INSTALLED_GPU_CCS_2)
   string(REPLACE " " ";" INSTALLED_GPU_CCS_3 "${INSTALLED_GPU_CCS_2}")
-  string(REPLACE "." "" CUDA_ARCH_LIST "${INSTALLED_GPU_CCS_3}")
+  string(REPLACE "." "" INSTALLED_GPU_CCS_4 "${INSTALLED_GPU_CCS_3}")
+  string(REPLACE "+PTX" "" CUDA_ARCH_LIST "${INSTALLED_GPU_CCS_4}")
   set(CMAKE_CUDA_ARCHITECTURES ${CUDA_ARCH_LIST})
   set_target_properties(gpuvmem PROPERTIES CUDA_ARCHITECTURES
                                            "${CUDA_ARCH_LIST}")