From 03a5865251b0fb469e15ad0010cb2279fae36608 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Wed, 31 Jan 2024 17:23:34 +0100
Subject: [PATCH] change attrs back to int64_t
---
.../backend/cpu/operator/GatherImpl_forward_kernels.hpp | 2 +-
.../backend/cpu/operator/SliceImpl_forward_kernels.hpp | 6 +++---
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp
index 7cb1f6c0..0d312e3c 100644
--- a/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp
@@ -40,7 +40,7 @@ void GatherImpl_cpu_forward_kernel(const typename Gather_Op::Attrs& attrs, const
preAxisElems *= inputDims[i];
}
- const std::vector<std::int32_t> indices = std::get<0>(attrs);
+ const std::vector<std::int64_t> indices = std::get<0>(attrs);
for (std::size_t i=0; i<preAxisElems; ++i)
{
for(std::size_t j=0; j<indices.size(); ++j)
diff --git a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
index c18a164b..d92e9008 100644
--- a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
@@ -32,9 +32,9 @@ void SliceImpl_cpu_forward_kernel(const typename Slice_Op::Attrs& attrs,
DimSize_t nbAxes = std::get<2>(attrs).size();
for (std::size_t i = 0; i < nbAxes; ++i) {
// For each slice operation get the params and cast them to size_t
- const std::int32_t axis_ = std::get<2>(attrs)[i];
- const std::int32_t start_ = std::get<0>(attrs)[i];
- const std::int32_t end_ = std::get<1>(attrs)[i];
+ const std::int64_t axis_ = std::get<2>(attrs)[i];
+ const std::int64_t start_ = std::get<0>(attrs)[i];
+ const std::int64_t end_ = std::get<1>(attrs)[i];
const std::size_t axis = axis_ >= 0 ? axis_ : static_cast<std::size_t>(axis_) + inputDims.size();
const std::size_t start = start_ >= 0 ? start_ : start_ + inputDims[axis];
const std::size_t end = end_ >= 0 ? end_ : end_ + inputDims[axis];
--
GitLab