diff --git a/include/aidge/backend/cuda/utils/CudaContext.hpp b/include/aidge/backend/cuda/utils/CudaContext.hpp index 7218cc24aed718f57a1866be74e7ba9124a5a7f1..f21886e502b9017aa55e250e7257d16bc5d04501 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"); }