From 2a4df30b4b7cd1cd64e40cfcf0dd25224a576ea0 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 5 Aug 2024 09:55:59 +0200
Subject: [PATCH] Fixed compatibility issue
---
include/aidge/backend/cuda/utils/CudaContext.hpp | 2 ++
1 file changed, 2 insertions(+)
diff --git a/include/aidge/backend/cuda/utils/CudaContext.hpp b/include/aidge/backend/cuda/utils/CudaContext.hpp
index 7218cc2..f21886e 100644
--- a/include/aidge/backend/cuda/utils/CudaContext.hpp
+++ b/include/aidge/backend/cuda/utils/CudaContext.hpp
@@ -157,8 +157,10 @@ namespace Aidge {
return CUDNN_DATA_UINT8;
case DataType::Int32:
return CUDNN_DATA_INT32;
+#if CUDNN_VERSION >= 8100
case DataType::Int64:
return CUDNN_DATA_INT64;
+#endif
default:
assert(false && "Unsupported CuDNN type");
}
--
GitLab