Skip to content
Snippets Groups Projects
Commit c48d6b2f authored by Charles Villard's avatar Charles Villard
Browse files

edit: CMake: added all gpu architecture target

parent b0e3c633
No related branches found
No related tags found
No related merge requests found
Pipeline #60755 failed
......@@ -80,11 +80,14 @@ message(STATUS "Cuda compiler version = ${CMAKE_CUDA_COMPILER_VERSION}")
# Define a preprocessor macro with the Cuda compiler version
add_definitions(-DCUDA_COMPILER_VERSION="${CMAKE_CUDA_COMPILER_VERSION}")
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86)
message(STATUS "CUDA STANDARD : ${CMAKE_CUDA_STANDARD}")
message(STATUS "CUDA ARCHITECTURE : ${CMAKE_CUDA_ARCHITECTURES}")
enable_language(CUDA)
find_package(CUDNN REQUIRED)
add_library(aidge::cudnn INTERFACE IMPORTED)
target_include_directories(aidge::cudnn INTERFACE ${CUDNN_INCLUDE_PATH})
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment