diff --git a/include/aidge/backend/cuda/data/TensorImpl.hpp b/include/aidge/backend/cuda/data/TensorImpl.hpp index 7d7b33f77237f40eb12d8aee794aaecf32804e15..176624b0df9f0a90eba5d8afe9496125f2aad898 100644 --- a/include/aidge/backend/cuda/data/TensorImpl.hpp +++ b/include/aidge/backend/cuda/data/TensorImpl.hpp @@ -235,7 +235,7 @@ static Registrar<Tensor> registrarTensorImpl_cuda_Float64( {"cuda", DataType::Float64}, Aidge::TensorImpl_cuda<double>::create); static Registrar<Tensor> registrarTensorImpl_cuda_Float32( {"cuda", DataType::Float32}, Aidge::TensorImpl_cuda<float>::create); -static Registrar<Tensor> registrarTensorImpl_cuda_Float32( +static Registrar<Tensor> registrarTensorImpl_cuda_Float16( {"cuda", DataType::Float16}, Aidge::TensorImpl_cuda<half_float::half>::create); static Registrar<Tensor> registrarTensorImpl_cuda_Int32( {"cuda", DataType::Int32}, Aidge::TensorImpl_cuda<int>::create);