From cda4b85e8437f5bd722f8451bf765f15f031edb7 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 28 Apr 2025 16:20:24 +0200
Subject: [PATCH] Added no bias indirection
---
aidge_export_cpp/kernels/convolution.hpp | 41 ++++++++++++++++++++++++
1 file changed, 41 insertions(+)
diff --git a/aidge_export_cpp/kernels/convolution.hpp b/aidge_export_cpp/kernels/convolution.hpp
index 1669ee8..5855654 100644
--- a/aidge_export_cpp/kernels/convolution.hpp
+++ b/aidge_export_cpp/kernels/convolution.hpp
@@ -272,4 +272,45 @@ void convolution_depthwise_forward(
}
}
+// Template specialization when biases are not given to the convolution
+template<int NB_CHANNELS,
+ int CHANNELS_HEIGHT, int CHANNELS_WIDTH,
+ int NB_OUTPUTS,
+ int OUTPUTS_HEIGHT, int OUTPUTS_WIDTH,
+ int PADDING_Y, int PADDING_X,
+ int STRIDE_Y, int STRIDE_X,
+ int DILATION_Y, int DILATION_X,
+ int KERNEL_HEIGHT, int KERNEL_WIDTH,
+ ActivationFunction_T ACTIVATION,
+ typename Input_T, typename Output_T,
+ typename Weight_T,
+ typename Rescaling_T>
+__attribute__((always_inline)) inline
+void convolution_depthwise_forward(
+ const Input_T* __restrict inputs,
+ Output_T* __restrict outputs,
+ const Weight_T* __restrict weights,
+ std::nullptr_t __restrict,
+ const Rescaling_T& __restrict rescaling)
+{
+ const float* b = nullptr;
+
+ convolution_depthwise_forward<NB_CHANNELS,
+ CHANNELS_HEIGHT,
+ CHANNELS_WIDTH,
+ NB_OUTPUTS,
+ OUTPUTS_HEIGHT,
+ OUTPUTS_WIDTH,
+ PADDING_Y,
+ PADDING_X,
+ STRIDE_Y,
+ STRIDE_X,
+ DILATION_Y,
+ DILATION_X,
+ KERNEL_HEIGHT,
+ KERNEL_WIDTH,
+ ACTIVATION>
+ (inputs, outputs, weights, b, rescaling);
+}
+
#endif // __AIDGE_EXPORT_CPP_KERNELS_CONVOLUTION__
--
GitLab