13de4cd12SGeorge Rokos //===----RTLs/cuda/src/rtl.cpp - Target RTLs Implementation ------- C++ -*-===//
23de4cd12SGeorge Rokos //
357b08b09SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
457b08b09SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
557b08b09SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
63de4cd12SGeorge Rokos //
73de4cd12SGeorge Rokos //===----------------------------------------------------------------------===//
83de4cd12SGeorge Rokos //
93de4cd12SGeorge Rokos // RTL for CUDA machine
103de4cd12SGeorge Rokos //
113de4cd12SGeorge Rokos //===----------------------------------------------------------------------===//
123de4cd12SGeorge Rokos 
133de4cd12SGeorge Rokos #include <cassert>
143de4cd12SGeorge Rokos #include <cstddef>
153de4cd12SGeorge Rokos #include <cuda.h>
163de4cd12SGeorge Rokos #include <list>
17a5153dbcSJohannes Doerfert #include <memory>
18feed674dSShilei Tian #include <mutex>
193de4cd12SGeorge Rokos #include <string>
20a15f8589SJoseph Huber #include <unordered_map>
213de4cd12SGeorge Rokos #include <vector>
223de4cd12SGeorge Rokos 
23ae95ceebSJoseph Huber #include "Debug.h"
241546d319SGeorge Rokos #include "omptargetplugin.h"
253de4cd12SGeorge Rokos 
263de4cd12SGeorge Rokos #define TARGET_NAME CUDA
27ae95ceebSJoseph Huber #define DEBUG_PREFIX "Target " GETNAME(TARGET_NAME) " RTL"
2878769ec4SMichael Kruse 
290871d6d5SShilei Tian #include "MemoryManager.h"
300871d6d5SShilei Tian 
3178769ec4SMichael Kruse // Utility for retrieving and printing CUDA error string.
32ae95ceebSJoseph Huber #ifdef OMPTARGET_DEBUG
3378769ec4SMichael Kruse #define CUDA_ERR_STRING(err)                                                   \
3478769ec4SMichael Kruse   do {                                                                         \
35ae95ceebSJoseph Huber     if (getDebugLevel() > 0) {                                                 \
3603111e5eSYe Luo       const char *errStr = nullptr;                                            \
3703111e5eSYe Luo       CUresult errStr_status = cuGetErrorString(err, &errStr);                 \
3803111e5eSYe Luo       if (errStr_status == CUDA_ERROR_INVALID_VALUE)                           \
392ce16810SJoseph Huber         REPORT("Unrecognized CUDA error code: %d\n", err);                     \
4003111e5eSYe Luo       else if (errStr_status == CUDA_SUCCESS)                                  \
412ce16810SJoseph Huber         REPORT("CUDA error is: %s\n", errStr);                                 \
4203111e5eSYe Luo       else {                                                                   \
432ce16810SJoseph Huber         REPORT("Unresolved CUDA error code: %d\n", err);                       \
442ce16810SJoseph Huber         REPORT("Unsuccessful cuGetErrorString return status: %d\n",            \
4503111e5eSYe Luo                errStr_status);                                                 \
4603111e5eSYe Luo       }                                                                        \
472ce16810SJoseph Huber     } else {                                                                   \
482ce16810SJoseph Huber       const char *errStr = nullptr;                                            \
492ce16810SJoseph Huber       CUresult errStr_status = cuGetErrorString(err, &errStr);                 \
502ce16810SJoseph Huber       if (errStr_status == CUDA_SUCCESS)                                       \
512ce16810SJoseph Huber         REPORT("%s \n", errStr);                                               \
5278769ec4SMichael Kruse     }                                                                          \
5378769ec4SMichael Kruse   } while (false)
54b305d26bSSergey Dmitriev #else // OMPTARGET_DEBUG
552ce16810SJoseph Huber #define CUDA_ERR_STRING(err)                                                   \
562ce16810SJoseph Huber   do {                                                                         \
572ce16810SJoseph Huber     const char *errStr = nullptr;                                              \
582ce16810SJoseph Huber     CUresult errStr_status = cuGetErrorString(err, &errStr);                   \
592ce16810SJoseph Huber     if (errStr_status == CUDA_SUCCESS)                                         \
602ce16810SJoseph Huber       REPORT("%s \n", errStr);                                                 \
612ce16810SJoseph Huber   } while (false)
62b305d26bSSergey Dmitriev #endif // OMPTARGET_DEBUG
633de4cd12SGeorge Rokos 
64d2f85d09SJose M Monsalve Diaz #define BOOL2TEXT(b) ((b) ? "Yes" : "No")
65d2f85d09SJose M Monsalve Diaz 
66a81c68aeSShilei Tian #include "elf_common.h"
673de4cd12SGeorge Rokos 
683de4cd12SGeorge Rokos /// Keep entries table per device.
693de4cd12SGeorge Rokos struct FuncOrGblEntryTy {
703de4cd12SGeorge Rokos   __tgt_target_table Table;
713de4cd12SGeorge Rokos   std::vector<__tgt_offload_entry> Entries;
723de4cd12SGeorge Rokos };
733de4cd12SGeorge Rokos 
743de4cd12SGeorge Rokos enum ExecutionModeType {
75c13df8e5SGeorge Rokos   SPMD,         // constructors, destructors,
76c13df8e5SGeorge Rokos                 // combined constructs (`teams distribute parallel for [simd]`)
77c13df8e5SGeorge Rokos   GENERIC,      // everything else
787d576392SJoseph Huber   SPMD_GENERIC, // Generic kernel with SPMD execution
793de4cd12SGeorge Rokos   NONE
803de4cd12SGeorge Rokos };
813de4cd12SGeorge Rokos 
82dc34b1c9SRon Lieberman /// Use a single entity to encode a kernel and a set of flags.
833de4cd12SGeorge Rokos struct KernelTy {
843de4cd12SGeorge Rokos   CUfunction Func;
853de4cd12SGeorge Rokos 
863de4cd12SGeorge Rokos   // execution mode of kernel
873de4cd12SGeorge Rokos   // 0 - SPMD mode (without master warp)
883de4cd12SGeorge Rokos   // 1 - Generic mode (with master warp)
897d576392SJoseph Huber   // 2 - SPMD mode execution with Generic mode semantics.
903de4cd12SGeorge Rokos   int8_t ExecutionMode;
913de4cd12SGeorge Rokos 
92aa27cfc1SJohannes Doerfert   /// Maximal number of threads per block for this kernel.
93aa27cfc1SJohannes Doerfert   int MaxThreadsPerBlock = 0;
94aa27cfc1SJohannes Doerfert 
953de4cd12SGeorge Rokos   KernelTy(CUfunction _Func, int8_t _ExecutionMode)
963de4cd12SGeorge Rokos       : Func(_Func), ExecutionMode(_ExecutionMode) {}
973de4cd12SGeorge Rokos };
983de4cd12SGeorge Rokos 
994c6a098aSKazuaki Ishizaki /// Device environment data
100feed674dSShilei Tian /// Manually sync with the deviceRTL side for now, move to a dedicated header
101feed674dSShilei Tian /// file later.
102e1c7a46dSGuansong Zhang struct omptarget_device_environmentTy {
103e1c7a46dSGuansong Zhang   int32_t debug_level;
104ec02c34bSJoseph Huber   uint32_t num_devices;
105ec02c34bSJoseph Huber   uint32_t device_num;
106*f1c821faSJoseph Huber   uint64_t dynamic_shared_size;
107e1c7a46dSGuansong Zhang };
108e1c7a46dSGuansong Zhang 
109feed674dSShilei Tian namespace {
110feed674dSShilei Tian bool checkResult(CUresult Err, const char *ErrMsg) {
111feed674dSShilei Tian   if (Err == CUDA_SUCCESS)
112feed674dSShilei Tian     return true;
113feed674dSShilei Tian 
1142ce16810SJoseph Huber   REPORT("%s", ErrMsg);
115feed674dSShilei Tian   CUDA_ERR_STRING(Err);
116feed674dSShilei Tian   return false;
117feed674dSShilei Tian }
1184031bb98SShilei Tian 
119a014fbbcSShilei Tian int memcpyDtoD(const void *SrcPtr, void *DstPtr, int64_t Size,
120a014fbbcSShilei Tian                CUstream Stream) {
12195f0d1edSJon Chesterfield   CUresult Err =
12295f0d1edSJon Chesterfield       cuMemcpyDtoDAsync((CUdeviceptr)DstPtr, (CUdeviceptr)SrcPtr, Size, Stream);
123a014fbbcSShilei Tian 
124a014fbbcSShilei Tian   if (Err != CUDA_SUCCESS) {
12529338459SJoseph Huber     DP("Error when copying data from device to device. Pointers: src "
126a014fbbcSShilei Tian        "= " DPxMOD ", dst = " DPxMOD ", size = %" PRId64 "\n",
127a014fbbcSShilei Tian        DPxPTR(SrcPtr), DPxPTR(DstPtr), Size);
128a014fbbcSShilei Tian     CUDA_ERR_STRING(Err);
129a014fbbcSShilei Tian     return OFFLOAD_FAIL;
130a014fbbcSShilei Tian   }
131a014fbbcSShilei Tian 
132a014fbbcSShilei Tian   return OFFLOAD_SUCCESS;
133a014fbbcSShilei Tian }
134a014fbbcSShilei Tian 
13529df4ab3SShilei Tian int createEvent(void **P) {
13629df4ab3SShilei Tian   CUevent Event = nullptr;
13729df4ab3SShilei Tian 
13829df4ab3SShilei Tian   CUresult Err = cuEventCreate(&Event, CU_EVENT_DEFAULT);
13929df4ab3SShilei Tian   if (Err != CUDA_SUCCESS) {
14029df4ab3SShilei Tian     DP("Error when creating event event = " DPxMOD "\n", DPxPTR(Event));
14129df4ab3SShilei Tian     CUDA_ERR_STRING(Err);
14229df4ab3SShilei Tian     return OFFLOAD_FAIL;
14329df4ab3SShilei Tian   }
14429df4ab3SShilei Tian 
14529df4ab3SShilei Tian   *P = Event;
14629df4ab3SShilei Tian 
14729df4ab3SShilei Tian   return OFFLOAD_SUCCESS;
14829df4ab3SShilei Tian }
14929df4ab3SShilei Tian 
15029df4ab3SShilei Tian int recordEvent(void *EventPtr, __tgt_async_info *AsyncInfo) {
15129df4ab3SShilei Tian   CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue);
15229df4ab3SShilei Tian   CUevent Event = reinterpret_cast<CUevent>(EventPtr);
15329df4ab3SShilei Tian 
15429df4ab3SShilei Tian   CUresult Err = cuEventRecord(Event, Stream);
15529df4ab3SShilei Tian   if (Err != CUDA_SUCCESS) {
15629df4ab3SShilei Tian     DP("Error when recording event. stream = " DPxMOD ", event = " DPxMOD "\n",
15729df4ab3SShilei Tian        DPxPTR(Stream), DPxPTR(Event));
15829df4ab3SShilei Tian     CUDA_ERR_STRING(Err);
15929df4ab3SShilei Tian     return OFFLOAD_FAIL;
16029df4ab3SShilei Tian   }
16129df4ab3SShilei Tian 
16229df4ab3SShilei Tian   return OFFLOAD_SUCCESS;
16329df4ab3SShilei Tian }
16429df4ab3SShilei Tian 
16529df4ab3SShilei Tian int syncEvent(void *EventPtr) {
16629df4ab3SShilei Tian   CUevent Event = reinterpret_cast<CUevent>(EventPtr);
16729df4ab3SShilei Tian 
16829df4ab3SShilei Tian   CUresult Err = cuEventSynchronize(Event);
16929df4ab3SShilei Tian   if (Err != CUDA_SUCCESS) {
17029df4ab3SShilei Tian     DP("Error when syncing event = " DPxMOD "\n", DPxPTR(Event));
17129df4ab3SShilei Tian     CUDA_ERR_STRING(Err);
17229df4ab3SShilei Tian     return OFFLOAD_FAIL;
17329df4ab3SShilei Tian   }
17429df4ab3SShilei Tian 
17529df4ab3SShilei Tian   return OFFLOAD_SUCCESS;
17629df4ab3SShilei Tian }
17729df4ab3SShilei Tian 
17829df4ab3SShilei Tian int destroyEvent(void *EventPtr) {
17929df4ab3SShilei Tian   CUevent Event = reinterpret_cast<CUevent>(EventPtr);
18029df4ab3SShilei Tian 
18129df4ab3SShilei Tian   CUresult Err = cuEventDestroy(Event);
18229df4ab3SShilei Tian   if (Err != CUDA_SUCCESS) {
18329df4ab3SShilei Tian     DP("Error when destroying event = " DPxMOD "\n", DPxPTR(Event));
18429df4ab3SShilei Tian     CUDA_ERR_STRING(Err);
18529df4ab3SShilei Tian     return OFFLOAD_FAIL;
18629df4ab3SShilei Tian   }
18729df4ab3SShilei Tian 
18829df4ab3SShilei Tian   return OFFLOAD_SUCCESS;
18929df4ab3SShilei Tian }
19029df4ab3SShilei Tian 
1914031bb98SShilei Tian // Structure contains per-device data
1924031bb98SShilei Tian struct DeviceDataTy {
1935272d29eSJohannes Doerfert   /// List that contains all the kernels.
1945272d29eSJohannes Doerfert   std::list<KernelTy> KernelsList;
1955272d29eSJohannes Doerfert 
1964031bb98SShilei Tian   std::list<FuncOrGblEntryTy> FuncGblEntries;
1975272d29eSJohannes Doerfert 
1984031bb98SShilei Tian   CUcontext Context = nullptr;
1994031bb98SShilei Tian   // Device properties
2004031bb98SShilei Tian   int ThreadsPerBlock = 0;
2014031bb98SShilei Tian   int BlocksPerGrid = 0;
2024031bb98SShilei Tian   int WarpSize = 0;
2034031bb98SShilei Tian   // OpenMP properties
2044031bb98SShilei Tian   int NumTeams = 0;
2054031bb98SShilei Tian   int NumThreads = 0;
2064031bb98SShilei Tian };
207feed674dSShilei Tian 
208feed674dSShilei Tian class StreamManagerTy {
209feed674dSShilei Tian   int NumberOfDevices;
2104031bb98SShilei Tian   // The initial size of stream pool
2114031bb98SShilei Tian   int EnvNumInitialStreams;
212feed674dSShilei Tian   // Per-device stream mutex
213feed674dSShilei Tian   std::vector<std::unique_ptr<std::mutex>> StreamMtx;
214feed674dSShilei Tian   // Per-device stream Id indicates the next available stream in the pool
215feed674dSShilei Tian   std::vector<int> NextStreamId;
216feed674dSShilei Tian   // Per-device stream pool
217feed674dSShilei Tian   std::vector<std::vector<CUstream>> StreamPool;
2184031bb98SShilei Tian   // Reference to per-device data
2194031bb98SShilei Tian   std::vector<DeviceDataTy> &DeviceData;
220feed674dSShilei Tian 
221feed674dSShilei Tian   // If there is no CUstream left in the pool, we will resize the pool to
222feed674dSShilei Tian   // allocate more CUstream. This function should be called with device mutex,
223feed674dSShilei Tian   // and we do not resize to smaller one.
224feed674dSShilei Tian   void resizeStreamPool(const int DeviceId, const size_t NewSize) {
225feed674dSShilei Tian     std::vector<CUstream> &Pool = StreamPool[DeviceId];
226feed674dSShilei Tian     const size_t CurrentSize = Pool.size();
227feed674dSShilei Tian     assert(NewSize > CurrentSize && "new size is not larger than current size");
228feed674dSShilei Tian 
2294031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
2304031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) {
231feed674dSShilei Tian       // We will return if cannot switch to the right context in case of
232feed674dSShilei Tian       // creating bunch of streams that are not corresponding to the right
233feed674dSShilei Tian       // device. The offloading will fail later because selected CUstream is
234feed674dSShilei Tian       // nullptr.
235feed674dSShilei Tian       return;
236feed674dSShilei Tian     }
237feed674dSShilei Tian 
2384031bb98SShilei Tian     Pool.resize(NewSize, nullptr);
2394031bb98SShilei Tian 
240feed674dSShilei Tian     for (size_t I = CurrentSize; I < NewSize; ++I) {
2414031bb98SShilei Tian       checkResult(cuStreamCreate(&Pool[I], CU_STREAM_NON_BLOCKING),
2424031bb98SShilei Tian                   "Error returned from cuStreamCreate\n");
243feed674dSShilei Tian     }
244feed674dSShilei Tian   }
245feed674dSShilei Tian 
246feed674dSShilei Tian public:
2474031bb98SShilei Tian   StreamManagerTy(const int NumberOfDevices,
2484031bb98SShilei Tian                   std::vector<DeviceDataTy> &DeviceData)
2494031bb98SShilei Tian       : NumberOfDevices(NumberOfDevices), EnvNumInitialStreams(32),
2504031bb98SShilei Tian         DeviceData(DeviceData) {
251feed674dSShilei Tian     StreamPool.resize(NumberOfDevices);
252feed674dSShilei Tian     NextStreamId.resize(NumberOfDevices);
253feed674dSShilei Tian     StreamMtx.resize(NumberOfDevices);
254feed674dSShilei Tian 
2554031bb98SShilei Tian     if (const char *EnvStr = getenv("LIBOMPTARGET_NUM_INITIAL_STREAMS"))
2564031bb98SShilei Tian       EnvNumInitialStreams = std::stoi(EnvStr);
257feed674dSShilei Tian 
258feed674dSShilei Tian     // Initialize the next stream id
259feed674dSShilei Tian     std::fill(NextStreamId.begin(), NextStreamId.end(), 0);
260feed674dSShilei Tian 
261feed674dSShilei Tian     // Initialize stream mutex
262feed674dSShilei Tian     for (std::unique_ptr<std::mutex> &Ptr : StreamMtx)
263feed674dSShilei Tian       Ptr = std::make_unique<std::mutex>();
264feed674dSShilei Tian   }
265feed674dSShilei Tian 
266feed674dSShilei Tian   ~StreamManagerTy() {
267feed674dSShilei Tian     // Destroy streams
268feed674dSShilei Tian     for (int I = 0; I < NumberOfDevices; ++I) {
2694031bb98SShilei Tian       checkResult(cuCtxSetCurrent(DeviceData[I].Context),
2704031bb98SShilei Tian                   "Error returned from cuCtxSetCurrent\n");
271feed674dSShilei Tian 
272feed674dSShilei Tian       for (CUstream &S : StreamPool[I]) {
2734031bb98SShilei Tian         if (S)
27495f0d1edSJon Chesterfield           checkResult(cuStreamDestroy(S),
27595f0d1edSJon Chesterfield                       "Error returned from cuStreamDestroy\n");
276feed674dSShilei Tian       }
277feed674dSShilei Tian     }
278feed674dSShilei Tian   }
279feed674dSShilei Tian 
280feed674dSShilei Tian   // Get a CUstream from pool. Per-device next stream id always points to the
281feed674dSShilei Tian   // next available CUstream. That means, CUstreams [0, id-1] have been
282feed674dSShilei Tian   // assigned, and [id,] are still available. If there is no CUstream left, we
283feed674dSShilei Tian   // will ask more CUstreams from CUDA RT. Each time a CUstream is assigned,
284feed674dSShilei Tian   // the id will increase one.
285feed674dSShilei Tian   // xxxxxs+++++++++
286feed674dSShilei Tian   //      ^
287feed674dSShilei Tian   //      id
288feed674dSShilei Tian   // After assignment, the pool becomes the following and s is assigned.
289feed674dSShilei Tian   // xxxxxs+++++++++
290feed674dSShilei Tian   //       ^
291feed674dSShilei Tian   //       id
292feed674dSShilei Tian   CUstream getStream(const int DeviceId) {
293feed674dSShilei Tian     const std::lock_guard<std::mutex> Lock(*StreamMtx[DeviceId]);
294feed674dSShilei Tian     int &Id = NextStreamId[DeviceId];
295feed674dSShilei Tian     // No CUstream left in the pool, we need to request from CUDA RT
296ad94fce8SJohannes Doerfert     if (Id == static_cast<int>(StreamPool[DeviceId].size())) {
297feed674dSShilei Tian       // By default we double the stream pool every time
298feed674dSShilei Tian       resizeStreamPool(DeviceId, Id * 2);
299feed674dSShilei Tian     }
300feed674dSShilei Tian     return StreamPool[DeviceId][Id++];
301feed674dSShilei Tian   }
302feed674dSShilei Tian 
303feed674dSShilei Tian   // Return a CUstream back to pool. As mentioned above, per-device next
304feed674dSShilei Tian   // stream is always points to the next available CUstream, so when we return
305feed674dSShilei Tian   // a CUstream, we need to first decrease the id, and then copy the CUstream
306feed674dSShilei Tian   // back.
307feed674dSShilei Tian   // It is worth noting that, the order of streams return might be different
308feed674dSShilei Tian   // from that they're assigned, that saying, at some point, there might be
309feed674dSShilei Tian   // two identical CUstreams.
310feed674dSShilei Tian   // xxax+a+++++
311feed674dSShilei Tian   //     ^
312feed674dSShilei Tian   //     id
313feed674dSShilei Tian   // However, it doesn't matter, because they're always on the two sides of
314feed674dSShilei Tian   // id. The left one will in the end be overwritten by another CUstream.
315feed674dSShilei Tian   // Therefore, after several execution, the order of pool might be different
316feed674dSShilei Tian   // from its initial state.
317feed674dSShilei Tian   void returnStream(const int DeviceId, CUstream Stream) {
318feed674dSShilei Tian     const std::lock_guard<std::mutex> Lock(*StreamMtx[DeviceId]);
319feed674dSShilei Tian     int &Id = NextStreamId[DeviceId];
320feed674dSShilei Tian     assert(Id > 0 && "Wrong stream ID");
321feed674dSShilei Tian     StreamPool[DeviceId][--Id] = Stream;
322feed674dSShilei Tian   }
323feed674dSShilei Tian 
3244031bb98SShilei Tian   bool initializeDeviceStreamPool(const int DeviceId) {
3254031bb98SShilei Tian     assert(StreamPool[DeviceId].empty() && "stream pool has been initialized");
3264031bb98SShilei Tian 
3274031bb98SShilei Tian     resizeStreamPool(DeviceId, EnvNumInitialStreams);
3284031bb98SShilei Tian 
3294031bb98SShilei Tian     // Check the size of stream pool
330ad94fce8SJohannes Doerfert     if (static_cast<int>(StreamPool[DeviceId].size()) != EnvNumInitialStreams)
3314031bb98SShilei Tian       return false;
3324031bb98SShilei Tian 
3334031bb98SShilei Tian     // Check whether each stream is valid
3344031bb98SShilei Tian     for (CUstream &S : StreamPool[DeviceId])
3354031bb98SShilei Tian       if (!S)
3364031bb98SShilei Tian         return false;
3374031bb98SShilei Tian 
3384031bb98SShilei Tian     return true;
339feed674dSShilei Tian   }
340feed674dSShilei Tian };
341feed674dSShilei Tian 
3424031bb98SShilei Tian class DeviceRTLTy {
3433de4cd12SGeorge Rokos   int NumberOfDevices;
3444031bb98SShilei Tian   // OpenMP environment properties
3453de4cd12SGeorge Rokos   int EnvNumTeams;
3463de4cd12SGeorge Rokos   int EnvTeamLimit;
347422adaa8SJoseph Huber   int EnvTeamThreadLimit;
3484031bb98SShilei Tian   // OpenMP requires flags
349c5fe030cSGheorghe-Teodor Bercea   int64_t RequiresFlags;
350*f1c821faSJoseph Huber   // Amount of dynamic shared memory to use at launch.
351*f1c821faSJoseph Huber   uint64_t DynamicMemorySize;
352c5fe030cSGheorghe-Teodor Bercea 
3534031bb98SShilei Tian   static constexpr const int HardTeamLimit = 1U << 16U; // 64k
3544031bb98SShilei Tian   static constexpr const int HardThreadLimit = 1024;
3554031bb98SShilei Tian   static constexpr const int DefaultNumTeams = 128;
3564031bb98SShilei Tian   static constexpr const int DefaultNumThreads = 128;
3573de4cd12SGeorge Rokos 
3584031bb98SShilei Tian   std::unique_ptr<StreamManagerTy> StreamManager;
3594031bb98SShilei Tian   std::vector<DeviceDataTy> DeviceData;
3604031bb98SShilei Tian   std::vector<CUmodule> Modules;
361feed674dSShilei Tian 
3620871d6d5SShilei Tian   /// A class responsible for interacting with device native runtime library to
3630871d6d5SShilei Tian   /// allocate and free memory.
3640871d6d5SShilei Tian   class CUDADeviceAllocatorTy : public DeviceAllocatorTy {
3650871d6d5SShilei Tian     const int DeviceId;
3660871d6d5SShilei Tian     const std::vector<DeviceDataTy> &DeviceData;
367a15f8589SJoseph Huber     std::unordered_map<void *, TargetAllocTy> HostPinnedAllocs;
3680871d6d5SShilei Tian 
3690871d6d5SShilei Tian   public:
3700871d6d5SShilei Tian     CUDADeviceAllocatorTy(int DeviceId, std::vector<DeviceDataTy> &DeviceData)
3710871d6d5SShilei Tian         : DeviceId(DeviceId), DeviceData(DeviceData) {}
3720871d6d5SShilei Tian 
373a15f8589SJoseph Huber     void *allocate(size_t Size, void *, TargetAllocTy Kind) override {
3740871d6d5SShilei Tian       if (Size == 0)
3750871d6d5SShilei Tian         return nullptr;
3760871d6d5SShilei Tian 
3770871d6d5SShilei Tian       CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
3780871d6d5SShilei Tian       if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
3790871d6d5SShilei Tian         return nullptr;
3800871d6d5SShilei Tian 
381a15f8589SJoseph Huber       void *MemAlloc = nullptr;
382a15f8589SJoseph Huber       switch (Kind) {
383a15f8589SJoseph Huber       case TARGET_ALLOC_DEFAULT:
384a15f8589SJoseph Huber       case TARGET_ALLOC_DEVICE:
3850871d6d5SShilei Tian         CUdeviceptr DevicePtr;
38695f0d1edSJon Chesterfield         Err = cuMemAlloc(&DevicePtr, Size);
387a15f8589SJoseph Huber         MemAlloc = (void *)DevicePtr;
38895f0d1edSJon Chesterfield         if (!checkResult(Err, "Error returned from cuMemAlloc\n"))
3890871d6d5SShilei Tian           return nullptr;
390a15f8589SJoseph Huber         break;
391a15f8589SJoseph Huber       case TARGET_ALLOC_HOST:
392a15f8589SJoseph Huber         void *HostPtr;
393a15f8589SJoseph Huber         Err = cuMemAllocHost(&HostPtr, Size);
394a15f8589SJoseph Huber         MemAlloc = HostPtr;
395a15f8589SJoseph Huber         if (!checkResult(Err, "Error returned from cuMemAllocHost\n"))
396a15f8589SJoseph Huber           return nullptr;
397a15f8589SJoseph Huber         HostPinnedAllocs[MemAlloc] = Kind;
398a15f8589SJoseph Huber         break;
399a15f8589SJoseph Huber       case TARGET_ALLOC_SHARED:
400a15f8589SJoseph Huber         CUdeviceptr SharedPtr;
401a15f8589SJoseph Huber         Err = cuMemAllocManaged(&SharedPtr, Size, CU_MEM_ATTACH_GLOBAL);
402a15f8589SJoseph Huber         MemAlloc = (void *)SharedPtr;
403a15f8589SJoseph Huber         if (!checkResult(Err, "Error returned from cuMemAllocManaged\n"))
404a15f8589SJoseph Huber           return nullptr;
405a15f8589SJoseph Huber         break;
406a15f8589SJoseph Huber       }
4070871d6d5SShilei Tian 
408a15f8589SJoseph Huber       return MemAlloc;
4090871d6d5SShilei Tian     }
4100871d6d5SShilei Tian 
4110871d6d5SShilei Tian     int free(void *TgtPtr) override {
4120871d6d5SShilei Tian       CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
4130871d6d5SShilei Tian       if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
4140871d6d5SShilei Tian         return OFFLOAD_FAIL;
4150871d6d5SShilei Tian 
416a15f8589SJoseph Huber       // Host pinned memory must be freed differently.
417a15f8589SJoseph Huber       TargetAllocTy Kind =
418a15f8589SJoseph Huber           (HostPinnedAllocs.find(TgtPtr) == HostPinnedAllocs.end())
419a15f8589SJoseph Huber               ? TARGET_ALLOC_DEFAULT
420a15f8589SJoseph Huber               : TARGET_ALLOC_HOST;
421a15f8589SJoseph Huber       switch (Kind) {
422a15f8589SJoseph Huber       case TARGET_ALLOC_DEFAULT:
423a15f8589SJoseph Huber       case TARGET_ALLOC_DEVICE:
424a15f8589SJoseph Huber       case TARGET_ALLOC_SHARED:
42595f0d1edSJon Chesterfield         Err = cuMemFree((CUdeviceptr)TgtPtr);
42695f0d1edSJon Chesterfield         if (!checkResult(Err, "Error returned from cuMemFree\n"))
4270871d6d5SShilei Tian           return OFFLOAD_FAIL;
428a15f8589SJoseph Huber         break;
429a15f8589SJoseph Huber       case TARGET_ALLOC_HOST:
430a15f8589SJoseph Huber         Err = cuMemFreeHost(TgtPtr);
431a15f8589SJoseph Huber         if (!checkResult(Err, "Error returned from cuMemFreeHost\n"))
432a15f8589SJoseph Huber           return OFFLOAD_FAIL;
433a15f8589SJoseph Huber         break;
434a15f8589SJoseph Huber       }
4350871d6d5SShilei Tian 
4360871d6d5SShilei Tian       return OFFLOAD_SUCCESS;
4370871d6d5SShilei Tian     }
4380871d6d5SShilei Tian   };
4390871d6d5SShilei Tian 
4400871d6d5SShilei Tian   /// A vector of device allocators
4410871d6d5SShilei Tian   std::vector<CUDADeviceAllocatorTy> DeviceAllocators;
4420871d6d5SShilei Tian 
4430871d6d5SShilei Tian   /// A vector of memory managers. Since the memory manager is non-copyable and
4440871d6d5SShilei Tian   // non-removable, we wrap them into std::unique_ptr.
4450871d6d5SShilei Tian   std::vector<std::unique_ptr<MemoryManagerTy>> MemoryManagers;
4460871d6d5SShilei Tian 
4470871d6d5SShilei Tian   /// Whether use memory manager
4480871d6d5SShilei Tian   bool UseMemoryManager = true;
4490871d6d5SShilei Tian 
4503de4cd12SGeorge Rokos   // Record entry point associated with device
4514031bb98SShilei Tian   void addOffloadEntry(const int DeviceId, const __tgt_offload_entry entry) {
4524031bb98SShilei Tian     FuncOrGblEntryTy &E = DeviceData[DeviceId].FuncGblEntries.back();
4533de4cd12SGeorge Rokos     E.Entries.push_back(entry);
4543de4cd12SGeorge Rokos   }
4553de4cd12SGeorge Rokos 
456ae209397SJoseph Huber   // Return a pointer to the entry associated with the pointer
457ae209397SJoseph Huber   const __tgt_offload_entry *getOffloadEntry(const int DeviceId,
458ae209397SJoseph Huber                                              const void *Addr) const {
4594031bb98SShilei Tian     for (const __tgt_offload_entry &Itr :
4604031bb98SShilei Tian          DeviceData[DeviceId].FuncGblEntries.back().Entries)
4614031bb98SShilei Tian       if (Itr.addr == Addr)
462ae209397SJoseph Huber         return &Itr;
4633de4cd12SGeorge Rokos 
464ae209397SJoseph Huber     return nullptr;
4653de4cd12SGeorge Rokos   }
4663de4cd12SGeorge Rokos 
4673de4cd12SGeorge Rokos   // Return the pointer to the target entries table
4684031bb98SShilei Tian   __tgt_target_table *getOffloadEntriesTable(const int DeviceId) {
4694031bb98SShilei Tian     FuncOrGblEntryTy &E = DeviceData[DeviceId].FuncGblEntries.back();
4703de4cd12SGeorge Rokos 
4714031bb98SShilei Tian     if (E.Entries.empty())
4724031bb98SShilei Tian       return nullptr;
4733de4cd12SGeorge Rokos 
4743de4cd12SGeorge Rokos     // Update table info according to the entries and return the pointer
4754031bb98SShilei Tian     E.Table.EntriesBegin = E.Entries.data();
4764031bb98SShilei Tian     E.Table.EntriesEnd = E.Entries.data() + E.Entries.size();
4773de4cd12SGeorge Rokos 
4783de4cd12SGeorge Rokos     return &E.Table;
4793de4cd12SGeorge Rokos   }
4803de4cd12SGeorge Rokos 
4813de4cd12SGeorge Rokos   // Clear entries table for a device
4824031bb98SShilei Tian   void clearOffloadEntriesTable(const int DeviceId) {
4834031bb98SShilei Tian     DeviceData[DeviceId].FuncGblEntries.emplace_back();
4844031bb98SShilei Tian     FuncOrGblEntryTy &E = DeviceData[DeviceId].FuncGblEntries.back();
4853de4cd12SGeorge Rokos     E.Entries.clear();
4864031bb98SShilei Tian     E.Table.EntriesBegin = E.Table.EntriesEnd = nullptr;
4873de4cd12SGeorge Rokos   }
4883de4cd12SGeorge Rokos 
4895449fbb5SJohannes Doerfert   CUstream getStream(const int DeviceId, __tgt_async_info *AsyncInfo) const {
4905449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
4914031bb98SShilei Tian 
4925449fbb5SJohannes Doerfert     if (!AsyncInfo->Queue)
4935449fbb5SJohannes Doerfert       AsyncInfo->Queue = StreamManager->getStream(DeviceId);
4944031bb98SShilei Tian 
4955449fbb5SJohannes Doerfert     return reinterpret_cast<CUstream>(AsyncInfo->Queue);
496b305d26bSSergey Dmitriev   }
4974031bb98SShilei Tian 
4984031bb98SShilei Tian public:
4994031bb98SShilei Tian   // This class should not be copied
5004031bb98SShilei Tian   DeviceRTLTy(const DeviceRTLTy &) = delete;
5014031bb98SShilei Tian   DeviceRTLTy(DeviceRTLTy &&) = delete;
5024031bb98SShilei Tian 
5034031bb98SShilei Tian   DeviceRTLTy()
5044031bb98SShilei Tian       : NumberOfDevices(0), EnvNumTeams(-1), EnvTeamLimit(-1),
505*f1c821faSJoseph Huber         EnvTeamThreadLimit(-1), RequiresFlags(OMP_REQ_UNDEFINED),
506*f1c821faSJoseph Huber         DynamicMemorySize(0) {
507b305d26bSSergey Dmitriev 
5083de4cd12SGeorge Rokos     DP("Start initializing CUDA\n");
5093de4cd12SGeorge Rokos 
5104031bb98SShilei Tian     CUresult Err = cuInit(0);
5117baff00eSJon Chesterfield     if (Err == CUDA_ERROR_INVALID_HANDLE) {
5127baff00eSJon Chesterfield       // Can't call cuGetErrorString if dlsym failed
5137baff00eSJon Chesterfield       DP("Failed to load CUDA shared library\n");
5147baff00eSJon Chesterfield       return;
5157baff00eSJon Chesterfield     }
5164031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuInit\n")) {
5173de4cd12SGeorge Rokos       return;
5183de4cd12SGeorge Rokos     }
5193de4cd12SGeorge Rokos 
5204031bb98SShilei Tian     Err = cuDeviceGetCount(&NumberOfDevices);
5214031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuDeviceGetCount\n"))
5223de4cd12SGeorge Rokos       return;
5233de4cd12SGeorge Rokos 
5243de4cd12SGeorge Rokos     if (NumberOfDevices == 0) {
5253de4cd12SGeorge Rokos       DP("There are no devices supporting CUDA.\n");
5263de4cd12SGeorge Rokos       return;
5273de4cd12SGeorge Rokos     }
5283de4cd12SGeorge Rokos 
5294031bb98SShilei Tian     DeviceData.resize(NumberOfDevices);
5303de4cd12SGeorge Rokos 
5313de4cd12SGeorge Rokos     // Get environment variables regarding teams
5324031bb98SShilei Tian     if (const char *EnvStr = getenv("OMP_TEAM_LIMIT")) {
5333de4cd12SGeorge Rokos       // OMP_TEAM_LIMIT has been set
5344031bb98SShilei Tian       EnvTeamLimit = std::stoi(EnvStr);
5353de4cd12SGeorge Rokos       DP("Parsed OMP_TEAM_LIMIT=%d\n", EnvTeamLimit);
5363de4cd12SGeorge Rokos     }
537422adaa8SJoseph Huber     if (const char *EnvStr = getenv("OMP_TEAMS_THREAD_LIMIT")) {
538422adaa8SJoseph Huber       // OMP_TEAMS_THREAD_LIMIT has been set
539422adaa8SJoseph Huber       EnvTeamThreadLimit = std::stoi(EnvStr);
540422adaa8SJoseph Huber       DP("Parsed OMP_TEAMS_THREAD_LIMIT=%d\n", EnvTeamThreadLimit);
541422adaa8SJoseph Huber     }
5424031bb98SShilei Tian     if (const char *EnvStr = getenv("OMP_NUM_TEAMS")) {
5433de4cd12SGeorge Rokos       // OMP_NUM_TEAMS has been set
5444031bb98SShilei Tian       EnvNumTeams = std::stoi(EnvStr);
5453de4cd12SGeorge Rokos       DP("Parsed OMP_NUM_TEAMS=%d\n", EnvNumTeams);
5463de4cd12SGeorge Rokos     }
547*f1c821faSJoseph Huber     if (const char *EnvStr = getenv("LIBOMPTARGET_SHARED_MEMORY_SIZE")) {
548*f1c821faSJoseph Huber       // LIBOMPTARGET_SHARED_MEMORY_SIZE has been set
549*f1c821faSJoseph Huber       DynamicMemorySize = std::stoi(EnvStr);
550*f1c821faSJoseph Huber       DP("Parsed LIBOMPTARGET_SHARED_MEMORY_SIZE", DynamicMemorySize);
551*f1c821faSJoseph Huber     }
552c5fe030cSGheorghe-Teodor Bercea 
553feed674dSShilei Tian     StreamManager =
5544031bb98SShilei Tian         std::make_unique<StreamManagerTy>(NumberOfDevices, DeviceData);
5550871d6d5SShilei Tian 
5560871d6d5SShilei Tian     for (int I = 0; I < NumberOfDevices; ++I)
5570871d6d5SShilei Tian       DeviceAllocators.emplace_back(I, DeviceData);
5580871d6d5SShilei Tian 
5590871d6d5SShilei Tian     // Get the size threshold from environment variable
5600871d6d5SShilei Tian     std::pair<size_t, bool> Res = MemoryManagerTy::getSizeThresholdFromEnv();
5610871d6d5SShilei Tian     UseMemoryManager = Res.second;
5620871d6d5SShilei Tian     size_t MemoryManagerThreshold = Res.first;
5630871d6d5SShilei Tian 
5640871d6d5SShilei Tian     if (UseMemoryManager)
5650871d6d5SShilei Tian       for (int I = 0; I < NumberOfDevices; ++I)
5660871d6d5SShilei Tian         MemoryManagers.emplace_back(std::make_unique<MemoryManagerTy>(
5670871d6d5SShilei Tian             DeviceAllocators[I], MemoryManagerThreshold));
5683de4cd12SGeorge Rokos   }
5693de4cd12SGeorge Rokos 
5704031bb98SShilei Tian   ~DeviceRTLTy() {
5710871d6d5SShilei Tian     // We first destruct memory managers in case that its dependent data are
5720871d6d5SShilei Tian     // destroyed before it.
5730871d6d5SShilei Tian     for (auto &M : MemoryManagers)
5740871d6d5SShilei Tian       M.release();
5750871d6d5SShilei Tian 
576feed674dSShilei Tian     StreamManager = nullptr;
577feed674dSShilei Tian 
5784031bb98SShilei Tian     for (CUmodule &M : Modules)
5794031bb98SShilei Tian       // Close module
5804031bb98SShilei Tian       if (M)
5814031bb98SShilei Tian         checkResult(cuModuleUnload(M), "Error returned from cuModuleUnload\n");
5824031bb98SShilei Tian 
5834031bb98SShilei Tian     for (DeviceDataTy &D : DeviceData) {
5844031bb98SShilei Tian       // Destroy context
585c5348aecSYe Luo       if (D.Context) {
586c5348aecSYe Luo         checkResult(cuCtxSetCurrent(D.Context),
587c5348aecSYe Luo                     "Error returned from cuCtxSetCurrent\n");
588c5348aecSYe Luo         CUdevice Device;
589c5348aecSYe Luo         checkResult(cuCtxGetDevice(&Device),
590c5348aecSYe Luo                     "Error returned from cuCtxGetDevice\n");
59195f0d1edSJon Chesterfield         checkResult(cuDevicePrimaryCtxRelease(Device),
59295f0d1edSJon Chesterfield                     "Error returned from cuDevicePrimaryCtxRelease\n");
593c5348aecSYe Luo       }
5943de4cd12SGeorge Rokos     }
5953de4cd12SGeorge Rokos   }
5963de4cd12SGeorge Rokos 
5974031bb98SShilei Tian   // Check whether a given DeviceId is valid
5984031bb98SShilei Tian   bool isValidDeviceId(const int DeviceId) const {
5994031bb98SShilei Tian     return DeviceId >= 0 && DeviceId < NumberOfDevices;
60032ed2927SShilei Tian   }
60132ed2927SShilei Tian 
602cb038927SShilei Tian   int getNumOfDevices() const { return NumberOfDevices; }
6034031bb98SShilei Tian 
6044031bb98SShilei Tian   void setRequiresFlag(const int64_t Flags) { this->RequiresFlags = Flags; }
6054031bb98SShilei Tian 
6064031bb98SShilei Tian   int initDevice(const int DeviceId) {
6074031bb98SShilei Tian     CUdevice Device;
6084031bb98SShilei Tian 
6094031bb98SShilei Tian     DP("Getting device %d\n", DeviceId);
6104031bb98SShilei Tian     CUresult Err = cuDeviceGet(&Device, DeviceId);
6114031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuDeviceGet\n"))
61232ed2927SShilei Tian       return OFFLOAD_FAIL;
6133de4cd12SGeorge Rokos 
614c5348aecSYe Luo     // Query the current flags of the primary context and set its flags if
615c5348aecSYe Luo     // it is inactive
616c5348aecSYe Luo     unsigned int FormerPrimaryCtxFlags = 0;
617c5348aecSYe Luo     int FormerPrimaryCtxIsActive = 0;
618c5348aecSYe Luo     Err = cuDevicePrimaryCtxGetState(Device, &FormerPrimaryCtxFlags,
619c5348aecSYe Luo                                      &FormerPrimaryCtxIsActive);
620c5348aecSYe Luo     if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxGetState\n"))
621c5348aecSYe Luo       return OFFLOAD_FAIL;
622c5348aecSYe Luo 
623c5348aecSYe Luo     if (FormerPrimaryCtxIsActive) {
624c5348aecSYe Luo       DP("The primary context is active, no change to its flags\n");
625c5348aecSYe Luo       if ((FormerPrimaryCtxFlags & CU_CTX_SCHED_MASK) !=
626c5348aecSYe Luo           CU_CTX_SCHED_BLOCKING_SYNC)
627c5348aecSYe Luo         DP("Warning the current flags are not CU_CTX_SCHED_BLOCKING_SYNC\n");
628c5348aecSYe Luo     } else {
629c5348aecSYe Luo       DP("The primary context is inactive, set its flags to "
630c5348aecSYe Luo          "CU_CTX_SCHED_BLOCKING_SYNC\n");
63195f0d1edSJon Chesterfield       Err = cuDevicePrimaryCtxSetFlags(Device, CU_CTX_SCHED_BLOCKING_SYNC);
63295f0d1edSJon Chesterfield       if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxSetFlags\n"))
633c5348aecSYe Luo         return OFFLOAD_FAIL;
634c5348aecSYe Luo     }
635c5348aecSYe Luo 
636c5348aecSYe Luo     // Retain the per device primary context and save it to use whenever this
637c5348aecSYe Luo     // device is selected.
638c5348aecSYe Luo     Err = cuDevicePrimaryCtxRetain(&DeviceData[DeviceId].Context, Device);
639c5348aecSYe Luo     if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxRetain\n"))
6403de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
6413de4cd12SGeorge Rokos 
6424031bb98SShilei Tian     Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
6434031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
6444031bb98SShilei Tian       return OFFLOAD_FAIL;
645a5153dbcSJohannes Doerfert 
646feed674dSShilei Tian     // Initialize stream pool
6474031bb98SShilei Tian     if (!StreamManager->initializeDeviceStreamPool(DeviceId))
6484031bb98SShilei Tian       return OFFLOAD_FAIL;
649a5153dbcSJohannes Doerfert 
650bb51d398SJonas Hahnfeld     // Query attributes to determine number of threads/block and blocks/grid.
6514031bb98SShilei Tian     int MaxGridDimX;
6524031bb98SShilei Tian     Err = cuDeviceGetAttribute(&MaxGridDimX, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X,
6534031bb98SShilei Tian                                Device);
6544031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
6554031bb98SShilei Tian       DP("Error getting max grid dimension, use default value %d\n",
6564031bb98SShilei Tian          DeviceRTLTy::DefaultNumTeams);
6574031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = DeviceRTLTy::DefaultNumTeams;
6584031bb98SShilei Tian     } else if (MaxGridDimX <= DeviceRTLTy::HardTeamLimit) {
6594031bb98SShilei Tian       DP("Using %d CUDA blocks per grid\n", MaxGridDimX);
6604031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = MaxGridDimX;
6613de4cd12SGeorge Rokos     } else {
6623de4cd12SGeorge Rokos       DP("Max CUDA blocks per grid %d exceeds the hard team limit %d, capping "
663bb51d398SJonas Hahnfeld          "at the hard limit\n",
6644031bb98SShilei Tian          MaxGridDimX, DeviceRTLTy::HardTeamLimit);
6654031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = DeviceRTLTy::HardTeamLimit;
6663de4cd12SGeorge Rokos     }
6673de4cd12SGeorge Rokos 
668bb51d398SJonas Hahnfeld     // We are only exploiting threads along the x axis.
6694031bb98SShilei Tian     int MaxBlockDimX;
6704031bb98SShilei Tian     Err = cuDeviceGetAttribute(&MaxBlockDimX,
6714031bb98SShilei Tian                                CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X, Device);
6724031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
6734031bb98SShilei Tian       DP("Error getting max block dimension, use default value %d\n",
6744031bb98SShilei Tian          DeviceRTLTy::DefaultNumThreads);
6754031bb98SShilei Tian       DeviceData[DeviceId].ThreadsPerBlock = DeviceRTLTy::DefaultNumThreads;
676422adaa8SJoseph Huber     } else {
6774031bb98SShilei Tian       DP("Using %d CUDA threads per block\n", MaxBlockDimX);
6784031bb98SShilei Tian       DeviceData[DeviceId].ThreadsPerBlock = MaxBlockDimX;
679422adaa8SJoseph Huber 
680422adaa8SJoseph Huber       if (EnvTeamThreadLimit > 0 &&
681422adaa8SJoseph Huber           DeviceData[DeviceId].ThreadsPerBlock > EnvTeamThreadLimit) {
682422adaa8SJoseph Huber         DP("Max CUDA threads per block %d exceeds the thread limit %d set by "
683422adaa8SJoseph Huber            "OMP_TEAMS_THREAD_LIMIT, capping at the limit\n",
684422adaa8SJoseph Huber            DeviceData[DeviceId].ThreadsPerBlock, EnvTeamThreadLimit);
685422adaa8SJoseph Huber         DeviceData[DeviceId].ThreadsPerBlock = EnvTeamThreadLimit;
686422adaa8SJoseph Huber       }
687422adaa8SJoseph Huber       if (DeviceData[DeviceId].ThreadsPerBlock > DeviceRTLTy::HardThreadLimit) {
6884031bb98SShilei Tian         DP("Max CUDA threads per block %d exceeds the hard thread limit %d, "
6894031bb98SShilei Tian            "capping at the hard limit\n",
690422adaa8SJoseph Huber            DeviceData[DeviceId].ThreadsPerBlock, DeviceRTLTy::HardThreadLimit);
6914031bb98SShilei Tian         DeviceData[DeviceId].ThreadsPerBlock = DeviceRTLTy::HardThreadLimit;
6923de4cd12SGeorge Rokos       }
693422adaa8SJoseph Huber     }
6943de4cd12SGeorge Rokos 
6954031bb98SShilei Tian     // Get and set warp size
6964031bb98SShilei Tian     int WarpSize;
6974031bb98SShilei Tian     Err =
6984031bb98SShilei Tian         cuDeviceGetAttribute(&WarpSize, CU_DEVICE_ATTRIBUTE_WARP_SIZE, Device);
6994031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
7004031bb98SShilei Tian       DP("Error getting warp size, assume default value 32\n");
7014031bb98SShilei Tian       DeviceData[DeviceId].WarpSize = 32;
702bb51d398SJonas Hahnfeld     } else {
7034031bb98SShilei Tian       DP("Using warp size %d\n", WarpSize);
7044031bb98SShilei Tian       DeviceData[DeviceId].WarpSize = WarpSize;
7053de4cd12SGeorge Rokos     }
7063de4cd12SGeorge Rokos 
7073de4cd12SGeorge Rokos     // Adjust teams to the env variables
7084031bb98SShilei Tian     if (EnvTeamLimit > 0 && DeviceData[DeviceId].BlocksPerGrid > EnvTeamLimit) {
7093de4cd12SGeorge Rokos       DP("Capping max CUDA blocks per grid to OMP_TEAM_LIMIT=%d\n",
7104031bb98SShilei Tian          EnvTeamLimit);
7114031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = EnvTeamLimit;
7123de4cd12SGeorge Rokos     }
7133de4cd12SGeorge Rokos 
7143817ba13SJoseph Huber     size_t StackLimit;
7153817ba13SJoseph Huber     size_t HeapLimit;
7163817ba13SJoseph Huber     if (const char *EnvStr = getenv("LIBOMPTARGET_STACK_SIZE")) {
7173817ba13SJoseph Huber       StackLimit = std::stol(EnvStr);
7183817ba13SJoseph Huber       if (cuCtxSetLimit(CU_LIMIT_STACK_SIZE, StackLimit) != CUDA_SUCCESS)
7193817ba13SJoseph Huber         return OFFLOAD_FAIL;
7203817ba13SJoseph Huber     } else {
7213817ba13SJoseph Huber       if (cuCtxGetLimit(&StackLimit, CU_LIMIT_STACK_SIZE) != CUDA_SUCCESS)
7223817ba13SJoseph Huber         return OFFLOAD_FAIL;
7233817ba13SJoseph Huber     }
7243817ba13SJoseph Huber     if (const char *EnvStr = getenv("LIBOMPTARGET_HEAP_SIZE")) {
7253817ba13SJoseph Huber       HeapLimit = std::stol(EnvStr);
7263817ba13SJoseph Huber       if (cuCtxSetLimit(CU_LIMIT_MALLOC_HEAP_SIZE, HeapLimit) != CUDA_SUCCESS)
7273817ba13SJoseph Huber         return OFFLOAD_FAIL;
7283817ba13SJoseph Huber     } else {
7293817ba13SJoseph Huber       if (cuCtxGetLimit(&HeapLimit, CU_LIMIT_MALLOC_HEAP_SIZE) != CUDA_SUCCESS)
7303817ba13SJoseph Huber         return OFFLOAD_FAIL;
7313817ba13SJoseph Huber     }
7323817ba13SJoseph Huber 
733119a9ea1SJoseph Huber     INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId,
734ae209397SJoseph Huber          "Device supports up to %d CUDA blocks and %d threads with a "
735ae209397SJoseph Huber          "warp size of %d\n",
736ae209397SJoseph Huber          DeviceData[DeviceId].BlocksPerGrid,
737ae209397SJoseph Huber          DeviceData[DeviceId].ThreadsPerBlock, DeviceData[DeviceId].WarpSize);
7383817ba13SJoseph Huber     INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId,
7393817ba13SJoseph Huber          "Device heap size is %d Bytes, device stack size is %d Bytes per "
7403817ba13SJoseph Huber          "thread\n",
7413817ba13SJoseph Huber          (int)HeapLimit, (int)StackLimit);
7423de4cd12SGeorge Rokos 
7433de4cd12SGeorge Rokos     // Set default number of teams
7444031bb98SShilei Tian     if (EnvNumTeams > 0) {
7453de4cd12SGeorge Rokos       DP("Default number of teams set according to environment %d\n",
7464031bb98SShilei Tian          EnvNumTeams);
7474031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = EnvNumTeams;
7483de4cd12SGeorge Rokos     } else {
7494031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = DeviceRTLTy::DefaultNumTeams;
7503de4cd12SGeorge Rokos       DP("Default number of teams set according to library's default %d\n",
7514031bb98SShilei Tian          DeviceRTLTy::DefaultNumTeams);
7523de4cd12SGeorge Rokos     }
7534031bb98SShilei Tian 
7544031bb98SShilei Tian     if (DeviceData[DeviceId].NumTeams > DeviceData[DeviceId].BlocksPerGrid) {
7553de4cd12SGeorge Rokos       DP("Default number of teams exceeds device limit, capping at %d\n",
7564031bb98SShilei Tian          DeviceData[DeviceId].BlocksPerGrid);
7574031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].BlocksPerGrid;
7583de4cd12SGeorge Rokos     }
7593de4cd12SGeorge Rokos 
7603de4cd12SGeorge Rokos     // Set default number of threads
7614031bb98SShilei Tian     DeviceData[DeviceId].NumThreads = DeviceRTLTy::DefaultNumThreads;
7623de4cd12SGeorge Rokos     DP("Default number of threads set according to library's default %d\n",
7634031bb98SShilei Tian        DeviceRTLTy::DefaultNumThreads);
7644031bb98SShilei Tian     if (DeviceData[DeviceId].NumThreads >
7654031bb98SShilei Tian         DeviceData[DeviceId].ThreadsPerBlock) {
7663de4cd12SGeorge Rokos       DP("Default number of threads exceeds device limit, capping at %d\n",
7674031bb98SShilei Tian          DeviceData[DeviceId].ThreadsPerBlock);
7684031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].ThreadsPerBlock;
7693de4cd12SGeorge Rokos     }
7703de4cd12SGeorge Rokos 
7713de4cd12SGeorge Rokos     return OFFLOAD_SUCCESS;
7723de4cd12SGeorge Rokos   }
7733de4cd12SGeorge Rokos 
7744031bb98SShilei Tian   __tgt_target_table *loadBinary(const int DeviceId,
7754031bb98SShilei Tian                                  const __tgt_device_image *Image) {
7764031bb98SShilei Tian     // Set the context we are using
7774031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
7784031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
7794031bb98SShilei Tian       return nullptr;
7803de4cd12SGeorge Rokos 
7813de4cd12SGeorge Rokos     // Clear the offload table as we are going to create a new one.
7824031bb98SShilei Tian     clearOffloadEntriesTable(DeviceId);
7833de4cd12SGeorge Rokos 
7843de4cd12SGeorge Rokos     // Create the module and extract the function pointers.
7854031bb98SShilei Tian     CUmodule Module;
7864031bb98SShilei Tian     DP("Load data from image " DPxMOD "\n", DPxPTR(Image->ImageStart));
7874031bb98SShilei Tian     Err = cuModuleLoadDataEx(&Module, Image->ImageStart, 0, nullptr, nullptr);
7884031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuModuleLoadDataEx\n"))
7894031bb98SShilei Tian       return nullptr;
7903de4cd12SGeorge Rokos 
7913de4cd12SGeorge Rokos     DP("CUDA module successfully loaded!\n");
7924031bb98SShilei Tian 
7934031bb98SShilei Tian     Modules.push_back(Module);
7943de4cd12SGeorge Rokos 
7953de4cd12SGeorge Rokos     // Find the symbols in the module by name.
7964031bb98SShilei Tian     const __tgt_offload_entry *HostBegin = Image->EntriesBegin;
7974031bb98SShilei Tian     const __tgt_offload_entry *HostEnd = Image->EntriesEnd;
7983de4cd12SGeorge Rokos 
7995272d29eSJohannes Doerfert     std::list<KernelTy> &KernelsList = DeviceData[DeviceId].KernelsList;
8004031bb98SShilei Tian     for (const __tgt_offload_entry *E = HostBegin; E != HostEnd; ++E) {
8014031bb98SShilei Tian       if (!E->addr) {
8024031bb98SShilei Tian         // We return nullptr when something like this happens, the host should
8034031bb98SShilei Tian         // have always something in the address to uniquely identify the target
8044031bb98SShilei Tian         // region.
8054031bb98SShilei Tian         DP("Invalid binary: host entry '<null>' (size = %zd)...\n", E->size);
8064031bb98SShilei Tian         return nullptr;
8073de4cd12SGeorge Rokos       }
8083de4cd12SGeorge Rokos 
8094031bb98SShilei Tian       if (E->size) {
8104031bb98SShilei Tian         __tgt_offload_entry Entry = *E;
8114031bb98SShilei Tian         CUdeviceptr CUPtr;
8124031bb98SShilei Tian         size_t CUSize;
81395f0d1edSJon Chesterfield         Err = cuModuleGetGlobal(&CUPtr, &CUSize, Module, E->name);
8144031bb98SShilei Tian         // We keep this style here because we need the name
8154031bb98SShilei Tian         if (Err != CUDA_SUCCESS) {
8162ce16810SJoseph Huber           REPORT("Loading global '%s' Failed\n", E->name);
8174031bb98SShilei Tian           CUDA_ERR_STRING(Err);
8184031bb98SShilei Tian           return nullptr;
8193de4cd12SGeorge Rokos         }
8203de4cd12SGeorge Rokos 
8214031bb98SShilei Tian         if (CUSize != E->size) {
8224031bb98SShilei Tian           DP("Loading global '%s' - size mismatch (%zd != %zd)\n", E->name,
8234031bb98SShilei Tian              CUSize, E->size);
8244031bb98SShilei Tian           return nullptr;
8253de4cd12SGeorge Rokos         }
8263de4cd12SGeorge Rokos 
8273de4cd12SGeorge Rokos         DP("Entry point " DPxMOD " maps to global %s (" DPxMOD ")\n",
8284031bb98SShilei Tian            DPxPTR(E - HostBegin), E->name, DPxPTR(CUPtr));
8294031bb98SShilei Tian 
8304031bb98SShilei Tian         Entry.addr = (void *)(CUPtr);
8313de4cd12SGeorge Rokos 
832aace6d28SGheorghe-Teodor Bercea         // Note: In the current implementation declare target variables
833aace6d28SGheorghe-Teodor Bercea         // can either be link or to. This means that once unified
834aace6d28SGheorghe-Teodor Bercea         // memory is activated via the requires directive, the variable
835aace6d28SGheorghe-Teodor Bercea         // can be used directly from the host in both cases.
836aace6d28SGheorghe-Teodor Bercea         // TODO: when variables types other than to or link are added,
8374c6a098aSKazuaki Ishizaki         // the below condition should be changed to explicitly
838aace6d28SGheorghe-Teodor Bercea         // check for to and link variables types:
8394031bb98SShilei Tian         // (RequiresFlags & OMP_REQ_UNIFIED_SHARED_MEMORY && (e->flags &
8404031bb98SShilei Tian         // OMP_DECLARE_TARGET_LINK || e->flags == OMP_DECLARE_TARGET_TO))
8414031bb98SShilei Tian         if (RequiresFlags & OMP_REQ_UNIFIED_SHARED_MEMORY) {
842aace6d28SGheorghe-Teodor Bercea           // If unified memory is present any target link or to variables
843c5fe030cSGheorghe-Teodor Bercea           // can access host addresses directly. There is no longer a
844c5fe030cSGheorghe-Teodor Bercea           // need for device copies.
84595f0d1edSJon Chesterfield           cuMemcpyHtoD(CUPtr, E->addr, sizeof(void *));
8464031bb98SShilei Tian           DP("Copy linked variable host address (" DPxMOD
8474031bb98SShilei Tian              ") to device address (" DPxMOD ")\n",
8484031bb98SShilei Tian              DPxPTR(*((void **)E->addr)), DPxPTR(CUPtr));
849c5fe030cSGheorghe-Teodor Bercea         }
850c5fe030cSGheorghe-Teodor Bercea 
8514031bb98SShilei Tian         addOffloadEntry(DeviceId, Entry);
8523de4cd12SGeorge Rokos 
8533de4cd12SGeorge Rokos         continue;
8543de4cd12SGeorge Rokos       }
8553de4cd12SGeorge Rokos 
8564031bb98SShilei Tian       CUfunction Func;
8574031bb98SShilei Tian       Err = cuModuleGetFunction(&Func, Module, E->name);
8584031bb98SShilei Tian       // We keep this style here because we need the name
8594031bb98SShilei Tian       if (Err != CUDA_SUCCESS) {
8602ce16810SJoseph Huber         REPORT("Loading '%s' Failed\n", E->name);
8614031bb98SShilei Tian         CUDA_ERR_STRING(Err);
8624031bb98SShilei Tian         return nullptr;
8633de4cd12SGeorge Rokos       }
8643de4cd12SGeorge Rokos 
8653de4cd12SGeorge Rokos       DP("Entry point " DPxMOD " maps to %s (" DPxMOD ")\n",
8664031bb98SShilei Tian          DPxPTR(E - HostBegin), E->name, DPxPTR(Func));
8673de4cd12SGeorge Rokos 
8683de4cd12SGeorge Rokos       // default value GENERIC (in case symbol is missing from cubin file)
8693de4cd12SGeorge Rokos       int8_t ExecModeVal = ExecutionModeType::GENERIC;
8704031bb98SShilei Tian       std::string ExecModeNameStr(E->name);
8713de4cd12SGeorge Rokos       ExecModeNameStr += "_exec_mode";
8723de4cd12SGeorge Rokos       const char *ExecModeName = ExecModeNameStr.c_str();
8733de4cd12SGeorge Rokos 
8743de4cd12SGeorge Rokos       CUdeviceptr ExecModePtr;
8754031bb98SShilei Tian       size_t CUSize;
87695f0d1edSJon Chesterfield       Err = cuModuleGetGlobal(&ExecModePtr, &CUSize, Module, ExecModeName);
8774031bb98SShilei Tian       if (Err == CUDA_SUCCESS) {
8784031bb98SShilei Tian         if (CUSize != sizeof(int8_t)) {
8793de4cd12SGeorge Rokos           DP("Loading global exec_mode '%s' - size mismatch (%zd != %zd)\n",
8804031bb98SShilei Tian              ExecModeName, CUSize, sizeof(int8_t));
8814031bb98SShilei Tian           return nullptr;
8823de4cd12SGeorge Rokos         }
8833de4cd12SGeorge Rokos 
88495f0d1edSJon Chesterfield         Err = cuMemcpyDtoH(&ExecModeVal, ExecModePtr, CUSize);
8854031bb98SShilei Tian         if (Err != CUDA_SUCCESS) {
8862ce16810SJoseph Huber           REPORT("Error when copying data from device to host. Pointers: "
8873de4cd12SGeorge Rokos                  "host = " DPxMOD ", device = " DPxMOD ", size = %zd\n",
8884031bb98SShilei Tian                  DPxPTR(&ExecModeVal), DPxPTR(ExecModePtr), CUSize);
8894031bb98SShilei Tian           CUDA_ERR_STRING(Err);
8904031bb98SShilei Tian           return nullptr;
8913de4cd12SGeorge Rokos         }
8923de4cd12SGeorge Rokos 
8937d576392SJoseph Huber         if (ExecModeVal < 0 || ExecModeVal > 2) {
8943de4cd12SGeorge Rokos           DP("Error wrong exec_mode value specified in cubin file: %d\n",
8953de4cd12SGeorge Rokos              ExecModeVal);
8964031bb98SShilei Tian           return nullptr;
8973de4cd12SGeorge Rokos         }
8983de4cd12SGeorge Rokos       } else {
899edb8acdcSJoseph Huber         DP("Loading global exec_mode '%s' - symbol missing, using default "
9004031bb98SShilei Tian            "value GENERIC (1)\n",
9014031bb98SShilei Tian            ExecModeName);
9023de4cd12SGeorge Rokos       }
9033de4cd12SGeorge Rokos 
9044031bb98SShilei Tian       KernelsList.emplace_back(Func, ExecModeVal);
9053de4cd12SGeorge Rokos 
9064031bb98SShilei Tian       __tgt_offload_entry Entry = *E;
9074031bb98SShilei Tian       Entry.addr = &KernelsList.back();
9084031bb98SShilei Tian       addOffloadEntry(DeviceId, Entry);
9093de4cd12SGeorge Rokos     }
9103de4cd12SGeorge Rokos 
911e1c7a46dSGuansong Zhang     // send device environment data to the device
912e1c7a46dSGuansong Zhang     {
913ec02c34bSJoseph Huber       // TODO: The device ID used here is not the real device ID used by OpenMP.
914ec02c34bSJoseph Huber       omptarget_device_environmentTy DeviceEnv{
915ec02c34bSJoseph Huber           0, static_cast<uint32_t>(NumberOfDevices),
916*f1c821faSJoseph Huber           static_cast<uint32_t>(DeviceId), DynamicMemorySize};
917e1c7a46dSGuansong Zhang 
918e1c7a46dSGuansong Zhang #ifdef OMPTARGET_DEBUG
9194031bb98SShilei Tian       if (const char *EnvStr = getenv("LIBOMPTARGET_DEVICE_RTL_DEBUG"))
9204031bb98SShilei Tian         DeviceEnv.debug_level = std::stoi(EnvStr);
921e1c7a46dSGuansong Zhang #endif
922e1c7a46dSGuansong Zhang 
9234031bb98SShilei Tian       const char *DeviceEnvName = "omptarget_device_environment";
9244031bb98SShilei Tian       CUdeviceptr DeviceEnvPtr;
9254031bb98SShilei Tian       size_t CUSize;
926e1c7a46dSGuansong Zhang 
92795f0d1edSJon Chesterfield       Err = cuModuleGetGlobal(&DeviceEnvPtr, &CUSize, Module, DeviceEnvName);
9284031bb98SShilei Tian       if (Err == CUDA_SUCCESS) {
9294031bb98SShilei Tian         if (CUSize != sizeof(DeviceEnv)) {
9302ce16810SJoseph Huber           REPORT(
9312ce16810SJoseph Huber               "Global device_environment '%s' - size mismatch (%zu != %zu)\n",
9324031bb98SShilei Tian               DeviceEnvName, CUSize, sizeof(int32_t));
9334031bb98SShilei Tian           CUDA_ERR_STRING(Err);
9344031bb98SShilei Tian           return nullptr;
935e1c7a46dSGuansong Zhang         }
936e1c7a46dSGuansong Zhang 
93795f0d1edSJon Chesterfield         Err = cuMemcpyHtoD(DeviceEnvPtr, &DeviceEnv, CUSize);
9384031bb98SShilei Tian         if (Err != CUDA_SUCCESS) {
9392ce16810SJoseph Huber           REPORT("Error when copying data from host to device. Pointers: "
940e1c7a46dSGuansong Zhang                  "host = " DPxMOD ", device = " DPxMOD ", size = %zu\n",
9414031bb98SShilei Tian                  DPxPTR(&DeviceEnv), DPxPTR(DeviceEnvPtr), CUSize);
9424031bb98SShilei Tian           CUDA_ERR_STRING(Err);
9434031bb98SShilei Tian           return nullptr;
944e1c7a46dSGuansong Zhang         }
945e1c7a46dSGuansong Zhang 
9464031bb98SShilei Tian         DP("Sending global device environment data %zu bytes\n", CUSize);
947e1c7a46dSGuansong Zhang       } else {
9484031bb98SShilei Tian         DP("Finding global device environment '%s' - symbol missing.\n",
9494031bb98SShilei Tian            DeviceEnvName);
9504031bb98SShilei Tian         DP("Continue, considering this is a device RTL which does not accept "
9514031bb98SShilei Tian            "environment setting.\n");
952e1c7a46dSGuansong Zhang       }
953e1c7a46dSGuansong Zhang     }
954e1c7a46dSGuansong Zhang 
9554031bb98SShilei Tian     return getOffloadEntriesTable(DeviceId);
9563de4cd12SGeorge Rokos   }
9573de4cd12SGeorge Rokos 
958a15f8589SJoseph Huber   void *dataAlloc(const int DeviceId, const int64_t Size,
959a15f8589SJoseph Huber                   const TargetAllocTy Kind) {
960a15f8589SJoseph Huber     switch (Kind) {
961a15f8589SJoseph Huber     case TARGET_ALLOC_DEFAULT:
962a15f8589SJoseph Huber     case TARGET_ALLOC_DEVICE:
9630871d6d5SShilei Tian       if (UseMemoryManager)
9640871d6d5SShilei Tian         return MemoryManagers[DeviceId]->allocate(Size, nullptr);
965a15f8589SJoseph Huber       else
966a15f8589SJoseph Huber         return DeviceAllocators[DeviceId].allocate(Size, nullptr, Kind);
967a15f8589SJoseph Huber     case TARGET_ALLOC_HOST:
968a15f8589SJoseph Huber     case TARGET_ALLOC_SHARED:
969a15f8589SJoseph Huber       return DeviceAllocators[DeviceId].allocate(Size, nullptr, Kind);
970a15f8589SJoseph Huber     }
9714031bb98SShilei Tian 
972a15f8589SJoseph Huber     REPORT("Invalid target data allocation kind or requested allocator not "
973a15f8589SJoseph Huber            "implemented yet\n");
974a15f8589SJoseph Huber 
975a15f8589SJoseph Huber     return nullptr;
9763de4cd12SGeorge Rokos   }
9773de4cd12SGeorge Rokos 
9784031bb98SShilei Tian   int dataSubmit(const int DeviceId, const void *TgtPtr, const void *HstPtr,
9795449fbb5SJohannes Doerfert                  const int64_t Size, __tgt_async_info *AsyncInfo) const {
9805449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
9813de4cd12SGeorge Rokos 
9824031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
9834031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
9843de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
9853de4cd12SGeorge Rokos 
9865449fbb5SJohannes Doerfert     CUstream Stream = getStream(DeviceId, AsyncInfo);
9873de4cd12SGeorge Rokos 
98895f0d1edSJon Chesterfield     Err = cuMemcpyHtoDAsync((CUdeviceptr)TgtPtr, HstPtr, Size, Stream);
9894031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
99029338459SJoseph Huber       DP("Error when copying data from host to device. Pointers: host "
9912ce16810SJoseph Huber          "= " DPxMOD ", device = " DPxMOD ", size = %" PRId64 "\n",
9924031bb98SShilei Tian          DPxPTR(HstPtr), DPxPTR(TgtPtr), Size);
9934031bb98SShilei Tian       CUDA_ERR_STRING(Err);
9943de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
9953de4cd12SGeorge Rokos     }
9963de4cd12SGeorge Rokos 
9973de4cd12SGeorge Rokos     return OFFLOAD_SUCCESS;
9983de4cd12SGeorge Rokos   }
9993de4cd12SGeorge Rokos 
10004031bb98SShilei Tian   int dataRetrieve(const int DeviceId, void *HstPtr, const void *TgtPtr,
10015449fbb5SJohannes Doerfert                    const int64_t Size, __tgt_async_info *AsyncInfo) const {
10025449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
10034031bb98SShilei Tian 
10044031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
10054031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
100603ff643dSShilei Tian       return OFFLOAD_FAIL;
100703ff643dSShilei Tian 
10085449fbb5SJohannes Doerfert     CUstream Stream = getStream(DeviceId, AsyncInfo);
100903ff643dSShilei Tian 
101095f0d1edSJon Chesterfield     Err = cuMemcpyDtoHAsync(HstPtr, (CUdeviceptr)TgtPtr, Size, Stream);
10114031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
101229338459SJoseph Huber       DP("Error when copying data from device to host. Pointers: host "
10132ce16810SJoseph Huber          "= " DPxMOD ", device = " DPxMOD ", size = %" PRId64 "\n",
10144031bb98SShilei Tian          DPxPTR(HstPtr), DPxPTR(TgtPtr), Size);
10154031bb98SShilei Tian       CUDA_ERR_STRING(Err);
10163de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
10173de4cd12SGeorge Rokos     }
10183de4cd12SGeorge Rokos 
10194031bb98SShilei Tian     return OFFLOAD_SUCCESS;
10204031bb98SShilei Tian   }
10214031bb98SShilei Tian 
1022a014fbbcSShilei Tian   int dataExchange(int SrcDevId, const void *SrcPtr, int DstDevId, void *DstPtr,
10235449fbb5SJohannes Doerfert                    int64_t Size, __tgt_async_info *AsyncInfo) const {
10245449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
1025a014fbbcSShilei Tian 
1026a014fbbcSShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[SrcDevId].Context);
1027a014fbbcSShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
1028a014fbbcSShilei Tian       return OFFLOAD_FAIL;
1029a014fbbcSShilei Tian 
10305449fbb5SJohannes Doerfert     CUstream Stream = getStream(SrcDevId, AsyncInfo);
1031a014fbbcSShilei Tian 
1032a014fbbcSShilei Tian     // If they are two devices, we try peer to peer copy first
1033a014fbbcSShilei Tian     if (SrcDevId != DstDevId) {
1034a014fbbcSShilei Tian       int CanAccessPeer = 0;
1035a014fbbcSShilei Tian       Err = cuDeviceCanAccessPeer(&CanAccessPeer, SrcDevId, DstDevId);
1036a014fbbcSShilei Tian       if (Err != CUDA_SUCCESS) {
10372ce16810SJoseph Huber         REPORT("Error returned from cuDeviceCanAccessPeer. src = %" PRId32
1038a014fbbcSShilei Tian                ", dst = %" PRId32 "\n",
1039a014fbbcSShilei Tian                SrcDevId, DstDevId);
1040a014fbbcSShilei Tian         CUDA_ERR_STRING(Err);
1041a014fbbcSShilei Tian         return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1042a014fbbcSShilei Tian       }
1043a014fbbcSShilei Tian 
1044a014fbbcSShilei Tian       if (!CanAccessPeer) {
1045a014fbbcSShilei Tian         DP("P2P memcpy not supported so fall back to D2D memcpy");
1046a014fbbcSShilei Tian         return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1047a014fbbcSShilei Tian       }
1048a014fbbcSShilei Tian 
1049a014fbbcSShilei Tian       Err = cuCtxEnablePeerAccess(DeviceData[DstDevId].Context, 0);
1050a014fbbcSShilei Tian       if (Err != CUDA_SUCCESS) {
10512ce16810SJoseph Huber         REPORT("Error returned from cuCtxEnablePeerAccess. src = %" PRId32
1052a014fbbcSShilei Tian                ", dst = %" PRId32 "\n",
1053a014fbbcSShilei Tian                SrcDevId, DstDevId);
1054a014fbbcSShilei Tian         CUDA_ERR_STRING(Err);
1055a014fbbcSShilei Tian         return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1056a014fbbcSShilei Tian       }
1057a014fbbcSShilei Tian 
1058a014fbbcSShilei Tian       Err = cuMemcpyPeerAsync((CUdeviceptr)DstPtr, DeviceData[DstDevId].Context,
1059a014fbbcSShilei Tian                               (CUdeviceptr)SrcPtr, DeviceData[SrcDevId].Context,
1060a014fbbcSShilei Tian                               Size, Stream);
1061a014fbbcSShilei Tian       if (Err == CUDA_SUCCESS)
1062a014fbbcSShilei Tian         return OFFLOAD_SUCCESS;
1063a014fbbcSShilei Tian 
106429338459SJoseph Huber       DP("Error returned from cuMemcpyPeerAsync. src_ptr = " DPxMOD
106529338459SJoseph Huber          ", src_id =%" PRId32 ", dst_ptr = " DPxMOD ", dst_id =%" PRId32 "\n",
106645bb073dSYe Luo          DPxPTR(SrcPtr), SrcDevId, DPxPTR(DstPtr), DstDevId);
1067a014fbbcSShilei Tian       CUDA_ERR_STRING(Err);
1068a014fbbcSShilei Tian     }
1069a014fbbcSShilei Tian 
1070a014fbbcSShilei Tian     return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1071a014fbbcSShilei Tian   }
1072a014fbbcSShilei Tian 
10730871d6d5SShilei Tian   int dataDelete(const int DeviceId, void *TgtPtr) {
10740871d6d5SShilei Tian     if (UseMemoryManager)
10750871d6d5SShilei Tian       return MemoryManagers[DeviceId]->free(TgtPtr);
10764031bb98SShilei Tian 
10770871d6d5SShilei Tian     return DeviceAllocators[DeviceId].free(TgtPtr);
10784031bb98SShilei Tian   }
10794031bb98SShilei Tian 
1080aa27cfc1SJohannes Doerfert   int runTargetTeamRegion(const int DeviceId, void *TgtEntryPtr, void **TgtArgs,
1081aa27cfc1SJohannes Doerfert                           ptrdiff_t *TgtOffsets, const int ArgNum,
1082aa27cfc1SJohannes Doerfert                           const int TeamNum, const int ThreadLimit,
10834031bb98SShilei Tian                           const unsigned int LoopTripCount,
10844031bb98SShilei Tian                           __tgt_async_info *AsyncInfo) const {
10854031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
10864031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
10874031bb98SShilei Tian       return OFFLOAD_FAIL;
10884031bb98SShilei Tian 
10893de4cd12SGeorge Rokos     // All args are references.
10904031bb98SShilei Tian     std::vector<void *> Args(ArgNum);
10914031bb98SShilei Tian     std::vector<void *> Ptrs(ArgNum);
10923de4cd12SGeorge Rokos 
10934031bb98SShilei Tian     for (int I = 0; I < ArgNum; ++I) {
10944031bb98SShilei Tian       Ptrs[I] = (void *)((intptr_t)TgtArgs[I] + TgtOffsets[I]);
10954031bb98SShilei Tian       Args[I] = &Ptrs[I];
10961546d319SGeorge Rokos     }
10973de4cd12SGeorge Rokos 
1098aa27cfc1SJohannes Doerfert     KernelTy *KernelInfo = reinterpret_cast<KernelTy *>(TgtEntryPtr);
10993de4cd12SGeorge Rokos 
1100aa27cfc1SJohannes Doerfert     int CudaThreadsPerBlock;
11014031bb98SShilei Tian     if (ThreadLimit > 0) {
11024031bb98SShilei Tian       DP("Setting CUDA threads per block to requested %d\n", ThreadLimit);
11034031bb98SShilei Tian       CudaThreadsPerBlock = ThreadLimit;
11043de4cd12SGeorge Rokos       // Add master warp if necessary
11053de4cd12SGeorge Rokos       if (KernelInfo->ExecutionMode == GENERIC) {
11064031bb98SShilei Tian         DP("Adding master warp: +%d threads\n", DeviceData[DeviceId].WarpSize);
11074031bb98SShilei Tian         CudaThreadsPerBlock += DeviceData[DeviceId].WarpSize;
11083de4cd12SGeorge Rokos       }
1109c13df8e5SGeorge Rokos     } else {
1110c13df8e5SGeorge Rokos       DP("Setting CUDA threads per block to default %d\n",
11114031bb98SShilei Tian          DeviceData[DeviceId].NumThreads);
11124031bb98SShilei Tian       CudaThreadsPerBlock = DeviceData[DeviceId].NumThreads;
1113c13df8e5SGeorge Rokos     }
11143de4cd12SGeorge Rokos 
11154031bb98SShilei Tian     if (CudaThreadsPerBlock > DeviceData[DeviceId].ThreadsPerBlock) {
11163de4cd12SGeorge Rokos       DP("Threads per block capped at device limit %d\n",
11174031bb98SShilei Tian          DeviceData[DeviceId].ThreadsPerBlock);
11184031bb98SShilei Tian       CudaThreadsPerBlock = DeviceData[DeviceId].ThreadsPerBlock;
11193de4cd12SGeorge Rokos     }
11203de4cd12SGeorge Rokos 
1121aa27cfc1SJohannes Doerfert     if (!KernelInfo->MaxThreadsPerBlock) {
1122aa27cfc1SJohannes Doerfert       Err = cuFuncGetAttribute(&KernelInfo->MaxThreadsPerBlock,
11234031bb98SShilei Tian                                CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK,
11244031bb98SShilei Tian                                KernelInfo->Func);
1125aa27cfc1SJohannes Doerfert       if (!checkResult(Err, "Error returned from cuFuncGetAttribute\n"))
1126aa27cfc1SJohannes Doerfert         return OFFLOAD_FAIL;
1127aa27cfc1SJohannes Doerfert     }
1128aa27cfc1SJohannes Doerfert 
1129aa27cfc1SJohannes Doerfert     if (KernelInfo->MaxThreadsPerBlock < CudaThreadsPerBlock) {
1130aa27cfc1SJohannes Doerfert       DP("Threads per block capped at kernel limit %d\n",
1131aa27cfc1SJohannes Doerfert          KernelInfo->MaxThreadsPerBlock);
1132aa27cfc1SJohannes Doerfert       CudaThreadsPerBlock = KernelInfo->MaxThreadsPerBlock;
11333de4cd12SGeorge Rokos     }
11343de4cd12SGeorge Rokos 
11354031bb98SShilei Tian     unsigned int CudaBlocksPerGrid;
11364031bb98SShilei Tian     if (TeamNum <= 0) {
11374031bb98SShilei Tian       if (LoopTripCount > 0 && EnvNumTeams < 0) {
1138c13df8e5SGeorge Rokos         if (KernelInfo->ExecutionMode == SPMD) {
11394031bb98SShilei Tian           // We have a combined construct, i.e. `target teams distribute
11404031bb98SShilei Tian           // parallel for [simd]`. We launch so many teams so that each thread
11414031bb98SShilei Tian           // will execute one iteration of the loop. round up to the nearest
11424031bb98SShilei Tian           // integer
11434031bb98SShilei Tian           CudaBlocksPerGrid = ((LoopTripCount - 1) / CudaThreadsPerBlock) + 1;
11447d576392SJoseph Huber         } else if (KernelInfo->ExecutionMode == GENERIC) {
1145c13df8e5SGeorge Rokos           // If we reach this point, then we have a non-combined construct, i.e.
1146c13df8e5SGeorge Rokos           // `teams distribute` with a nested `parallel for` and each team is
1147c13df8e5SGeorge Rokos           // assigned one iteration of the `distribute` loop. E.g.:
1148c13df8e5SGeorge Rokos           //
1149c13df8e5SGeorge Rokos           // #pragma omp target teams distribute
1150c13df8e5SGeorge Rokos           // for(...loop_tripcount...) {
1151c13df8e5SGeorge Rokos           //   #pragma omp parallel for
1152c13df8e5SGeorge Rokos           //   for(...) {}
1153c13df8e5SGeorge Rokos           // }
1154c13df8e5SGeorge Rokos           //
1155c13df8e5SGeorge Rokos           // Threads within a team will execute the iterations of the `parallel`
1156c13df8e5SGeorge Rokos           // loop.
11574031bb98SShilei Tian           CudaBlocksPerGrid = LoopTripCount;
11587d576392SJoseph Huber         } else if (KernelInfo->ExecutionMode == SPMD_GENERIC) {
11597d576392SJoseph Huber           // If we reach this point, then we are executing a kernel that was
11607d576392SJoseph Huber           // transformed from Generic-mode to SPMD-mode. This kernel has
11617d576392SJoseph Huber           // SPMD-mode execution, but needs its blocks to be scheduled
11627d576392SJoseph Huber           // differently because the current loop trip count only applies to the
11637d576392SJoseph Huber           // `teams distribute` region and will create var too few blocks using
11647d576392SJoseph Huber           // the regular SPMD-mode method.
11657d576392SJoseph Huber           CudaBlocksPerGrid = LoopTripCount;
1166a158d366SJoseph Huber         } else {
1167a158d366SJoseph Huber           REPORT("Unknown execution mode: %d\n", KernelInfo->ExecutionMode);
1168a158d366SJoseph Huber           return OFFLOAD_FAIL;
1169c13df8e5SGeorge Rokos         }
117045bb073dSYe Luo         DP("Using %d teams due to loop trip count %" PRIu32
11714031bb98SShilei Tian            " and number of threads per block %d\n",
11724031bb98SShilei Tian            CudaBlocksPerGrid, LoopTripCount, CudaThreadsPerBlock);
11733de4cd12SGeorge Rokos       } else {
11744031bb98SShilei Tian         DP("Using default number of teams %d\n", DeviceData[DeviceId].NumTeams);
11754031bb98SShilei Tian         CudaBlocksPerGrid = DeviceData[DeviceId].NumTeams;
11763de4cd12SGeorge Rokos       }
11774031bb98SShilei Tian     } else if (TeamNum > DeviceData[DeviceId].BlocksPerGrid) {
11783de4cd12SGeorge Rokos       DP("Capping number of teams to team limit %d\n",
11794031bb98SShilei Tian          DeviceData[DeviceId].BlocksPerGrid);
11804031bb98SShilei Tian       CudaBlocksPerGrid = DeviceData[DeviceId].BlocksPerGrid;
11813de4cd12SGeorge Rokos     } else {
11824031bb98SShilei Tian       DP("Using requested number of teams %d\n", TeamNum);
11834031bb98SShilei Tian       CudaBlocksPerGrid = TeamNum;
11843de4cd12SGeorge Rokos     }
11853de4cd12SGeorge Rokos 
1186119a9ea1SJoseph Huber     INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId,
1187ae209397SJoseph Huber          "Launching kernel %s with %d blocks and %d threads in %s "
1188ae209397SJoseph Huber          "mode\n",
1189ae209397SJoseph Huber          (getOffloadEntry(DeviceId, TgtEntryPtr))
1190ae209397SJoseph Huber              ? getOffloadEntry(DeviceId, TgtEntryPtr)->name
1191ae209397SJoseph Huber              : "(null)",
1192ae209397SJoseph Huber          CudaBlocksPerGrid, CudaThreadsPerBlock,
11937d576392SJoseph Huber          (KernelInfo->ExecutionMode != SPMD
119476c0c0caSJoseph Huber               ? (KernelInfo->ExecutionMode == GENERIC ? "Generic"
119576c0c0caSJoseph Huber                                                       : "SPMD-Generic")
11967d576392SJoseph Huber               : "SPMD"));
11973de4cd12SGeorge Rokos 
11984031bb98SShilei Tian     CUstream Stream = getStream(DeviceId, AsyncInfo);
11994031bb98SShilei Tian     Err = cuLaunchKernel(KernelInfo->Func, CudaBlocksPerGrid, /* gridDimY */ 1,
12004031bb98SShilei Tian                          /* gridDimZ */ 1, CudaThreadsPerBlock,
12014031bb98SShilei Tian                          /* blockDimY */ 1, /* blockDimZ */ 1,
1202*f1c821faSJoseph Huber                          DynamicMemorySize, Stream, &Args[0], nullptr);
12034031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuLaunchKernel\n"))
12044031bb98SShilei Tian       return OFFLOAD_FAIL;
12054031bb98SShilei Tian 
12064031bb98SShilei Tian     DP("Launch of entry point at " DPxMOD " successful!\n",
12074031bb98SShilei Tian        DPxPTR(TgtEntryPtr));
12084031bb98SShilei Tian 
12094031bb98SShilei Tian     return OFFLOAD_SUCCESS;
12104031bb98SShilei Tian   }
12114031bb98SShilei Tian 
12125449fbb5SJohannes Doerfert   int synchronize(const int DeviceId, __tgt_async_info *AsyncInfo) const {
12135449fbb5SJohannes Doerfert     CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue);
12144031bb98SShilei Tian     CUresult Err = cuStreamSynchronize(Stream);
12153de4cd12SGeorge Rokos 
12164031bb98SShilei Tian     // Once the stream is synchronized, return it to stream pool and reset
12175449fbb5SJohannes Doerfert     // AsyncInfo. This is to make sure the synchronization only works for its
12184031bb98SShilei Tian     // own tasks.
12195449fbb5SJohannes Doerfert     StreamManager->returnStream(DeviceId,
12205449fbb5SJohannes Doerfert                                 reinterpret_cast<CUstream>(AsyncInfo->Queue));
12215449fbb5SJohannes Doerfert     AsyncInfo->Queue = nullptr;
12223de4cd12SGeorge Rokos 
122316064e71SJoseph Huber     if (Err != CUDA_SUCCESS) {
122429338459SJoseph Huber       DP("Error when synchronizing stream. stream = " DPxMOD
122516064e71SJoseph Huber          ", async info ptr = " DPxMOD "\n",
122616064e71SJoseph Huber          DPxPTR(Stream), DPxPTR(AsyncInfo));
122716064e71SJoseph Huber       CUDA_ERR_STRING(Err);
122816064e71SJoseph Huber     }
122916064e71SJoseph Huber     return (Err == CUDA_SUCCESS) ? OFFLOAD_SUCCESS : OFFLOAD_FAIL;
12303de4cd12SGeorge Rokos   }
1231d2f85d09SJose M Monsalve Diaz 
1232d2f85d09SJose M Monsalve Diaz   void printDeviceInfo(int32_t device_id) {
1233d2f85d09SJose M Monsalve Diaz     char TmpChar[1000];
1234d2f85d09SJose M Monsalve Diaz     std::string TmpStr;
1235d2f85d09SJose M Monsalve Diaz     size_t TmpSt;
1236d2f85d09SJose M Monsalve Diaz     int TmpInt, TmpInt2, TmpInt3;
1237d2f85d09SJose M Monsalve Diaz 
1238d2f85d09SJose M Monsalve Diaz     CUdevice Device;
1239d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGet(&Device, device_id),
1240d2f85d09SJose M Monsalve Diaz                 "Error returned from cuCtxGetDevice\n");
1241d2f85d09SJose M Monsalve Diaz 
1242d2f85d09SJose M Monsalve Diaz     cuDriverGetVersion(&TmpInt);
1243d2f85d09SJose M Monsalve Diaz     printf("    CUDA Driver Version: \t\t%d \n", TmpInt);
1244d2f85d09SJose M Monsalve Diaz     printf("    CUDA Device Number: \t\t%d \n", device_id);
1245d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetName(TmpChar, 1000, Device),
1246d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetName\n");
1247d2f85d09SJose M Monsalve Diaz     printf("    Device Name: \t\t\t%s \n", TmpChar);
1248d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceTotalMem(&TmpSt, Device),
1249d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceTotalMem\n");
1250d2f85d09SJose M Monsalve Diaz     printf("    Global Memory Size: \t\t%zu bytes \n", TmpSt);
1251d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1252d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT, Device),
1253d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1254d2f85d09SJose M Monsalve Diaz     printf("    Number of Multiprocessors: \t\t%d \n", TmpInt);
1255d2f85d09SJose M Monsalve Diaz     checkResult(
1256d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_GPU_OVERLAP, Device),
1257d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1258d2f85d09SJose M Monsalve Diaz     printf("    Concurrent Copy and Execution: \t%s \n", BOOL2TEXT(TmpInt));
1259d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1260d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY, Device),
1261d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1262d2f85d09SJose M Monsalve Diaz     printf("    Total Constant Memory: \t\t%d bytes\n", TmpInt);
1263d2f85d09SJose M Monsalve Diaz     checkResult(
1264d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(
1265d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK, Device),
1266d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1267d2f85d09SJose M Monsalve Diaz     printf("    Max Shared Memory per Block: \t%d bytes \n", TmpInt);
126828939b6aSShilei Tian     checkResult(
126928939b6aSShilei Tian         cuDeviceGetAttribute(
1270d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK, Device),
1271d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1272d2f85d09SJose M Monsalve Diaz     printf("    Registers per Block: \t\t%d \n", TmpInt);
1273d2f85d09SJose M Monsalve Diaz     checkResult(
1274d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_WARP_SIZE, Device),
1275d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1276d2f85d09SJose M Monsalve Diaz     printf("    Warp Size: \t\t\t\t%d Threads \n", TmpInt);
1277d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1278d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK, Device),
1279d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1280d2f85d09SJose M Monsalve Diaz     printf("    Maximum Threads per Block: \t\t%d \n", TmpInt);
128128939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
128228939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X, Device),
1283d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
128428939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
128528939b6aSShilei Tian                     &TmpInt2, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y, Device),
1286d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
128728939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
128828939b6aSShilei Tian                     &TmpInt3, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z, Device),
1289d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1290d2f85d09SJose M Monsalve Diaz     printf("    Maximum Block Dimensions: \t\t%d, %d, %d \n", TmpInt, TmpInt2,
1291d2f85d09SJose M Monsalve Diaz            TmpInt3);
129228939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
129328939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X, Device),
1294d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
129528939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
129628939b6aSShilei Tian                     &TmpInt2, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y, Device),
1297d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
129828939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
129928939b6aSShilei Tian                     &TmpInt3, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z, Device),
1300d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1301d2f85d09SJose M Monsalve Diaz     printf("    Maximum Grid Dimensions: \t\t%d x %d x %d \n", TmpInt, TmpInt2,
1302d2f85d09SJose M Monsalve Diaz            TmpInt3);
1303d2f85d09SJose M Monsalve Diaz     checkResult(
1304d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_MAX_PITCH, Device),
1305d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1306d2f85d09SJose M Monsalve Diaz     printf("    Maximum Memory Pitch: \t\t%d bytes \n", TmpInt);
1307d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1308d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT, Device),
1309d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1310d2f85d09SJose M Monsalve Diaz     printf("    Texture Alignment: \t\t\t%d bytes \n", TmpInt);
1311d2f85d09SJose M Monsalve Diaz     checkResult(
1312d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_CLOCK_RATE, Device),
1313d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1314d2f85d09SJose M Monsalve Diaz     printf("    Clock Rate: \t\t\t%d kHz\n", TmpInt);
1315d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1316d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT, Device),
1317d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1318d2f85d09SJose M Monsalve Diaz     printf("    Execution Timeout: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1319d2f85d09SJose M Monsalve Diaz     checkResult(
1320d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_INTEGRATED, Device),
1321d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1322d2f85d09SJose M Monsalve Diaz     printf("    Integrated Device: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1323d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1324d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY, Device),
1325d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1326d2f85d09SJose M Monsalve Diaz     printf("    Can Map Host Memory: \t\t%s \n", BOOL2TEXT(TmpInt));
1327d2f85d09SJose M Monsalve Diaz     checkResult(
1328d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_MODE, Device),
1329d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1330d2f85d09SJose M Monsalve Diaz     if (TmpInt == CU_COMPUTEMODE_DEFAULT)
1331d2f85d09SJose M Monsalve Diaz       TmpStr = "DEFAULT";
1332d2f85d09SJose M Monsalve Diaz     else if (TmpInt == CU_COMPUTEMODE_PROHIBITED)
1333d2f85d09SJose M Monsalve Diaz       TmpStr = "PROHIBITED";
1334d2f85d09SJose M Monsalve Diaz     else if (TmpInt == CU_COMPUTEMODE_EXCLUSIVE_PROCESS)
1335d2f85d09SJose M Monsalve Diaz       TmpStr = "EXCLUSIVE PROCESS";
1336d2f85d09SJose M Monsalve Diaz     else
1337d2f85d09SJose M Monsalve Diaz       TmpStr = "unknown";
1338d2f85d09SJose M Monsalve Diaz     printf("    Compute Mode: \t\t\t%s \n", TmpStr.c_str());
1339d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1340d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS, Device),
1341d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1342d2f85d09SJose M Monsalve Diaz     printf("    Concurrent Kernels: \t\t%s \n", BOOL2TEXT(TmpInt));
1343d2f85d09SJose M Monsalve Diaz     checkResult(
1344d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_ECC_ENABLED, Device),
1345d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1346d2f85d09SJose M Monsalve Diaz     printf("    ECC Enabled: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1347d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1348d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE, Device),
1349d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1350d2f85d09SJose M Monsalve Diaz     printf("    Memory Clock Rate: \t\t\t%d kHz\n", TmpInt);
135128939b6aSShilei Tian     checkResult(
135228939b6aSShilei Tian         cuDeviceGetAttribute(
1353d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH, Device),
1354d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1355d2f85d09SJose M Monsalve Diaz     printf("    Memory Bus Width: \t\t\t%d bits\n", TmpInt);
135628939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE,
135728939b6aSShilei Tian                                      Device),
1358d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1359d2f85d09SJose M Monsalve Diaz     printf("    L2 Cache Size: \t\t\t%d bytes \n", TmpInt);
136028939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
136128939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR,
136228939b6aSShilei Tian                     Device),
1363d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1364d2f85d09SJose M Monsalve Diaz     printf("    Max Threads Per SMP: \t\t%d \n", TmpInt);
1365d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1366d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT, Device),
1367d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1368d2f85d09SJose M Monsalve Diaz     printf("    Async Engines: \t\t\t%s (%d) \n", BOOL2TEXT(TmpInt), TmpInt);
1369d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1370d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING, Device),
1371d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1372d2f85d09SJose M Monsalve Diaz     printf("    Unified Addressing: \t\t%s \n", BOOL2TEXT(TmpInt));
137328939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
137428939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY, Device),
1375d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1376d2f85d09SJose M Monsalve Diaz     printf("    Managed Memory: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1377d2f85d09SJose M Monsalve Diaz     checkResult(
1378d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(
1379d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS, Device),
1380d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1381d2f85d09SJose M Monsalve Diaz     printf("    Concurrent Managed Memory: \t\t%s \n", BOOL2TEXT(TmpInt));
1382d2f85d09SJose M Monsalve Diaz     checkResult(
1383d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(
1384d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED, Device),
1385d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1386d2f85d09SJose M Monsalve Diaz     printf("    Preemption Supported: \t\t%s \n", BOOL2TEXT(TmpInt));
1387d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1388d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH, Device),
1389d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1390d2f85d09SJose M Monsalve Diaz     printf("    Cooperative Launch: \t\t%s \n", BOOL2TEXT(TmpInt));
139128939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
139228939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD, Device),
1393d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1394d2f85d09SJose M Monsalve Diaz     printf("    Multi-Device Boars: \t\t%s \n", BOOL2TEXT(TmpInt));
139528939b6aSShilei Tian     checkResult(
139628939b6aSShilei Tian         cuDeviceGetAttribute(
139728939b6aSShilei Tian             &TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, Device),
1398d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
139928939b6aSShilei Tian     checkResult(
140028939b6aSShilei Tian         cuDeviceGetAttribute(
140128939b6aSShilei Tian             &TmpInt2, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, Device),
1402d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1403d2f85d09SJose M Monsalve Diaz     printf("    Compute Capabilities: \t\t%d%d \n", TmpInt, TmpInt2);
1404d2f85d09SJose M Monsalve Diaz   }
140529df4ab3SShilei Tian 
140629df4ab3SShilei Tian   int waitEvent(const int DeviceId, __tgt_async_info *AsyncInfo,
140729df4ab3SShilei Tian                 void *EventPtr) const {
140829df4ab3SShilei Tian     CUstream Stream = getStream(DeviceId, AsyncInfo);
140929df4ab3SShilei Tian     CUevent Event = reinterpret_cast<CUevent>(EventPtr);
141029df4ab3SShilei Tian 
141129df4ab3SShilei Tian     // We don't use CU_EVENT_WAIT_DEFAULT here as it is only available from
141229df4ab3SShilei Tian     // specific CUDA version, and defined as 0x0. In previous version, per CUDA
141329df4ab3SShilei Tian     // API document, that argument has to be 0x0.
141429df4ab3SShilei Tian     CUresult Err = cuStreamWaitEvent(Stream, Event, 0);
141529df4ab3SShilei Tian     if (Err != CUDA_SUCCESS) {
141629df4ab3SShilei Tian       DP("Error when waiting event. stream = " DPxMOD ", event = " DPxMOD "\n",
141729df4ab3SShilei Tian          DPxPTR(Stream), DPxPTR(Event));
141829df4ab3SShilei Tian       CUDA_ERR_STRING(Err);
141929df4ab3SShilei Tian       return OFFLOAD_FAIL;
142029df4ab3SShilei Tian     }
142129df4ab3SShilei Tian 
142229df4ab3SShilei Tian     return OFFLOAD_SUCCESS;
142329df4ab3SShilei Tian   }
14244031bb98SShilei Tian };
14254031bb98SShilei Tian 
14264031bb98SShilei Tian DeviceRTLTy DeviceRTL;
14274031bb98SShilei Tian } // namespace
14284031bb98SShilei Tian 
14294031bb98SShilei Tian // Exposed library API function
14304031bb98SShilei Tian #ifdef __cplusplus
14314031bb98SShilei Tian extern "C" {
14324031bb98SShilei Tian #endif
14334031bb98SShilei Tian 
14344031bb98SShilei Tian int32_t __tgt_rtl_is_valid_binary(__tgt_device_image *image) {
14354031bb98SShilei Tian   return elf_check_machine(image, /* EM_CUDA */ 190);
14364031bb98SShilei Tian }
14374031bb98SShilei Tian 
14384031bb98SShilei Tian int32_t __tgt_rtl_number_of_devices() { return DeviceRTL.getNumOfDevices(); }
14394031bb98SShilei Tian 
14404031bb98SShilei Tian int64_t __tgt_rtl_init_requires(int64_t RequiresFlags) {
144140470eb2SGeorge Rokos   DP("Init requires flags to %" PRId64 "\n", RequiresFlags);
14424031bb98SShilei Tian   DeviceRTL.setRequiresFlag(RequiresFlags);
14434031bb98SShilei Tian   return RequiresFlags;
14444031bb98SShilei Tian }
14454031bb98SShilei Tian 
1446a014fbbcSShilei Tian int32_t __tgt_rtl_is_data_exchangable(int32_t src_dev_id, int dst_dev_id) {
1447a014fbbcSShilei Tian   if (DeviceRTL.isValidDeviceId(src_dev_id) &&
1448a014fbbcSShilei Tian       DeviceRTL.isValidDeviceId(dst_dev_id))
1449a014fbbcSShilei Tian     return 1;
1450a014fbbcSShilei Tian 
1451a014fbbcSShilei Tian   return 0;
1452a014fbbcSShilei Tian }
1453a014fbbcSShilei Tian 
14544031bb98SShilei Tian int32_t __tgt_rtl_init_device(int32_t device_id) {
14554031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14564031bb98SShilei Tian 
14574031bb98SShilei Tian   return DeviceRTL.initDevice(device_id);
14584031bb98SShilei Tian }
14594031bb98SShilei Tian 
14604031bb98SShilei Tian __tgt_target_table *__tgt_rtl_load_binary(int32_t device_id,
14614031bb98SShilei Tian                                           __tgt_device_image *image) {
14624031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14634031bb98SShilei Tian 
14644031bb98SShilei Tian   return DeviceRTL.loadBinary(device_id, image);
14654031bb98SShilei Tian }
14664031bb98SShilei Tian 
14672468fdd9SGeorge Rokos void *__tgt_rtl_data_alloc(int32_t device_id, int64_t size, void *,
14682468fdd9SGeorge Rokos                            int32_t kind) {
14694031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14704031bb98SShilei Tian 
1471a15f8589SJoseph Huber   return DeviceRTL.dataAlloc(device_id, size, (TargetAllocTy)kind);
14724031bb98SShilei Tian }
14734031bb98SShilei Tian 
14744031bb98SShilei Tian int32_t __tgt_rtl_data_submit(int32_t device_id, void *tgt_ptr, void *hst_ptr,
14754031bb98SShilei Tian                               int64_t size) {
14764031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14774031bb98SShilei Tian 
14785449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
14794031bb98SShilei Tian   const int32_t rc = __tgt_rtl_data_submit_async(device_id, tgt_ptr, hst_ptr,
14805449fbb5SJohannes Doerfert                                                  size, &AsyncInfo);
14814031bb98SShilei Tian   if (rc != OFFLOAD_SUCCESS)
14824031bb98SShilei Tian     return OFFLOAD_FAIL;
14834031bb98SShilei Tian 
14845449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
14854031bb98SShilei Tian }
14864031bb98SShilei Tian 
14874031bb98SShilei Tian int32_t __tgt_rtl_data_submit_async(int32_t device_id, void *tgt_ptr,
14884031bb98SShilei Tian                                     void *hst_ptr, int64_t size,
14894031bb98SShilei Tian                                     __tgt_async_info *async_info_ptr) {
14904031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14914031bb98SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
14924031bb98SShilei Tian 
14934031bb98SShilei Tian   return DeviceRTL.dataSubmit(device_id, tgt_ptr, hst_ptr, size,
14944031bb98SShilei Tian                               async_info_ptr);
14954031bb98SShilei Tian }
14964031bb98SShilei Tian 
14974031bb98SShilei Tian int32_t __tgt_rtl_data_retrieve(int32_t device_id, void *hst_ptr, void *tgt_ptr,
14984031bb98SShilei Tian                                 int64_t size) {
14994031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15004031bb98SShilei Tian 
15015449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
15024031bb98SShilei Tian   const int32_t rc = __tgt_rtl_data_retrieve_async(device_id, hst_ptr, tgt_ptr,
15035449fbb5SJohannes Doerfert                                                    size, &AsyncInfo);
15044031bb98SShilei Tian   if (rc != OFFLOAD_SUCCESS)
15054031bb98SShilei Tian     return OFFLOAD_FAIL;
15064031bb98SShilei Tian 
15075449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
15084031bb98SShilei Tian }
15094031bb98SShilei Tian 
15104031bb98SShilei Tian int32_t __tgt_rtl_data_retrieve_async(int32_t device_id, void *hst_ptr,
15114031bb98SShilei Tian                                       void *tgt_ptr, int64_t size,
15124031bb98SShilei Tian                                       __tgt_async_info *async_info_ptr) {
15134031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15144031bb98SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
15154031bb98SShilei Tian 
15164031bb98SShilei Tian   return DeviceRTL.dataRetrieve(device_id, hst_ptr, tgt_ptr, size,
15174031bb98SShilei Tian                                 async_info_ptr);
15184031bb98SShilei Tian }
15194031bb98SShilei Tian 
1520a014fbbcSShilei Tian int32_t __tgt_rtl_data_exchange_async(int32_t src_dev_id, void *src_ptr,
1521a014fbbcSShilei Tian                                       int dst_dev_id, void *dst_ptr,
1522a014fbbcSShilei Tian                                       int64_t size,
15235449fbb5SJohannes Doerfert                                       __tgt_async_info *AsyncInfo) {
1524a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(src_dev_id) && "src_dev_id is invalid");
1525a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(dst_dev_id) && "dst_dev_id is invalid");
15265449fbb5SJohannes Doerfert   assert(AsyncInfo && "AsyncInfo is nullptr");
1527a014fbbcSShilei Tian 
1528a014fbbcSShilei Tian   return DeviceRTL.dataExchange(src_dev_id, src_ptr, dst_dev_id, dst_ptr, size,
15295449fbb5SJohannes Doerfert                                 AsyncInfo);
1530a014fbbcSShilei Tian }
1531a014fbbcSShilei Tian 
1532a014fbbcSShilei Tian int32_t __tgt_rtl_data_exchange(int32_t src_dev_id, void *src_ptr,
1533a014fbbcSShilei Tian                                 int32_t dst_dev_id, void *dst_ptr,
1534a014fbbcSShilei Tian                                 int64_t size) {
1535a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(src_dev_id) && "src_dev_id is invalid");
1536a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(dst_dev_id) && "dst_dev_id is invalid");
1537a014fbbcSShilei Tian 
15385449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
1539a014fbbcSShilei Tian   const int32_t rc = __tgt_rtl_data_exchange_async(
15405449fbb5SJohannes Doerfert       src_dev_id, src_ptr, dst_dev_id, dst_ptr, size, &AsyncInfo);
1541a014fbbcSShilei Tian   if (rc != OFFLOAD_SUCCESS)
1542a014fbbcSShilei Tian     return OFFLOAD_FAIL;
1543a014fbbcSShilei Tian 
15445449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(src_dev_id, &AsyncInfo);
1545a014fbbcSShilei Tian }
1546a014fbbcSShilei Tian 
15474031bb98SShilei Tian int32_t __tgt_rtl_data_delete(int32_t device_id, void *tgt_ptr) {
15484031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15494031bb98SShilei Tian 
15504031bb98SShilei Tian   return DeviceRTL.dataDelete(device_id, tgt_ptr);
15514031bb98SShilei Tian }
15524031bb98SShilei Tian 
15534031bb98SShilei Tian int32_t __tgt_rtl_run_target_team_region(int32_t device_id, void *tgt_entry_ptr,
15544031bb98SShilei Tian                                          void **tgt_args,
15554031bb98SShilei Tian                                          ptrdiff_t *tgt_offsets,
15564031bb98SShilei Tian                                          int32_t arg_num, int32_t team_num,
15574031bb98SShilei Tian                                          int32_t thread_limit,
15584031bb98SShilei Tian                                          uint64_t loop_tripcount) {
15594031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15604031bb98SShilei Tian 
15615449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
15624031bb98SShilei Tian   const int32_t rc = __tgt_rtl_run_target_team_region_async(
15634031bb98SShilei Tian       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num,
15645449fbb5SJohannes Doerfert       thread_limit, loop_tripcount, &AsyncInfo);
15654031bb98SShilei Tian   if (rc != OFFLOAD_SUCCESS)
15664031bb98SShilei Tian     return OFFLOAD_FAIL;
15674031bb98SShilei Tian 
15685449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
15694031bb98SShilei Tian }
15704031bb98SShilei Tian 
15714031bb98SShilei Tian int32_t __tgt_rtl_run_target_team_region_async(
15724031bb98SShilei Tian     int32_t device_id, void *tgt_entry_ptr, void **tgt_args,
15734031bb98SShilei Tian     ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t team_num,
15744031bb98SShilei Tian     int32_t thread_limit, uint64_t loop_tripcount,
15754031bb98SShilei Tian     __tgt_async_info *async_info_ptr) {
15764031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15774031bb98SShilei Tian 
15784031bb98SShilei Tian   return DeviceRTL.runTargetTeamRegion(
15794031bb98SShilei Tian       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num,
15804031bb98SShilei Tian       thread_limit, loop_tripcount, async_info_ptr);
15814031bb98SShilei Tian }
15823de4cd12SGeorge Rokos 
15833de4cd12SGeorge Rokos int32_t __tgt_rtl_run_target_region(int32_t device_id, void *tgt_entry_ptr,
158432ed2927SShilei Tian                                     void **tgt_args, ptrdiff_t *tgt_offsets,
158503ff643dSShilei Tian                                     int32_t arg_num) {
15864031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15874031bb98SShilei Tian 
15885449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
15894031bb98SShilei Tian   const int32_t rc = __tgt_rtl_run_target_region_async(
15905449fbb5SJohannes Doerfert       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, &AsyncInfo);
159103ff643dSShilei Tian   if (rc != OFFLOAD_SUCCESS)
159203ff643dSShilei Tian     return OFFLOAD_FAIL;
159303ff643dSShilei Tian 
15945449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
159503ff643dSShilei Tian }
159603ff643dSShilei Tian 
159703ff643dSShilei Tian int32_t __tgt_rtl_run_target_region_async(int32_t device_id,
159803ff643dSShilei Tian                                           void *tgt_entry_ptr, void **tgt_args,
159903ff643dSShilei Tian                                           ptrdiff_t *tgt_offsets,
160032ed2927SShilei Tian                                           int32_t arg_num,
16014031bb98SShilei Tian                                           __tgt_async_info *async_info_ptr) {
16024031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
16034031bb98SShilei Tian 
160403ff643dSShilei Tian   return __tgt_rtl_run_target_team_region_async(
16054031bb98SShilei Tian       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num,
16064031bb98SShilei Tian       /* team num*/ 1, /* thread_limit */ 1, /* loop_tripcount */ 0,
16074031bb98SShilei Tian       async_info_ptr);
160832ed2927SShilei Tian }
160932ed2927SShilei Tian 
16104031bb98SShilei Tian int32_t __tgt_rtl_synchronize(int32_t device_id,
16114031bb98SShilei Tian                               __tgt_async_info *async_info_ptr) {
16124031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
16134031bb98SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
16144031bb98SShilei Tian   assert(async_info_ptr->Queue && "async_info_ptr->Queue is nullptr");
161532ed2927SShilei Tian 
16164031bb98SShilei Tian   return DeviceRTL.synchronize(device_id, async_info_ptr);
16173de4cd12SGeorge Rokos }
16183de4cd12SGeorge Rokos 
16192b6f2008SJoseph Huber void __tgt_rtl_set_info_flag(uint32_t NewInfoLevel) {
162059b68490SJoseph Huber   std::atomic<uint32_t> &InfoLevel = getInfoLevelInternal();
16212b6f2008SJoseph Huber   InfoLevel.store(NewInfoLevel);
16222b6f2008SJoseph Huber }
16232b6f2008SJoseph Huber 
1624d2f85d09SJose M Monsalve Diaz void __tgt_rtl_print_device_info(int32_t device_id) {
1625d2f85d09SJose M Monsalve Diaz   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
1626d2f85d09SJose M Monsalve Diaz   DeviceRTL.printDeviceInfo(device_id);
1627d2f85d09SJose M Monsalve Diaz }
1628d2f85d09SJose M Monsalve Diaz 
162929df4ab3SShilei Tian int32_t __tgt_rtl_create_event(int32_t device_id, void **event) {
163029df4ab3SShilei Tian   assert(event && "event is nullptr");
163129df4ab3SShilei Tian   return createEvent(event);
163229df4ab3SShilei Tian }
163329df4ab3SShilei Tian 
163429df4ab3SShilei Tian int32_t __tgt_rtl_record_event(int32_t device_id, void *event_ptr,
163529df4ab3SShilei Tian                                __tgt_async_info *async_info_ptr) {
163629df4ab3SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
163729df4ab3SShilei Tian   assert(async_info_ptr->Queue && "async_info_ptr->Queue is nullptr");
163829df4ab3SShilei Tian   assert(event_ptr && "event_ptr is nullptr");
163929df4ab3SShilei Tian 
164029df4ab3SShilei Tian   return recordEvent(event_ptr, async_info_ptr);
164129df4ab3SShilei Tian }
164229df4ab3SShilei Tian 
164329df4ab3SShilei Tian int32_t __tgt_rtl_wait_event(int32_t device_id, void *event_ptr,
164429df4ab3SShilei Tian                              __tgt_async_info *async_info_ptr) {
164529df4ab3SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
164629df4ab3SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
164729df4ab3SShilei Tian   assert(event_ptr && "event is nullptr");
164829df4ab3SShilei Tian 
164929df4ab3SShilei Tian   return DeviceRTL.waitEvent(device_id, async_info_ptr, event_ptr);
165029df4ab3SShilei Tian }
165129df4ab3SShilei Tian 
165229df4ab3SShilei Tian int32_t __tgt_rtl_sync_event(int32_t device_id, void *event_ptr) {
165329df4ab3SShilei Tian   assert(event_ptr && "event is nullptr");
165429df4ab3SShilei Tian 
165529df4ab3SShilei Tian   return syncEvent(event_ptr);
165629df4ab3SShilei Tian }
165729df4ab3SShilei Tian 
165829df4ab3SShilei Tian int32_t __tgt_rtl_destroy_event(int32_t device_id, void *event_ptr) {
165929df4ab3SShilei Tian   assert(event_ptr && "event is nullptr");
166029df4ab3SShilei Tian 
166129df4ab3SShilei Tian   return destroyEvent(event_ptr);
166229df4ab3SShilei Tian }
166329df4ab3SShilei Tian 
16643de4cd12SGeorge Rokos #ifdef __cplusplus
16653de4cd12SGeorge Rokos }
16663de4cd12SGeorge Rokos #endif
1667