Skip to content
Snippets Groups Projects
Commit 3be71485 authored by Jerome Hue's avatar Jerome Hue
Browse files

Include <cmath> in HeavisideImpl kernels

parent e7c9550d
No related branches found
No related tags found
2 merge requests!166Update 0.5.0 -> 0.6.0,!146Implement a backward for Heaviside
Pipeline #68012 failed
......@@ -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_) {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment