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

Initialize conv2d layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-04 22:16:03 +01:00
parent f37320594a
commit cfc5c46d5e
5 changed files with 148 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
CMakeLists.txt
View File

@@ -13,6 +13,7 @@ set(LIBRARY_SOURCES
src/kernels/activations.cu
src/kernels/padding.cu
src/layers/dense.cu
src/layers/conv2d.cu
)
set(CMAKE_CUDA_ARCHITECTURES 75)