We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 71be020 commit 1dc0071Copy full SHA for 1dc0071
mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
@@ -296,6 +296,12 @@ extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuSetDefaultDevice(int32_t device) {
296
defaultDevice = device;
297
}
298
299
+///
300
+/// Runtime methods using CUDA 12.0+ driver
301
302
+
303
+#if (CUDA_VERSION >= 12000)
304
305
extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuTensorMapEncodeTiled(
306
CUtensorMap *tensorMap, // Tensor map object
307
CUtensorMapDataType tensorDataType, // Tensor data type
@@ -378,6 +384,7 @@ extern "C" MLIR_CUDA_WRAPPERS_EXPORT void *mgpuTensorMapEncodeTiledMemref(
378
384
sizeof(CUtensorMap)));
379
385
return reinterpret_cast<void *>(dTensorMap);
380
386
387
+#endif
381
388
382
389
#ifdef MLIR_ENABLE_CUDA_CUSPARSE
383
390
0 commit comments