diff --git a/include/aidge/backend/cpu/operator/ReLUImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/ReLUImpl_forward_kernels.hpp index 90b22c5fa8526115122fef9a0f58322af513b302..aa533786d3ce5b6f5cd501b6ba74b1be2823d407 100644 --- a/include/aidge/backend/cpu/operator/ReLUImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/ReLUImpl_forward_kernels.hpp @@ -25,7 +25,7 @@ void ReLUImpl_cpu_forward_kernel(std::size_t inputLenght, const I* input = static_cast<const I*>(input_); O* output = static_cast<O*>(output_); -#pragma omp parallel for if (inputLenght > 1024) +//#pragma omp parallel for if (inputLenght > 1024) for (std::size_t i = 0; i < inputLenght; ++i) { output[i] = input[i] > 0 ? input[i] : 0; } diff --git a/include/aidge/backend/cpu/operator/SigmoidImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/SigmoidImpl_forward_kernels.hpp index 96303312aae067c6955c96331f7cd7d959de53a7..a53650942540e6368855ffe19e2f7f651ab5b6bc 100644 --- a/include/aidge/backend/cpu/operator/SigmoidImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/SigmoidImpl_forward_kernels.hpp @@ -25,7 +25,7 @@ void SigmoidImpl_cpu_forward_kernel(std::size_t inputLenght, const I* input = static_cast<const I*>(input_); O* output = static_cast<O*>(output_); -#pragma omp parallel for if (inputLenght > 1024) +//#pragma omp parallel for if (inputLenght > 1024) for (std::size_t i = 0; i < inputLenght; ++i) { output[i] = static_cast<O>(1.0) / (static_cast<O>(1.0) + std::exp(-input[i])); } diff --git a/include/aidge/backend/cpu/operator/TanhImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/TanhImpl_forward_kernels.hpp index 3012aae9e4a8a587efde0b8221b8c55c4d832345..9e57b6dfcb0da322f5b21944fb10ec7a10cd0ab8 100644 --- a/include/aidge/backend/cpu/operator/TanhImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/TanhImpl_forward_kernels.hpp @@ -25,7 +25,7 @@ void TanhImpl_cpu_forward_kernel(std::size_t inputLenght, const I* input = static_cast<const I*>(input_); O* output = static_cast<O*>(output_); -#pragma omp parallel for if (inputLenght > 1024) +//#pragma omp parallel for if (inputLenght > 1024) for (std::size_t i = 0; i < inputLenght; ++i) { output[i] = std::tanh(input[i]); }