diff --git a/CMakeLists.txt b/CMakeLists.txt index b9802b681a7832874d4728c03b1d4befbb50bb13..eeafe753bb1c0271b527e3f59cd07f83ae3db105 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -206,7 +206,6 @@ if(ARB_WITH_GPU) target_compile_options(arbor-private-deps INTERFACE $<$<COMPILE_LANGUAGE:CUDA>:-gencode=arch=compute_37,code=sm_37>) target_compile_options(arbor-private-deps INTERFACE $<$<COMPILE_LANGUAGE:CUDA>:-gencode=arch=compute_60,code=sm_60>) target_compile_options(arbor-private-deps INTERFACE $<$<COMPILE_LANGUAGE:CUDA>:-gencode=arch=compute_70,code=sm_70>) - endif() # Use libunwind if available for pretty printing stack traces diff --git a/arbor/CMakeLists.txt b/arbor/CMakeLists.txt index 87ec5150f5e0a907d66909a4fa7a0fd23e2b490c..6664cb56097cc8341454e63a8b0403dbee87f188 100644 --- a/arbor/CMakeLists.txt +++ b/arbor/CMakeLists.txt @@ -92,6 +92,10 @@ add_library(arbor-private-headers INTERFACE) target_include_directories(arbor-private-headers INTERFACE "$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}>") +if(ARB_WITH_GPU) + target_include_directories(arbor-private-headers INTERFACE ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}) +endif() + install(TARGETS arbor-private-headers EXPORT arbor-targets) # Mechanisms, generated from .mod files; sets arbor_mechanism_sources