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

Implement simple input layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-12 21:16:46 +01:00
parent 9d91896f13
commit 708164e4d0
5 changed files with 82 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@@ -7,7 +7,6 @@ project(CUDANet
find_package(CUDAToolkit REQUIRED)
include_directories(${CUDAToolkit_INCLUDE_DIRS})
# Add project source files for the library
set(LIBRARY_SOURCES
src/utils/cuda_helper.cu
src/kernels/activations.cu
@@ -15,6 +14,7 @@ set(LIBRARY_SOURCES
src/kernels/matmul.cu
src/layers/dense.cu
src/layers/conv2d.cu
src/layers/input.cu
)
set(CMAKE_CUDA_ARCHITECTURES 75)