From 452a65e6aad19ec2a5f46cf35463fc1c73580d95 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Tue, 26 Nov 2024 19:55:24 +0000
Subject: [PATCH] fix 'And' kernel in case of equal dimensions for inputs
---
include/aidge/backend/cpu/operator/AndImpl_kernels.hpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/include/aidge/backend/cpu/operator/AndImpl_kernels.hpp b/include/aidge/backend/cpu/operator/AndImpl_kernels.hpp
index def1ebfe..73b710e0 100644
--- a/include/aidge/backend/cpu/operator/AndImpl_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/AndImpl_kernels.hpp
@@ -62,7 +62,7 @@ void EqualImpl_cpu_forward_kernel(std::vector<std::size_t> dims0,
const std::size_t input0_contiguous_size = std::accumulate(dims0.cbegin(), dims0.cend(), std::size_t(1), std::multiplies<std::size_t>());
for (std::size_t i = 0; i < input0_contiguous_size; ++i)
{
- output[i] = static_cast<O>(input_0[i] + input_1[i]);
+ output[i] = static_cast<O>(input_0[i] == input_1[i]);
}
return;
}
--
GitLab