Fix formatting, use CUDA_VERSION instead of CUDART_VERSION

oneapi-src · Nov 3, 2023 · 7ca370d · 7ca370d
1 parent 3fd39f3
commit 7ca370d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/source/adapters/cuda/command_buffer.cpp b/source/adapters/cuda/command_buffer.cpp
@@ -140,12 +140,13 @@ urCommandBufferReleaseExp(ur_exp_command_buffer_handle_t hCommandBuffer) {
 UR_APIEXPORT ur_result_t UR_APICALL
 urCommandBufferFinalizeExp(ur_exp_command_buffer_handle_t hCommandBuffer) {
   try {
-#if CUDART_VERSION >= 12000
+#if CUDA_VERSION >= 12000
     UR_CHECK_ERROR(cuGraphInstantiate(&hCommandBuffer->CudaGraphExec,
                                       hCommandBuffer->CudaGraph, 0));
 #else
     UR_CHECK_ERROR(cuGraphInstantiate(&hCommandBuffer->CudaGraphExec,
-                                      hCommandBuffer->CudaGraph, nullptr, nullptr, 0));
+                                      hCommandBuffer->CudaGraph, nullptr,
+                                      nullptr, 0));
 #endif
   } catch (...) {
     return UR_RESULT_ERROR_UNKNOWN;