diff --git a/CMakeLists.txt b/CMakeLists.txt index 00d6d6cd484ad4a7cb506185ec4ae01978f3f5ee..1b5a43fc9f6012f0ae1cd4bf63e65b25eeb92e52 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -5,7 +5,7 @@ file(STRINGS "${CMAKE_SOURCE_DIR}/version.txt" version) project(aidge_learning VERSION ${version} - DESCRIPTION "Functions and alogrithms to train models in the AIDGE framework" + DESCRIPTION "Functions and alogrithms to train models in the AIDGE framework" LANGUAGES CXX) message(STATUS "Project name: ${CMAKE_PROJECT_NAME}") @@ -42,12 +42,6 @@ if(NOT $ENV{AIDGE_INSTALL} STREQUAL "") endif() find_package(aidge_core REQUIRED) -# Enable CUDA if backend_cuda is found -find_package(aidge_backend_cuda) - -if(aidge_backend_cuda_FOUND) - enable_language(CUDA) -endif() ############################################## # Create target and set properties diff --git a/unit_tests/CMakeLists.txt b/unit_tests/CMakeLists.txt index 39af9ec9b609b8b2c5685699ac014406ed64df72..07d35e3360735434412ac7bbd659e3d8ad8240de 100644 --- a/unit_tests/CMakeLists.txt +++ b/unit_tests/CMakeLists.txt @@ -45,11 +45,11 @@ if(aidge_backend_cuda_FOUND) # Enable CUDA language support and separable compilation for the target enable_language(CUDA) set_target_properties(${tests_exe} PROPERTIES CUDA_SEPARABLE_COMPILATION ON) - + target_include_directories(${tests_exe} PRIVATE ${CUDAToolkit_INCLUDE_DIRS}) # Link manually specified CUDA libraries if the targets are not available target_link_libraries(${tests_exe} - PUBLIC + PUBLIC _aidge_backend_cuda CUDA::cudart CUDA::cublas