diff --git a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp --- a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp +++ b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp @@ -296,6 +296,12 @@ defaultDevice = device; } +/// +/// Runtime methods using CUDA 12.0+ driver +/// + +#if (CUDA_VERSION >= 12000) + extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuTensorMapEncodeTiled( CUtensorMap *tensorMap, // Tensor map object CUtensorMapDataType tensorDataType, // Tensor data type @@ -378,6 +384,7 @@ sizeof(CUtensorMap))); return reinterpret_cast(dTensorMap); } +#endif #ifdef MLIR_ENABLE_CUDA_CUSPARSE