diff --git a/include/aidge/backend/cuda/data/TensorImpl.hpp b/include/aidge/backend/cuda/data/TensorImpl.hpp index 96045781647f93f0627ca0853a0cdaa66a08af83..28b193610309fee51e3fd2d384fc0e84f6a57404 100644 --- a/include/aidge/backend/cuda/data/TensorImpl.hpp +++ b/include/aidge/backend/cuda/data/TensorImpl.hpp @@ -255,7 +255,7 @@ 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); + {"cuda", DataType::Int32}, Aidge::TensorImpl_cuda<int32_t>::create); } // namespace } // namespace Aidge