From bc86ed1782ff23f272f3913226e01985a5ef2ce4 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Thu, 11 Apr 2024 19:50:54 +0200
Subject: [PATCH] Add activation to pooling layers

---
 src/layers/avg_pooling.cu | 2 ++
 src/layers/max_pooling.cu | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/layers/avg_pooling.cu b/src/layers/avg_pooling.cu
index aedcc13..b37bb9b 100644
--- a/src/layers/avg_pooling.cu
+++ b/src/layers/avg_pooling.cu
@@ -43,5 +43,7 @@ float* AvgPooling2D::forward(const float* d_input) {
         d_input, d_output, inputSize, outputSize, nChannels, poolingSize, stride
     );
 
+    activation.activate(d_output);
+
     return d_output;
 }
\ No newline at end of file
diff --git a/src/layers/max_pooling.cu b/src/layers/max_pooling.cu
index 3f1f426..f6ae351 100644
--- a/src/layers/max_pooling.cu
+++ b/src/layers/max_pooling.cu
@@ -46,5 +46,7 @@ float* MaxPooling2D::forward(const float* d_input) {
         d_input, d_output, inputSize, outputSize, nChannels, poolingSize, stride
     );
 
+    activation.activate(d_output);
+
     return d_output;
 }
\ No newline at end of file