[llvm-branch-commits] [openmp] fe8b4ac - [OpenMP] Recover non-blocking target nowait disabled by D141232

[Tobias Hieta via llvm-branch-commits]

Author: Ye Luo
Date: 2023-02-16T09:49:09+01:00
New Revision: fe8b4acbb578523a3a3fa5bf324b9ab9274c6f47

URL: https://github.com/llvm/llvm-project/commit/fe8b4acbb578523a3a3fa5bf324b9ab9274c6f47
DIFF: https://github.com/llvm/llvm-project/commit/fe8b4acbb578523a3a3fa5bf324b9ab9274c6f47.diff

LOG: [OpenMP] Recover non-blocking target nowait disabled by D141232

Reviewed By: jdoerfert

Differential Revision: https://reviews.llvm.org/D143871

(cherry picked from commit 0d4e55ba69c1dd04913c59d39c764c8070e55c60)

Added: 
    

Modified: 
    openmp/libomptarget/src/interface.cpp

Removed: 
    


################################################################################
diff  --git a/openmp/libomptarget/src/interface.cpp b/openmp/libomptarget/src/interface.cpp
index 91dda62b0c2d5..beea0c26210bc 100644
--- a/openmp/libomptarget/src/interface.cpp
+++ b/openmp/libomptarget/src/interface.cpp
@@ -299,8 +299,12 @@ EXTERN int __tgt_target_kernel(ident_t *Loc, int64_t DeviceId, int32_t NumTeams,
                                int32_t ThreadLimit, void *HostPtr,
                                KernelArgsTy *KernelArgs) {
   TIMESCOPE_WITH_IDENT(Loc);
-  return targetKernel<AsyncInfoTy>(Loc, DeviceId, NumTeams, ThreadLimit,
-                                   HostPtr, KernelArgs);
+  if (KernelArgs->Flags.NoWait)
+    return targetKernel<TaskAsyncInfoWrapperTy>(
+        Loc, DeviceId, NumTeams, ThreadLimit, HostPtr, KernelArgs);
+  else
+    return targetKernel<AsyncInfoTy>(Loc, DeviceId, NumTeams, ThreadLimit,
+                                     HostPtr, KernelArgs);
 }
 
 /// Implements a target kernel entry that replays a pre-recorded kernel.