From 3be714857faff91636fc0736b11654ae404c4ee8 Mon Sep 17 00:00:00 2001
From: Jerome Hue <jerome.hue@cea.fr>
Date: Mon, 17 Mar 2025 16:06:57 +0100
Subject: [PATCH] Include <cmath> in HeavisideImpl kernels
---
include/aidge/backend/cpu/operator/HeavisideImpl_kernels.hpp | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/include/aidge/backend/cpu/operator/HeavisideImpl_kernels.hpp b/include/aidge/backend/cpu/operator/HeavisideImpl_kernels.hpp
index 4e2a7db2..92f12fbe 100644
--- a/include/aidge/backend/cpu/operator/HeavisideImpl_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/HeavisideImpl_kernels.hpp
@@ -15,6 +15,7 @@
#include "aidge/utils/Registrar.hpp"
#include <cstddef> // std::size_t
+#include <cmath>
#include "aidge/backend/cpu/operator/HeavisideImpl.hpp"
#include "aidge/utils/ErrorHandling.hpp"
@@ -37,7 +38,7 @@ void HeavisideImplCpuForwardKernel(std::size_t inputLength,
// Surrogate Gradient
template <class O, class GO, class GI>
-void HeavisideImplCpuBackwardKernel(std::size_t inputLength,
+void HeavisideImplCpuBackwardKernel(std::size_t inputLength,
const void* output_,
const void* grad_output_,
void* grad_input_) {
--
GitLab