Skip to content
Snippets Groups Projects
Commit d55b42c5 authored by Olivier BICHLER's avatar Olivier BICHLER Committed by Maxence Naud
Browse files

Hotfix for more generic registry

parent 82ec2c54
No related branches found
No related tags found
No related merge requests found
This commit is part of merge request !123. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
include/aidge/backend/cpu/operator/ResizeImpl_kernels.hpp
+ 19
− 18
View file @ d55b42c5
...@@ -99,30 +99,31 @@ void ResizeImpl_cpu_forward_kernel( ...@@ -99,30 +99,31 @@ void ResizeImpl_cpu_forward_kernel(
} }
return; return;
} }
// Kernels registration to implementation entry point // Kernels registration to implementation entry point
REGISTRAR(ResizeImpl_cpu, REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int16}, {{{DataType::Int16},
{DataType::Float32}, {DataType::Any},
{DataType::Float32}, {DataType::Any},
{DataType::UInt64}}, {DataType::Any}},
{DataType::Int16}}, {DataType::Int16}},
{ProdConso::inPlaceModel, {ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int16_t>, ResizeImpl_cpu_forward_kernel<int16_t>,
nullptr}); nullptr});
REGISTRAR(ResizeImpl_cpu, REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int32}, {{{DataType::Int32},
{DataType::Float32}, {DataType::Any},
{DataType::Float32}, {DataType::Any},
{DataType::UInt64}}, {DataType::Any}},
{DataType::Int32}}, {DataType::Int32}},
{ProdConso::inPlaceModel, {ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int32_t>, ResizeImpl_cpu_forward_kernel<int32_t>,
nullptr}); nullptr});
REGISTRAR(ResizeImpl_cpu, REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int64}, {{{DataType::Int64},
{DataType::Float32}, {DataType::Any},
{DataType::Float32}, {DataType::Any},
{DataType::Int64}}, {DataType::Any}},
{DataType::UInt64}}, {DataType::UInt64}},
{ProdConso::inPlaceModel, {ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int64_t>, ResizeImpl_cpu_forward_kernel<int64_t>,
...@@ -130,27 +131,27 @@ REGISTRAR(ResizeImpl_cpu, ...@@ -130,27 +131,27 @@ REGISTRAR(ResizeImpl_cpu,
REGISTRAR(ResizeImpl_cpu, REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float16}, {{{DataType::Float16},
{DataType::Float32}, {DataType::Any},
{DataType::Float32}, {DataType::Any},
{DataType::UInt64}}, {DataType::Any}},
{DataType::Float16}}, {DataType::Float16}},
{ProdConso::inPlaceModel, {ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<half_float::half>, ResizeImpl_cpu_forward_kernel<half_float::half>,
nullptr}); nullptr});
REGISTRAR(ResizeImpl_cpu, REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float32}, {{{DataType::Float32},
{DataType::Float32}, {DataType::Any},
{DataType::Float32}, {DataType::Any},
{DataType::UInt64}}, {DataType::Any}},
{DataType::Float32}}, {DataType::Float32}},
{ProdConso::inPlaceModel, {ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<float>, ResizeImpl_cpu_forward_kernel<float>,
nullptr}); nullptr});
REGISTRAR(ResizeImpl_cpu, REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float64}, {{{DataType::Float64},
{DataType::Float32}, {DataType::Any},
{DataType::Float32}, {DataType::Any},
{DataType::UInt64}}, {DataType::Any}},
{DataType::Float64}}, {DataType::Float64}},
{ProdConso::inPlaceModel, {ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<double>, ResizeImpl_cpu_forward_kernel<double>,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent