Implement max pooling layer

2025-11-08 10:44:22 +00:00 · 2024-03-19 22:04:58 +01:00
parent 364715ff70
commit a0fc1b00ae
9 changed files with 245 additions and 21 deletions
--- a/src/kernels/convolution.cu
+++ b/src/kernels/convolution.cu
@@ -2,9 +2,9 @@

 #include "convolution.cuh"

-using namespace CUDANet::Kernels;
+using namespace CUDANet;

-__global__ void convolution(
+__global__ void Kernels::convolution(
    const float* __restrict__ d_input,
    const float* __restrict__ d_kernel,
    float* __restrict__ d_output,