From 202bd1d0177c83c938559cda3db762046375f2e5 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Thu, 23 May 2024 17:52:21 +0200
Subject: [PATCH] fix SliceSize for Slice kernel
---
.../aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
index df2299f6..6b9adc69 100644
--- a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
@@ -44,7 +44,7 @@ void SliceImpl_cpu_forward_kernel(const Slice_Op::Attrs &attrs, const std::vecto
static_cast<DimSize_t>(std::get<1>(attrs)[i] + static_cast<std::int64_t>(inputDims[axis]));
std::int64_t step = std::get<3>(attrs)[i];
- std::size_t sliceSize = static_cast<std::size_t>(std::ceil(static_cast<float>(end - start) / static_cast<float>(step)));
+ std::size_t sliceSize = static_cast<std::size_t>(std::ceil((static_cast<float>(end) - static_cast<float>(start)) / static_cast<float>(step)));
if ( i > 0) {
outputAccumulation = new I[totalSize];
--
GitLab