Set CUcontext not via Device in cuda adapter.

Signed-off-by: JackAKirk <jack.kirk@codeplay.com>
oneapi-src · Mar 1, 2024 · 2077bc6 · 2077bc6
1 parent 29ac4c7
commit 2077bc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/adapters/cuda/event.cpp b/source/adapters/cuda/event.cpp
@@ -222,7 +222,7 @@ urEventWait(uint32_t numEvents, const ur_event_handle_t *phEventWaitList) {
   try {
     auto WaitFunc = [](ur_event_handle_t Event) -> ur_result_t {
       UR_ASSERT(Event, UR_RESULT_ERROR_INVALID_EVENT);
-      ScopedContext Active(Event->getQueue()->getDevice());
+      ScopedContext Active(Event->getContext());
 
       return Event->wait();
     };