From 7581ea6363e277f9264a8d0d3218ec675f7438db Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Thu, 23 May 2024 18:06:41 +0200
Subject: [PATCH] fix output type 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 6b9adc69..dc274115 100644
--- a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp
@@ -24,7 +24,7 @@ namespace Aidge {
template<class I, class O>
void SliceImpl_cpu_forward_kernel(const Slice_Op::Attrs &attrs, const std::vector<DimSize_t>&inputDims, const void *input_, void *output_){
const I* input = static_cast<const I*>(input_);
- I* output = static_cast<I*>(output_);
+ O* output = static_cast<O*>(output_);
const std::size_t nbDims = inputDims.size();
std::vector<DimSize_t> dims = inputDims;
--
GitLab