Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Split tests to cpu and cuda

Browse Source
This commit is contained in:
Matúš Námešný
2024-09-07 21:24:34 +02:00
parent f8220f0ec1
commit 591507ed21
16 changed files with 29 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
CMakeLists.txt
View File

@@ -17,8 +17,12 @@ endif()
if(USE_CUDA)
enable_language(CUDA)
add_definitions(-DUSE_CUDA)
message(STATUS "Building library with CUDA support")
else()
message(STATUS "Building library without CUDA support")
endif()
file(GLOB_RECURSE CPU_SOURCES
src/layers/*.cpp
src/model/*.cpp