From a3973f0b2110a76086757643957715098586018d Mon Sep 17 00:00:00 2001 From: LordMathis Date: Mon, 11 Mar 2024 21:05:38 +0100 Subject: [PATCH] Add activation to conv2d --- src/layers/conv2d.cu | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/src/layers/conv2d.cu b/src/layers/conv2d.cu index 4518c01..cc156d7 100644 --- a/src/layers/conv2d.cu +++ b/src/layers/conv2d.cu @@ -122,6 +122,19 @@ void Layers::Conv2d::forward(const float* d_input, float* d_output) { d_biases, d_output, d_output, biases.size() ); + switch (activation) { + case SIGMOID: + Kernels::sigmoid<<<1, outputSize>>>(d_output, d_output, outputSize); + break; + + case RELU: + Kernels::relu<<<1, outputSize>>>(d_output, d_output, outputSize); + break; + + default: + break; + } + CUDA_CHECK(cudaDeviceSynchronize()); }