From b76f54942333531f202deea496c5b319d7317be0 Mon Sep 17 00:00:00 2001
From: Mickael GUIBERT <mickael.guibert@cea.fr>
Date: Tue, 1 Apr 2025 12:46:52 +0000
Subject: [PATCH] [Fix] Suppress constexpr and simplified the choice of
rounding
---
aidge_export_cpp/kernels/pooling.hpp | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/aidge_export_cpp/kernels/pooling.hpp b/aidge_export_cpp/kernels/pooling.hpp
index 0829cc4..85fece5 100644
--- a/aidge_export_cpp/kernels/pooling.hpp
+++ b/aidge_export_cpp/kernels/pooling.hpp
@@ -116,11 +116,9 @@ void pooling_forward(
}
}
- if constexpr (std::is_integral<Output_T>::value) {
- outputs[oOffset + output] = (Output_T) std::round((float)sum / (POOL_HEIGHT * POOL_WIDTH));
- } else {
- outputs[oOffset + output] = (Output_T) (sum / (POOL_HEIGHT * POOL_WIDTH));
- }
+ outputs[oOffset + output] = static_cast<Output_T>(
+ std::is_integral<Output_T>::value ? std::round((float)sum / (POOL_HEIGHT * POOL_WIDTH)) : sum / (POOL_HEIGHT * POOL_WIDTH)
+ );
}
else {
--
GitLab