Merge pull request #1682 from JackAKirk/launch-attr-cuda1180

kbenzie · web-flow · commit 15752ab5d674 · 2024-05-30T10:17:49.000+01:00
[cuda] Add cuda 11.8 #ifdef for `cuEnqueueKernelEx`.
diff --git a/source/adapters/cuda/enqueue.cpp b/source/adapters/cuda/enqueue.cpp
@@ -553,7 +553,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueKernelLaunchCustomExp(
                           pLocalWorkSize, numEventsInWaitList, phEventWaitList,
                           phEvent);
   }
-
+#if CUDA_VERSION >= 11080
   // Preconditions
   UR_ASSERT(hQueue->getDevice() == hKernel->getProgram()->getDevice(),
             UR_RESULT_ERROR_INVALID_KERNEL);
@@ -722,6 +722,11 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueKernelLaunchCustomExp(
     return Err;
   }
   return UR_RESULT_SUCCESS;
+#else
+  setErrorMessage("This feature requires cuda 11.8 or later.",
+                  UR_RESULT_ERROR_ADAPTER_SPECIFIC);
+  return UR_RESULT_ERROR_ADAPTER_SPECIFIC;
+#endif // CUDA_VERSION >= 11080
 }
 
 /// Set parameters for general 3D memory copy.

Original file line number	Diff line number	Diff line change
`@@ -553,7 +553,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueKernelLaunchCustomExp(`
`553`	`553`	`pLocalWorkSize, numEventsInWaitList, phEventWaitList,`
`554`	`554`	`phEvent);`
`555`	`555`	`}`
`556`		`-`
	`556`	`+#if CUDA_VERSION >= 11080`
`557`	`557`	`// Preconditions`
`558`	`558`	`UR_ASSERT(hQueue->getDevice() == hKernel->getProgram()->getDevice(),`
`559`	`559`	`UR_RESULT_ERROR_INVALID_KERNEL);`
`@@ -722,6 +722,11 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueKernelLaunchCustomExp(`
`722`	`722`	`return Err;`
`723`	`723`	`}`
`724`	`724`	`return UR_RESULT_SUCCESS;`
	`725`	`+#else`
	`726`	`+ setErrorMessage("This feature requires cuda 11.8 or later.",`
	`727`	`+ UR_RESULT_ERROR_ADAPTER_SPECIFIC);`
	`728`	`+ return UR_RESULT_ERROR_ADAPTER_SPECIFIC;`
	`729`	`+#endif // CUDA_VERSION >= 11080`
`725`	`730`	`}`
`726`	`731`
`727`	`732`	`/// Set parameters for general 3D memory copy.`