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

Restructure cuda backend

Browse Source
This commit is contained in:
Matúš Námešný
2024-09-05 22:23:47 +02:00
parent 65727dfee8
commit f8220f0ec1
19 changed files with 69 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
CMakeLists.txt
View File

@@ -20,6 +20,7 @@ if(USE_CUDA)
endif()
file(GLOB_RECURSE CPU_SOURCES
src/layers/*.cpp
src/model/*.cpp
)
@@ -27,10 +28,11 @@ set(LIBRARY_SOURCES ${CPU_SOURCES})
if(USE_CUDA)
file(GLOB_RECURSE CUDA_SOURCES
src/*.cu
src/cuda/utils/*.cu
src/cuda/kernels/*.cu
src/cuda/layers/*.cu
src/backends/cuda/*.cu
src/backends/cuda/utils/*.cu
src/backends/cuda/kernels/*.cu
src/backends/cuda/layers/*.cu
src/layers/*.cu # To be removed
)
set(LIBRARY_SOURCES ${LIBRARY_SOURCES} ${CUDA_SOURCES})
endif()