mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
Working conv2d forward
This commit is contained in:
@@ -11,6 +11,7 @@ include_directories(${CUDAToolkit_INCLUDE_DIRS})
|
||||
set(LIBRARY_SOURCES
|
||||
src/utils/cuda_helper.cu
|
||||
src/kernels/activations.cu
|
||||
src/kernels/convolution.cu
|
||||
src/kernels/padding.cu
|
||||
src/kernels/matrix_math.cu
|
||||
src/layers/dense.cu
|
||||
|
||||
Reference in New Issue
Block a user