From d55b42c57bb39e2d74b2d342106e0c810fc1c543 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Wed, 18 Dec 2024 15:06:43 +0100
Subject: [PATCH] Hotfix for more generic registry
---
.../cpu/operator/ResizeImpl_kernels.hpp | 37 ++++++++++---------
1 file changed, 19 insertions(+), 18 deletions(-)
diff --git a/include/aidge/backend/cpu/operator/ResizeImpl_kernels.hpp b/include/aidge/backend/cpu/operator/ResizeImpl_kernels.hpp
index 6a22ff4e..6449417b 100644
--- a/include/aidge/backend/cpu/operator/ResizeImpl_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/ResizeImpl_kernels.hpp
@@ -99,30 +99,31 @@ void ResizeImpl_cpu_forward_kernel(
}
return;
}
+
// Kernels registration to implementation entry point
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int16},
- {DataType::Float32},
- {DataType::Float32},
- {DataType::UInt64}},
+ {DataType::Any},
+ {DataType::Any},
+ {DataType::Any}},
{DataType::Int16}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int16_t>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int32},
- {DataType::Float32},
- {DataType::Float32},
- {DataType::UInt64}},
+ {DataType::Any},
+ {DataType::Any},
+ {DataType::Any}},
{DataType::Int32}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int32_t>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int64},
- {DataType::Float32},
- {DataType::Float32},
- {DataType::Int64}},
+ {DataType::Any},
+ {DataType::Any},
+ {DataType::Any}},
{DataType::UInt64}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int64_t>,
@@ -130,27 +131,27 @@ REGISTRAR(ResizeImpl_cpu,
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float16},
- {DataType::Float32},
- {DataType::Float32},
- {DataType::UInt64}},
+ {DataType::Any},
+ {DataType::Any},
+ {DataType::Any}},
{DataType::Float16}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<half_float::half>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float32},
- {DataType::Float32},
- {DataType::Float32},
- {DataType::UInt64}},
+ {DataType::Any},
+ {DataType::Any},
+ {DataType::Any}},
{DataType::Float32}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<float>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float64},
- {DataType::Float32},
- {DataType::Float32},
- {DataType::UInt64}},
+ {DataType::Any},
+ {DataType::Any},
+ {DataType::Any}},
{DataType::Float64}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<double>,
--
GitLab