mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 22:34:22 +00:00
Update CMakeLists.txt
This commit is contained in:
@@ -23,8 +23,8 @@ endif()
|
|||||||
|
|
||||||
|
|
||||||
file(GLOB_RECURSE CPU_SOURCES
|
file(GLOB_RECURSE CPU_SOURCES
|
||||||
|
src/*.cpp
|
||||||
src/layers/*.cpp
|
src/layers/*.cpp
|
||||||
src/model/*.cpp
|
|
||||||
)
|
)
|
||||||
|
|
||||||
set(LIBRARY_SOURCES ${CPU_SOURCES})
|
set(LIBRARY_SOURCES ${CPU_SOURCES})
|
||||||
@@ -32,10 +32,7 @@ set(LIBRARY_SOURCES ${CPU_SOURCES})
|
|||||||
if(USE_CUDA)
|
if(USE_CUDA)
|
||||||
file(GLOB_RECURSE CUDA_SOURCES
|
file(GLOB_RECURSE CUDA_SOURCES
|
||||||
src/backends/cuda/*.cu
|
src/backends/cuda/*.cu
|
||||||
src/backends/cuda/utils/*.cu
|
|
||||||
src/backends/cuda/kernels/*.cu
|
src/backends/cuda/kernels/*.cu
|
||||||
src/backends/cuda/layers/*.cu
|
|
||||||
src/layers/*.cu # To be removed
|
|
||||||
)
|
)
|
||||||
set(LIBRARY_SOURCES ${LIBRARY_SOURCES} ${CUDA_SOURCES})
|
set(LIBRARY_SOURCES ${LIBRARY_SOURCES} ${CUDA_SOURCES})
|
||||||
endif()
|
endif()
|
||||||
@@ -52,11 +49,6 @@ endif()
|
|||||||
# Set include directories for the library
|
# Set include directories for the library
|
||||||
target_include_directories(${PROJECT_NAME} PUBLIC
|
target_include_directories(${PROJECT_NAME} PUBLIC
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/include
|
${CMAKE_CURRENT_SOURCE_DIR}/include
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/include/utils
|
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/include/kernels
|
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/include/layers
|
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/include/model
|
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/src
|
|
||||||
)
|
)
|
||||||
|
|
||||||
set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 20)
|
set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 20)
|
||||||
|
|||||||
Reference in New Issue
Block a user