From 764923ca1d684b49dc023b2c93193e7de538da85 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Wed, 13 Dec 2023 17:08:06 +0100
Subject: [PATCH] change Slice attr to int64
---
.../backend/cpu/operator/SliceImpl_forward_kernels.hpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
index c501a55e..58cfe400 100644
--- a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
@@ -33,9 +33,9 @@ void SliceImpl_cpu_forward_kernel(const typename Slice_Op::Attrs& attrs,
for(std::size_t i=0; i<nbAxes;++i)
{
// For each slice operation get the params and cast them to size_t
- std::int32_t axis_ = std::get<2>(attrs)[i];
- std::int32_t start_ = std::get<0>(attrs)[i];
- std::int32_t end_ = std::get<1>(attrs)[i];
+ std::int64_t axis_ = std::get<2>(attrs)[i];
+ std::int64_t start_ = std::get<0>(attrs)[i];
+ std::int64_t end_ = std::get<1>(attrs)[i];
std::size_t axis = axis_>=0?axis_:axis_+inputDims.size();
std::size_t start = start_>=0?start_:start_+inputDims[axis];
std::size_t end = end_>=0?end_:end_+inputDims[axis];
--
GitLab