From 04631d98f5cc785835d6cc0bf7adbd75ba218dbc Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Thu, 15 Aug 2024 10:48:04 +0200
Subject: [PATCH] Fix critical error
---
include/aidge/backend/cuda/data/TensorImpl.hpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/include/aidge/backend/cuda/data/TensorImpl.hpp b/include/aidge/backend/cuda/data/TensorImpl.hpp
index 9604578..28b1936 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
--
GitLab