Revert CUDA change

Signed-off-by: Larsen, Steffen <steffen.larsen@intel.com>
oneapi-src · Feb 29, 2024 · b7b0a6f · b7b0a6f
1 parent 578f347
commit b7b0a6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/adapters/cuda/enqueue.cpp b/source/adapters/cuda/enqueue.cpp
@@ -1764,7 +1764,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueTimestampRecordingExp(
   ur_result_t Result = UR_RESULT_SUCCESS;
   std::unique_ptr<ur_event_handle_t_> RetImplEvent{nullptr};
   try {
-    ScopedContext Active(hQueue->getDevice());
+    ScopedContext Active(hQueue->getContext());
     CUstream CuStream = hQueue->getNextComputeStream();
 
     UR_CHECK_ERROR(enqueueEventsWait(hQueue, CuStream, numEventsInWaitList,