[OpenMP] Replace CUDART_VERSION with CUDA_VERSION

llvm · Nov 6, 2023 · 002f422 · 002f422
1 parent 726ee40
commit 002f422
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openmp/libomptarget/plugins-nextgen/cuda/src/rtl.cpp b/openmp/libomptarget/plugins-nextgen/cuda/src/rtl.cpp
@@ -37,7 +37,7 @@ struct CUDAKernelTy;
 struct CUDADeviceTy;
 struct CUDAPluginTy;
 
-#if (defined(CUDART_VERSION) && (CUDART_VERSION < 11000))
+#if (defined(CUDA_VERSION) && (CUDA_VERSION < 11000))
 /// Forward declarations for all Virtual Memory Management
 /// related data structures and functions. This is necessary
 /// for older cuda versions.
@@ -543,7 +543,7 @@ struct CUDADeviceTy : public GenericDeviceTy {
 
   /// CUDA support VA management
   bool supportVAManagement() const override {
-#if (defined(CUDART_VERSION) && (CUDART_VERSION >= 11000))
+#if (defined(CUDA_VERSION) && (CUDA_VERSION >= 11000))
     return true;
 #else
     return false;