[OpenMP][FIX] Ensure to set the context for wait events if necessary

[tianshilei1992]

LG

openmp/libomptarget/plugins/cuda/src/rtl.cpp

int32_t __tgt_rtl_record_event(int32_t device_id, void *event_ptr,
   return recordEvent(event_ptr, async_info_ptr);
 }
 
 int32_t __tgt_rtl_wait_event(int32_t device_id, void *event_ptr,
                              __tgt_async_info *async_info_ptr) {
   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
   assert(async_info_ptr && "async_info_ptr is nullptr");
   assert(event_ptr && "event is nullptr");
-  // NOTE: We might not need to set context for event sync.
+  // If we don't have a queue we need to set the context.
+  if (!async_info_ptr->Queue &&
+      DeviceRTL.setContext(device_id) != OFFLOAD_SUCCESS)
+    return OFFLOAD_FAIL;
   return DeviceRTL.waitEvent(device_id, async_info_ptr, event_ptr);
 }
 
 int32_t __tgt_rtl_sync_event(int32_t device_id, void *event_ptr) {
   assert(event_ptr && "event is nullptr");
   // NOTE: We might not need to set context for event sync.
   return syncEvent(event_ptr);
 }