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;
106f1c821faSJoseph 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;
350f1c821faSJoseph Huber   // Amount of dynamic shared memory to use at launch.
351f1c821faSJoseph 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),
505f1c821faSJoseph Huber         EnvTeamThreadLimit(-1), RequiresFlags(OMP_REQ_UNDEFINED),
506f1c821faSJoseph 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     }
547f1c821faSJoseph Huber     if (const char *EnvStr = getenv("LIBOMPTARGET_SHARED_MEMORY_SIZE")) {
548f1c821faSJoseph Huber       // LIBOMPTARGET_SHARED_MEMORY_SIZE has been set
549f1c821faSJoseph Huber       DynamicMemorySize = std::stoi(EnvStr);
550*49e976c9SShilei Tian       DP("Parsed LIBOMPTARGET_SHARED_MEMORY_SIZE = %" PRIu64 "\n",
551*49e976c9SShilei Tian          DynamicMemorySize);
552f1c821faSJoseph Huber     }
553c5fe030cSGheorghe-Teodor Bercea 
554feed674dSShilei Tian     StreamManager =
5554031bb98SShilei Tian         std::make_unique<StreamManagerTy>(NumberOfDevices, DeviceData);
5560871d6d5SShilei Tian 
5570871d6d5SShilei Tian     for (int I = 0; I < NumberOfDevices; ++I)
5580871d6d5SShilei Tian       DeviceAllocators.emplace_back(I, DeviceData);
5590871d6d5SShilei Tian 
5600871d6d5SShilei Tian     // Get the size threshold from environment variable
5610871d6d5SShilei Tian     std::pair<size_t, bool> Res = MemoryManagerTy::getSizeThresholdFromEnv();
5620871d6d5SShilei Tian     UseMemoryManager = Res.second;
5630871d6d5SShilei Tian     size_t MemoryManagerThreshold = Res.first;
5640871d6d5SShilei Tian 
5650871d6d5SShilei Tian     if (UseMemoryManager)
5660871d6d5SShilei Tian       for (int I = 0; I < NumberOfDevices; ++I)
5670871d6d5SShilei Tian         MemoryManagers.emplace_back(std::make_unique<MemoryManagerTy>(
5680871d6d5SShilei Tian             DeviceAllocators[I], MemoryManagerThreshold));
5693de4cd12SGeorge Rokos   }
5703de4cd12SGeorge Rokos 
5714031bb98SShilei Tian   ~DeviceRTLTy() {
5720871d6d5SShilei Tian     // We first destruct memory managers in case that its dependent data are
5730871d6d5SShilei Tian     // destroyed before it.
5740871d6d5SShilei Tian     for (auto &M : MemoryManagers)
5750871d6d5SShilei Tian       M.release();
5760871d6d5SShilei Tian 
577feed674dSShilei Tian     StreamManager = nullptr;
578feed674dSShilei Tian 
5794031bb98SShilei Tian     for (CUmodule &M : Modules)
5804031bb98SShilei Tian       // Close module
5814031bb98SShilei Tian       if (M)
5824031bb98SShilei Tian         checkResult(cuModuleUnload(M), "Error returned from cuModuleUnload\n");
5834031bb98SShilei Tian 
5844031bb98SShilei Tian     for (DeviceDataTy &D : DeviceData) {
5854031bb98SShilei Tian       // Destroy context
586c5348aecSYe Luo       if (D.Context) {
587c5348aecSYe Luo         checkResult(cuCtxSetCurrent(D.Context),
588c5348aecSYe Luo                     "Error returned from cuCtxSetCurrent\n");
589c5348aecSYe Luo         CUdevice Device;
590c5348aecSYe Luo         checkResult(cuCtxGetDevice(&Device),
591c5348aecSYe Luo                     "Error returned from cuCtxGetDevice\n");
59295f0d1edSJon Chesterfield         checkResult(cuDevicePrimaryCtxRelease(Device),
59395f0d1edSJon Chesterfield                     "Error returned from cuDevicePrimaryCtxRelease\n");
594c5348aecSYe Luo       }
5953de4cd12SGeorge Rokos     }
5963de4cd12SGeorge Rokos   }
5973de4cd12SGeorge Rokos 
5984031bb98SShilei Tian   // Check whether a given DeviceId is valid
5994031bb98SShilei Tian   bool isValidDeviceId(const int DeviceId) const {
6004031bb98SShilei Tian     return DeviceId >= 0 && DeviceId < NumberOfDevices;
60132ed2927SShilei Tian   }
60232ed2927SShilei Tian 
603cb038927SShilei Tian   int getNumOfDevices() const { return NumberOfDevices; }
6044031bb98SShilei Tian 
6054031bb98SShilei Tian   void setRequiresFlag(const int64_t Flags) { this->RequiresFlags = Flags; }
6064031bb98SShilei Tian 
6074031bb98SShilei Tian   int initDevice(const int DeviceId) {
6084031bb98SShilei Tian     CUdevice Device;
6094031bb98SShilei Tian 
6104031bb98SShilei Tian     DP("Getting device %d\n", DeviceId);
6114031bb98SShilei Tian     CUresult Err = cuDeviceGet(&Device, DeviceId);
6124031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuDeviceGet\n"))
61332ed2927SShilei Tian       return OFFLOAD_FAIL;
6143de4cd12SGeorge Rokos 
615c5348aecSYe Luo     // Query the current flags of the primary context and set its flags if
616c5348aecSYe Luo     // it is inactive
617c5348aecSYe Luo     unsigned int FormerPrimaryCtxFlags = 0;
618c5348aecSYe Luo     int FormerPrimaryCtxIsActive = 0;
619c5348aecSYe Luo     Err = cuDevicePrimaryCtxGetState(Device, &FormerPrimaryCtxFlags,
620c5348aecSYe Luo                                      &FormerPrimaryCtxIsActive);
621c5348aecSYe Luo     if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxGetState\n"))
622c5348aecSYe Luo       return OFFLOAD_FAIL;
623c5348aecSYe Luo 
624c5348aecSYe Luo     if (FormerPrimaryCtxIsActive) {
625c5348aecSYe Luo       DP("The primary context is active, no change to its flags\n");
626c5348aecSYe Luo       if ((FormerPrimaryCtxFlags & CU_CTX_SCHED_MASK) !=
627c5348aecSYe Luo           CU_CTX_SCHED_BLOCKING_SYNC)
628c5348aecSYe Luo         DP("Warning the current flags are not CU_CTX_SCHED_BLOCKING_SYNC\n");
629c5348aecSYe Luo     } else {
630c5348aecSYe Luo       DP("The primary context is inactive, set its flags to "
631c5348aecSYe Luo          "CU_CTX_SCHED_BLOCKING_SYNC\n");
63295f0d1edSJon Chesterfield       Err = cuDevicePrimaryCtxSetFlags(Device, CU_CTX_SCHED_BLOCKING_SYNC);
63395f0d1edSJon Chesterfield       if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxSetFlags\n"))
634c5348aecSYe Luo         return OFFLOAD_FAIL;
635c5348aecSYe Luo     }
636c5348aecSYe Luo 
637c5348aecSYe Luo     // Retain the per device primary context and save it to use whenever this
638c5348aecSYe Luo     // device is selected.
639c5348aecSYe Luo     Err = cuDevicePrimaryCtxRetain(&DeviceData[DeviceId].Context, Device);
640c5348aecSYe Luo     if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxRetain\n"))
6413de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
6423de4cd12SGeorge Rokos 
6434031bb98SShilei Tian     Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
6444031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
6454031bb98SShilei Tian       return OFFLOAD_FAIL;
646a5153dbcSJohannes Doerfert 
647feed674dSShilei Tian     // Initialize stream pool
6484031bb98SShilei Tian     if (!StreamManager->initializeDeviceStreamPool(DeviceId))
6494031bb98SShilei Tian       return OFFLOAD_FAIL;
650a5153dbcSJohannes Doerfert 
651bb51d398SJonas Hahnfeld     // Query attributes to determine number of threads/block and blocks/grid.
6524031bb98SShilei Tian     int MaxGridDimX;
6534031bb98SShilei Tian     Err = cuDeviceGetAttribute(&MaxGridDimX, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X,
6544031bb98SShilei Tian                                Device);
6554031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
6564031bb98SShilei Tian       DP("Error getting max grid dimension, use default value %d\n",
6574031bb98SShilei Tian          DeviceRTLTy::DefaultNumTeams);
6584031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = DeviceRTLTy::DefaultNumTeams;
6594031bb98SShilei Tian     } else if (MaxGridDimX <= DeviceRTLTy::HardTeamLimit) {
6604031bb98SShilei Tian       DP("Using %d CUDA blocks per grid\n", MaxGridDimX);
6614031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = MaxGridDimX;
6623de4cd12SGeorge Rokos     } else {
6633de4cd12SGeorge Rokos       DP("Max CUDA blocks per grid %d exceeds the hard team limit %d, capping "
664bb51d398SJonas Hahnfeld          "at the hard limit\n",
6654031bb98SShilei Tian          MaxGridDimX, DeviceRTLTy::HardTeamLimit);
6664031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = DeviceRTLTy::HardTeamLimit;
6673de4cd12SGeorge Rokos     }
6683de4cd12SGeorge Rokos 
669bb51d398SJonas Hahnfeld     // We are only exploiting threads along the x axis.
6704031bb98SShilei Tian     int MaxBlockDimX;
6714031bb98SShilei Tian     Err = cuDeviceGetAttribute(&MaxBlockDimX,
6724031bb98SShilei Tian                                CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X, Device);
6734031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
6744031bb98SShilei Tian       DP("Error getting max block dimension, use default value %d\n",
6754031bb98SShilei Tian          DeviceRTLTy::DefaultNumThreads);
6764031bb98SShilei Tian       DeviceData[DeviceId].ThreadsPerBlock = DeviceRTLTy::DefaultNumThreads;
677422adaa8SJoseph Huber     } else {
6784031bb98SShilei Tian       DP("Using %d CUDA threads per block\n", MaxBlockDimX);
6794031bb98SShilei Tian       DeviceData[DeviceId].ThreadsPerBlock = MaxBlockDimX;
680422adaa8SJoseph Huber 
681422adaa8SJoseph Huber       if (EnvTeamThreadLimit > 0 &&
682422adaa8SJoseph Huber           DeviceData[DeviceId].ThreadsPerBlock > EnvTeamThreadLimit) {
683422adaa8SJoseph Huber         DP("Max CUDA threads per block %d exceeds the thread limit %d set by "
684422adaa8SJoseph Huber            "OMP_TEAMS_THREAD_LIMIT, capping at the limit\n",
685422adaa8SJoseph Huber            DeviceData[DeviceId].ThreadsPerBlock, EnvTeamThreadLimit);
686422adaa8SJoseph Huber         DeviceData[DeviceId].ThreadsPerBlock = EnvTeamThreadLimit;
687422adaa8SJoseph Huber       }
688422adaa8SJoseph Huber       if (DeviceData[DeviceId].ThreadsPerBlock > DeviceRTLTy::HardThreadLimit) {
6894031bb98SShilei Tian         DP("Max CUDA threads per block %d exceeds the hard thread limit %d, "
6904031bb98SShilei Tian            "capping at the hard limit\n",
691422adaa8SJoseph Huber            DeviceData[DeviceId].ThreadsPerBlock, DeviceRTLTy::HardThreadLimit);
6924031bb98SShilei Tian         DeviceData[DeviceId].ThreadsPerBlock = DeviceRTLTy::HardThreadLimit;
6933de4cd12SGeorge Rokos       }
694422adaa8SJoseph Huber     }
6953de4cd12SGeorge Rokos 
6964031bb98SShilei Tian     // Get and set warp size
6974031bb98SShilei Tian     int WarpSize;
6984031bb98SShilei Tian     Err =
6994031bb98SShilei Tian         cuDeviceGetAttribute(&WarpSize, CU_DEVICE_ATTRIBUTE_WARP_SIZE, Device);
7004031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
7014031bb98SShilei Tian       DP("Error getting warp size, assume default value 32\n");
7024031bb98SShilei Tian       DeviceData[DeviceId].WarpSize = 32;
703bb51d398SJonas Hahnfeld     } else {
7044031bb98SShilei Tian       DP("Using warp size %d\n", WarpSize);
7054031bb98SShilei Tian       DeviceData[DeviceId].WarpSize = WarpSize;
7063de4cd12SGeorge Rokos     }
7073de4cd12SGeorge Rokos 
7083de4cd12SGeorge Rokos     // Adjust teams to the env variables
7094031bb98SShilei Tian     if (EnvTeamLimit > 0 && DeviceData[DeviceId].BlocksPerGrid > EnvTeamLimit) {
7103de4cd12SGeorge Rokos       DP("Capping max CUDA blocks per grid to OMP_TEAM_LIMIT=%d\n",
7114031bb98SShilei Tian          EnvTeamLimit);
7124031bb98SShilei Tian       DeviceData[DeviceId].BlocksPerGrid = EnvTeamLimit;
7133de4cd12SGeorge Rokos     }
7143de4cd12SGeorge Rokos 
7153817ba13SJoseph Huber     size_t StackLimit;
7163817ba13SJoseph Huber     size_t HeapLimit;
7173817ba13SJoseph Huber     if (const char *EnvStr = getenv("LIBOMPTARGET_STACK_SIZE")) {
7183817ba13SJoseph Huber       StackLimit = std::stol(EnvStr);
7193817ba13SJoseph Huber       if (cuCtxSetLimit(CU_LIMIT_STACK_SIZE, StackLimit) != CUDA_SUCCESS)
7203817ba13SJoseph Huber         return OFFLOAD_FAIL;
7213817ba13SJoseph Huber     } else {
7223817ba13SJoseph Huber       if (cuCtxGetLimit(&StackLimit, CU_LIMIT_STACK_SIZE) != CUDA_SUCCESS)
7233817ba13SJoseph Huber         return OFFLOAD_FAIL;
7243817ba13SJoseph Huber     }
7253817ba13SJoseph Huber     if (const char *EnvStr = getenv("LIBOMPTARGET_HEAP_SIZE")) {
7263817ba13SJoseph Huber       HeapLimit = std::stol(EnvStr);
7273817ba13SJoseph Huber       if (cuCtxSetLimit(CU_LIMIT_MALLOC_HEAP_SIZE, HeapLimit) != CUDA_SUCCESS)
7283817ba13SJoseph Huber         return OFFLOAD_FAIL;
7293817ba13SJoseph Huber     } else {
7303817ba13SJoseph Huber       if (cuCtxGetLimit(&HeapLimit, CU_LIMIT_MALLOC_HEAP_SIZE) != CUDA_SUCCESS)
7313817ba13SJoseph Huber         return OFFLOAD_FAIL;
7323817ba13SJoseph Huber     }
7333817ba13SJoseph Huber 
734119a9ea1SJoseph Huber     INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId,
735ae209397SJoseph Huber          "Device supports up to %d CUDA blocks and %d threads with a "
736ae209397SJoseph Huber          "warp size of %d\n",
737ae209397SJoseph Huber          DeviceData[DeviceId].BlocksPerGrid,
738ae209397SJoseph Huber          DeviceData[DeviceId].ThreadsPerBlock, DeviceData[DeviceId].WarpSize);
7393817ba13SJoseph Huber     INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId,
7403817ba13SJoseph Huber          "Device heap size is %d Bytes, device stack size is %d Bytes per "
7413817ba13SJoseph Huber          "thread\n",
7423817ba13SJoseph Huber          (int)HeapLimit, (int)StackLimit);
7433de4cd12SGeorge Rokos 
7443de4cd12SGeorge Rokos     // Set default number of teams
7454031bb98SShilei Tian     if (EnvNumTeams > 0) {
7463de4cd12SGeorge Rokos       DP("Default number of teams set according to environment %d\n",
7474031bb98SShilei Tian          EnvNumTeams);
7484031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = EnvNumTeams;
7493de4cd12SGeorge Rokos     } else {
7504031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = DeviceRTLTy::DefaultNumTeams;
7513de4cd12SGeorge Rokos       DP("Default number of teams set according to library's default %d\n",
7524031bb98SShilei Tian          DeviceRTLTy::DefaultNumTeams);
7533de4cd12SGeorge Rokos     }
7544031bb98SShilei Tian 
7554031bb98SShilei Tian     if (DeviceData[DeviceId].NumTeams > DeviceData[DeviceId].BlocksPerGrid) {
7563de4cd12SGeorge Rokos       DP("Default number of teams exceeds device limit, capping at %d\n",
7574031bb98SShilei Tian          DeviceData[DeviceId].BlocksPerGrid);
7584031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].BlocksPerGrid;
7593de4cd12SGeorge Rokos     }
7603de4cd12SGeorge Rokos 
7613de4cd12SGeorge Rokos     // Set default number of threads
7624031bb98SShilei Tian     DeviceData[DeviceId].NumThreads = DeviceRTLTy::DefaultNumThreads;
7633de4cd12SGeorge Rokos     DP("Default number of threads set according to library's default %d\n",
7644031bb98SShilei Tian        DeviceRTLTy::DefaultNumThreads);
7654031bb98SShilei Tian     if (DeviceData[DeviceId].NumThreads >
7664031bb98SShilei Tian         DeviceData[DeviceId].ThreadsPerBlock) {
7673de4cd12SGeorge Rokos       DP("Default number of threads exceeds device limit, capping at %d\n",
7684031bb98SShilei Tian          DeviceData[DeviceId].ThreadsPerBlock);
7694031bb98SShilei Tian       DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].ThreadsPerBlock;
7703de4cd12SGeorge Rokos     }
7713de4cd12SGeorge Rokos 
7723de4cd12SGeorge Rokos     return OFFLOAD_SUCCESS;
7733de4cd12SGeorge Rokos   }
7743de4cd12SGeorge Rokos 
7754031bb98SShilei Tian   __tgt_target_table *loadBinary(const int DeviceId,
7764031bb98SShilei Tian                                  const __tgt_device_image *Image) {
7774031bb98SShilei Tian     // Set the context we are using
7784031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
7794031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
7804031bb98SShilei Tian       return nullptr;
7813de4cd12SGeorge Rokos 
7823de4cd12SGeorge Rokos     // Clear the offload table as we are going to create a new one.
7834031bb98SShilei Tian     clearOffloadEntriesTable(DeviceId);
7843de4cd12SGeorge Rokos 
7853de4cd12SGeorge Rokos     // Create the module and extract the function pointers.
7864031bb98SShilei Tian     CUmodule Module;
7874031bb98SShilei Tian     DP("Load data from image " DPxMOD "\n", DPxPTR(Image->ImageStart));
7884031bb98SShilei Tian     Err = cuModuleLoadDataEx(&Module, Image->ImageStart, 0, nullptr, nullptr);
7894031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuModuleLoadDataEx\n"))
7904031bb98SShilei Tian       return nullptr;
7913de4cd12SGeorge Rokos 
7923de4cd12SGeorge Rokos     DP("CUDA module successfully loaded!\n");
7934031bb98SShilei Tian 
7944031bb98SShilei Tian     Modules.push_back(Module);
7953de4cd12SGeorge Rokos 
7963de4cd12SGeorge Rokos     // Find the symbols in the module by name.
7974031bb98SShilei Tian     const __tgt_offload_entry *HostBegin = Image->EntriesBegin;
7984031bb98SShilei Tian     const __tgt_offload_entry *HostEnd = Image->EntriesEnd;
7993de4cd12SGeorge Rokos 
8005272d29eSJohannes Doerfert     std::list<KernelTy> &KernelsList = DeviceData[DeviceId].KernelsList;
8014031bb98SShilei Tian     for (const __tgt_offload_entry *E = HostBegin; E != HostEnd; ++E) {
8024031bb98SShilei Tian       if (!E->addr) {
8034031bb98SShilei Tian         // We return nullptr when something like this happens, the host should
8044031bb98SShilei Tian         // have always something in the address to uniquely identify the target
8054031bb98SShilei Tian         // region.
8064031bb98SShilei Tian         DP("Invalid binary: host entry '<null>' (size = %zd)...\n", E->size);
8074031bb98SShilei Tian         return nullptr;
8083de4cd12SGeorge Rokos       }
8093de4cd12SGeorge Rokos 
8104031bb98SShilei Tian       if (E->size) {
8114031bb98SShilei Tian         __tgt_offload_entry Entry = *E;
8124031bb98SShilei Tian         CUdeviceptr CUPtr;
8134031bb98SShilei Tian         size_t CUSize;
81495f0d1edSJon Chesterfield         Err = cuModuleGetGlobal(&CUPtr, &CUSize, Module, E->name);
8154031bb98SShilei Tian         // We keep this style here because we need the name
8164031bb98SShilei Tian         if (Err != CUDA_SUCCESS) {
8172ce16810SJoseph Huber           REPORT("Loading global '%s' Failed\n", E->name);
8184031bb98SShilei Tian           CUDA_ERR_STRING(Err);
8194031bb98SShilei Tian           return nullptr;
8203de4cd12SGeorge Rokos         }
8213de4cd12SGeorge Rokos 
8224031bb98SShilei Tian         if (CUSize != E->size) {
8234031bb98SShilei Tian           DP("Loading global '%s' - size mismatch (%zd != %zd)\n", E->name,
8244031bb98SShilei Tian              CUSize, E->size);
8254031bb98SShilei Tian           return nullptr;
8263de4cd12SGeorge Rokos         }
8273de4cd12SGeorge Rokos 
8283de4cd12SGeorge Rokos         DP("Entry point " DPxMOD " maps to global %s (" DPxMOD ")\n",
8294031bb98SShilei Tian            DPxPTR(E - HostBegin), E->name, DPxPTR(CUPtr));
8304031bb98SShilei Tian 
8314031bb98SShilei Tian         Entry.addr = (void *)(CUPtr);
8323de4cd12SGeorge Rokos 
833aace6d28SGheorghe-Teodor Bercea         // Note: In the current implementation declare target variables
834aace6d28SGheorghe-Teodor Bercea         // can either be link or to. This means that once unified
835aace6d28SGheorghe-Teodor Bercea         // memory is activated via the requires directive, the variable
836aace6d28SGheorghe-Teodor Bercea         // can be used directly from the host in both cases.
837aace6d28SGheorghe-Teodor Bercea         // TODO: when variables types other than to or link are added,
8384c6a098aSKazuaki Ishizaki         // the below condition should be changed to explicitly
839aace6d28SGheorghe-Teodor Bercea         // check for to and link variables types:
8404031bb98SShilei Tian         // (RequiresFlags & OMP_REQ_UNIFIED_SHARED_MEMORY && (e->flags &
8414031bb98SShilei Tian         // OMP_DECLARE_TARGET_LINK || e->flags == OMP_DECLARE_TARGET_TO))
8424031bb98SShilei Tian         if (RequiresFlags & OMP_REQ_UNIFIED_SHARED_MEMORY) {
843aace6d28SGheorghe-Teodor Bercea           // If unified memory is present any target link or to variables
844c5fe030cSGheorghe-Teodor Bercea           // can access host addresses directly. There is no longer a
845c5fe030cSGheorghe-Teodor Bercea           // need for device copies.
84695f0d1edSJon Chesterfield           cuMemcpyHtoD(CUPtr, E->addr, sizeof(void *));
8474031bb98SShilei Tian           DP("Copy linked variable host address (" DPxMOD
8484031bb98SShilei Tian              ") to device address (" DPxMOD ")\n",
8494031bb98SShilei Tian              DPxPTR(*((void **)E->addr)), DPxPTR(CUPtr));
850c5fe030cSGheorghe-Teodor Bercea         }
851c5fe030cSGheorghe-Teodor Bercea 
8524031bb98SShilei Tian         addOffloadEntry(DeviceId, Entry);
8533de4cd12SGeorge Rokos 
8543de4cd12SGeorge Rokos         continue;
8553de4cd12SGeorge Rokos       }
8563de4cd12SGeorge Rokos 
8574031bb98SShilei Tian       CUfunction Func;
8584031bb98SShilei Tian       Err = cuModuleGetFunction(&Func, Module, E->name);
8594031bb98SShilei Tian       // We keep this style here because we need the name
8604031bb98SShilei Tian       if (Err != CUDA_SUCCESS) {
8612ce16810SJoseph Huber         REPORT("Loading '%s' Failed\n", E->name);
8624031bb98SShilei Tian         CUDA_ERR_STRING(Err);
8634031bb98SShilei Tian         return nullptr;
8643de4cd12SGeorge Rokos       }
8653de4cd12SGeorge Rokos 
8663de4cd12SGeorge Rokos       DP("Entry point " DPxMOD " maps to %s (" DPxMOD ")\n",
8674031bb98SShilei Tian          DPxPTR(E - HostBegin), E->name, DPxPTR(Func));
8683de4cd12SGeorge Rokos 
8693de4cd12SGeorge Rokos       // default value GENERIC (in case symbol is missing from cubin file)
8703de4cd12SGeorge Rokos       int8_t ExecModeVal = ExecutionModeType::GENERIC;
8714031bb98SShilei Tian       std::string ExecModeNameStr(E->name);
8723de4cd12SGeorge Rokos       ExecModeNameStr += "_exec_mode";
8733de4cd12SGeorge Rokos       const char *ExecModeName = ExecModeNameStr.c_str();
8743de4cd12SGeorge Rokos 
8753de4cd12SGeorge Rokos       CUdeviceptr ExecModePtr;
8764031bb98SShilei Tian       size_t CUSize;
87795f0d1edSJon Chesterfield       Err = cuModuleGetGlobal(&ExecModePtr, &CUSize, Module, ExecModeName);
8784031bb98SShilei Tian       if (Err == CUDA_SUCCESS) {
8794031bb98SShilei Tian         if (CUSize != sizeof(int8_t)) {
8803de4cd12SGeorge Rokos           DP("Loading global exec_mode '%s' - size mismatch (%zd != %zd)\n",
8814031bb98SShilei Tian              ExecModeName, CUSize, sizeof(int8_t));
8824031bb98SShilei Tian           return nullptr;
8833de4cd12SGeorge Rokos         }
8843de4cd12SGeorge Rokos 
88595f0d1edSJon Chesterfield         Err = cuMemcpyDtoH(&ExecModeVal, ExecModePtr, CUSize);
8864031bb98SShilei Tian         if (Err != CUDA_SUCCESS) {
8872ce16810SJoseph Huber           REPORT("Error when copying data from device to host. Pointers: "
8883de4cd12SGeorge Rokos                  "host = " DPxMOD ", device = " DPxMOD ", size = %zd\n",
8894031bb98SShilei Tian                  DPxPTR(&ExecModeVal), DPxPTR(ExecModePtr), CUSize);
8904031bb98SShilei Tian           CUDA_ERR_STRING(Err);
8914031bb98SShilei Tian           return nullptr;
8923de4cd12SGeorge Rokos         }
8933de4cd12SGeorge Rokos 
8947d576392SJoseph Huber         if (ExecModeVal < 0 || ExecModeVal > 2) {
8953de4cd12SGeorge Rokos           DP("Error wrong exec_mode value specified in cubin file: %d\n",
8963de4cd12SGeorge Rokos              ExecModeVal);
8974031bb98SShilei Tian           return nullptr;
8983de4cd12SGeorge Rokos         }
8993de4cd12SGeorge Rokos       } else {
900edb8acdcSJoseph Huber         DP("Loading global exec_mode '%s' - symbol missing, using default "
9014031bb98SShilei Tian            "value GENERIC (1)\n",
9024031bb98SShilei Tian            ExecModeName);
9033de4cd12SGeorge Rokos       }
9043de4cd12SGeorge Rokos 
9054031bb98SShilei Tian       KernelsList.emplace_back(Func, ExecModeVal);
9063de4cd12SGeorge Rokos 
9074031bb98SShilei Tian       __tgt_offload_entry Entry = *E;
9084031bb98SShilei Tian       Entry.addr = &KernelsList.back();
9094031bb98SShilei Tian       addOffloadEntry(DeviceId, Entry);
9103de4cd12SGeorge Rokos     }
9113de4cd12SGeorge Rokos 
912e1c7a46dSGuansong Zhang     // send device environment data to the device
913e1c7a46dSGuansong Zhang     {
914ec02c34bSJoseph Huber       // TODO: The device ID used here is not the real device ID used by OpenMP.
915ec02c34bSJoseph Huber       omptarget_device_environmentTy DeviceEnv{
916ec02c34bSJoseph Huber           0, static_cast<uint32_t>(NumberOfDevices),
917f1c821faSJoseph Huber           static_cast<uint32_t>(DeviceId), DynamicMemorySize};
918e1c7a46dSGuansong Zhang 
919e1c7a46dSGuansong Zhang #ifdef OMPTARGET_DEBUG
9204031bb98SShilei Tian       if (const char *EnvStr = getenv("LIBOMPTARGET_DEVICE_RTL_DEBUG"))
9214031bb98SShilei Tian         DeviceEnv.debug_level = std::stoi(EnvStr);
922e1c7a46dSGuansong Zhang #endif
923e1c7a46dSGuansong Zhang 
9244031bb98SShilei Tian       const char *DeviceEnvName = "omptarget_device_environment";
9254031bb98SShilei Tian       CUdeviceptr DeviceEnvPtr;
9264031bb98SShilei Tian       size_t CUSize;
927e1c7a46dSGuansong Zhang 
92895f0d1edSJon Chesterfield       Err = cuModuleGetGlobal(&DeviceEnvPtr, &CUSize, Module, DeviceEnvName);
9294031bb98SShilei Tian       if (Err == CUDA_SUCCESS) {
9304031bb98SShilei Tian         if (CUSize != sizeof(DeviceEnv)) {
9312ce16810SJoseph Huber           REPORT(
9322ce16810SJoseph Huber               "Global device_environment '%s' - size mismatch (%zu != %zu)\n",
9334031bb98SShilei Tian               DeviceEnvName, CUSize, sizeof(int32_t));
9344031bb98SShilei Tian           CUDA_ERR_STRING(Err);
9354031bb98SShilei Tian           return nullptr;
936e1c7a46dSGuansong Zhang         }
937e1c7a46dSGuansong Zhang 
93895f0d1edSJon Chesterfield         Err = cuMemcpyHtoD(DeviceEnvPtr, &DeviceEnv, CUSize);
9394031bb98SShilei Tian         if (Err != CUDA_SUCCESS) {
9402ce16810SJoseph Huber           REPORT("Error when copying data from host to device. Pointers: "
941e1c7a46dSGuansong Zhang                  "host = " DPxMOD ", device = " DPxMOD ", size = %zu\n",
9424031bb98SShilei Tian                  DPxPTR(&DeviceEnv), DPxPTR(DeviceEnvPtr), CUSize);
9434031bb98SShilei Tian           CUDA_ERR_STRING(Err);
9444031bb98SShilei Tian           return nullptr;
945e1c7a46dSGuansong Zhang         }
946e1c7a46dSGuansong Zhang 
9474031bb98SShilei Tian         DP("Sending global device environment data %zu bytes\n", CUSize);
948e1c7a46dSGuansong Zhang       } else {
9494031bb98SShilei Tian         DP("Finding global device environment '%s' - symbol missing.\n",
9504031bb98SShilei Tian            DeviceEnvName);
9514031bb98SShilei Tian         DP("Continue, considering this is a device RTL which does not accept "
9524031bb98SShilei Tian            "environment setting.\n");
953e1c7a46dSGuansong Zhang       }
954e1c7a46dSGuansong Zhang     }
955e1c7a46dSGuansong Zhang 
9564031bb98SShilei Tian     return getOffloadEntriesTable(DeviceId);
9573de4cd12SGeorge Rokos   }
9583de4cd12SGeorge Rokos 
959a15f8589SJoseph Huber   void *dataAlloc(const int DeviceId, const int64_t Size,
960a15f8589SJoseph Huber                   const TargetAllocTy Kind) {
961a15f8589SJoseph Huber     switch (Kind) {
962a15f8589SJoseph Huber     case TARGET_ALLOC_DEFAULT:
963a15f8589SJoseph Huber     case TARGET_ALLOC_DEVICE:
9640871d6d5SShilei Tian       if (UseMemoryManager)
9650871d6d5SShilei Tian         return MemoryManagers[DeviceId]->allocate(Size, nullptr);
966a15f8589SJoseph Huber       else
967a15f8589SJoseph Huber         return DeviceAllocators[DeviceId].allocate(Size, nullptr, Kind);
968a15f8589SJoseph Huber     case TARGET_ALLOC_HOST:
969a15f8589SJoseph Huber     case TARGET_ALLOC_SHARED:
970a15f8589SJoseph Huber       return DeviceAllocators[DeviceId].allocate(Size, nullptr, Kind);
971a15f8589SJoseph Huber     }
9724031bb98SShilei Tian 
973a15f8589SJoseph Huber     REPORT("Invalid target data allocation kind or requested allocator not "
974a15f8589SJoseph Huber            "implemented yet\n");
975a15f8589SJoseph Huber 
976a15f8589SJoseph Huber     return nullptr;
9773de4cd12SGeorge Rokos   }
9783de4cd12SGeorge Rokos 
9794031bb98SShilei Tian   int dataSubmit(const int DeviceId, const void *TgtPtr, const void *HstPtr,
9805449fbb5SJohannes Doerfert                  const int64_t Size, __tgt_async_info *AsyncInfo) const {
9815449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
9823de4cd12SGeorge Rokos 
9834031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
9844031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
9853de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
9863de4cd12SGeorge Rokos 
9875449fbb5SJohannes Doerfert     CUstream Stream = getStream(DeviceId, AsyncInfo);
9883de4cd12SGeorge Rokos 
98995f0d1edSJon Chesterfield     Err = cuMemcpyHtoDAsync((CUdeviceptr)TgtPtr, HstPtr, Size, Stream);
9904031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
99129338459SJoseph Huber       DP("Error when copying data from host to device. Pointers: host "
9922ce16810SJoseph Huber          "= " DPxMOD ", device = " DPxMOD ", size = %" PRId64 "\n",
9934031bb98SShilei Tian          DPxPTR(HstPtr), DPxPTR(TgtPtr), Size);
9944031bb98SShilei Tian       CUDA_ERR_STRING(Err);
9953de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
9963de4cd12SGeorge Rokos     }
9973de4cd12SGeorge Rokos 
9983de4cd12SGeorge Rokos     return OFFLOAD_SUCCESS;
9993de4cd12SGeorge Rokos   }
10003de4cd12SGeorge Rokos 
10014031bb98SShilei Tian   int dataRetrieve(const int DeviceId, void *HstPtr, const void *TgtPtr,
10025449fbb5SJohannes Doerfert                    const int64_t Size, __tgt_async_info *AsyncInfo) const {
10035449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
10044031bb98SShilei Tian 
10054031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
10064031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
100703ff643dSShilei Tian       return OFFLOAD_FAIL;
100803ff643dSShilei Tian 
10095449fbb5SJohannes Doerfert     CUstream Stream = getStream(DeviceId, AsyncInfo);
101003ff643dSShilei Tian 
101195f0d1edSJon Chesterfield     Err = cuMemcpyDtoHAsync(HstPtr, (CUdeviceptr)TgtPtr, Size, Stream);
10124031bb98SShilei Tian     if (Err != CUDA_SUCCESS) {
101329338459SJoseph Huber       DP("Error when copying data from device to host. Pointers: host "
10142ce16810SJoseph Huber          "= " DPxMOD ", device = " DPxMOD ", size = %" PRId64 "\n",
10154031bb98SShilei Tian          DPxPTR(HstPtr), DPxPTR(TgtPtr), Size);
10164031bb98SShilei Tian       CUDA_ERR_STRING(Err);
10173de4cd12SGeorge Rokos       return OFFLOAD_FAIL;
10183de4cd12SGeorge Rokos     }
10193de4cd12SGeorge Rokos 
10204031bb98SShilei Tian     return OFFLOAD_SUCCESS;
10214031bb98SShilei Tian   }
10224031bb98SShilei Tian 
1023a014fbbcSShilei Tian   int dataExchange(int SrcDevId, const void *SrcPtr, int DstDevId, void *DstPtr,
10245449fbb5SJohannes Doerfert                    int64_t Size, __tgt_async_info *AsyncInfo) const {
10255449fbb5SJohannes Doerfert     assert(AsyncInfo && "AsyncInfo is nullptr");
1026a014fbbcSShilei Tian 
1027a014fbbcSShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[SrcDevId].Context);
1028a014fbbcSShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
1029a014fbbcSShilei Tian       return OFFLOAD_FAIL;
1030a014fbbcSShilei Tian 
10315449fbb5SJohannes Doerfert     CUstream Stream = getStream(SrcDevId, AsyncInfo);
1032a014fbbcSShilei Tian 
1033a014fbbcSShilei Tian     // If they are two devices, we try peer to peer copy first
1034a014fbbcSShilei Tian     if (SrcDevId != DstDevId) {
1035a014fbbcSShilei Tian       int CanAccessPeer = 0;
1036a014fbbcSShilei Tian       Err = cuDeviceCanAccessPeer(&CanAccessPeer, SrcDevId, DstDevId);
1037a014fbbcSShilei Tian       if (Err != CUDA_SUCCESS) {
10382ce16810SJoseph Huber         REPORT("Error returned from cuDeviceCanAccessPeer. src = %" PRId32
1039a014fbbcSShilei Tian                ", dst = %" PRId32 "\n",
1040a014fbbcSShilei Tian                SrcDevId, DstDevId);
1041a014fbbcSShilei Tian         CUDA_ERR_STRING(Err);
1042a014fbbcSShilei Tian         return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1043a014fbbcSShilei Tian       }
1044a014fbbcSShilei Tian 
1045a014fbbcSShilei Tian       if (!CanAccessPeer) {
1046a014fbbcSShilei Tian         DP("P2P memcpy not supported so fall back to D2D memcpy");
1047a014fbbcSShilei Tian         return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1048a014fbbcSShilei Tian       }
1049a014fbbcSShilei Tian 
1050a014fbbcSShilei Tian       Err = cuCtxEnablePeerAccess(DeviceData[DstDevId].Context, 0);
1051a014fbbcSShilei Tian       if (Err != CUDA_SUCCESS) {
10522ce16810SJoseph Huber         REPORT("Error returned from cuCtxEnablePeerAccess. src = %" PRId32
1053a014fbbcSShilei Tian                ", dst = %" PRId32 "\n",
1054a014fbbcSShilei Tian                SrcDevId, DstDevId);
1055a014fbbcSShilei Tian         CUDA_ERR_STRING(Err);
1056a014fbbcSShilei Tian         return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1057a014fbbcSShilei Tian       }
1058a014fbbcSShilei Tian 
1059a014fbbcSShilei Tian       Err = cuMemcpyPeerAsync((CUdeviceptr)DstPtr, DeviceData[DstDevId].Context,
1060a014fbbcSShilei Tian                               (CUdeviceptr)SrcPtr, DeviceData[SrcDevId].Context,
1061a014fbbcSShilei Tian                               Size, Stream);
1062a014fbbcSShilei Tian       if (Err == CUDA_SUCCESS)
1063a014fbbcSShilei Tian         return OFFLOAD_SUCCESS;
1064a014fbbcSShilei Tian 
106529338459SJoseph Huber       DP("Error returned from cuMemcpyPeerAsync. src_ptr = " DPxMOD
106629338459SJoseph Huber          ", src_id =%" PRId32 ", dst_ptr = " DPxMOD ", dst_id =%" PRId32 "\n",
106745bb073dSYe Luo          DPxPTR(SrcPtr), SrcDevId, DPxPTR(DstPtr), DstDevId);
1068a014fbbcSShilei Tian       CUDA_ERR_STRING(Err);
1069a014fbbcSShilei Tian     }
1070a014fbbcSShilei Tian 
1071a014fbbcSShilei Tian     return memcpyDtoD(SrcPtr, DstPtr, Size, Stream);
1072a014fbbcSShilei Tian   }
1073a014fbbcSShilei Tian 
10740871d6d5SShilei Tian   int dataDelete(const int DeviceId, void *TgtPtr) {
10750871d6d5SShilei Tian     if (UseMemoryManager)
10760871d6d5SShilei Tian       return MemoryManagers[DeviceId]->free(TgtPtr);
10774031bb98SShilei Tian 
10780871d6d5SShilei Tian     return DeviceAllocators[DeviceId].free(TgtPtr);
10794031bb98SShilei Tian   }
10804031bb98SShilei Tian 
1081aa27cfc1SJohannes Doerfert   int runTargetTeamRegion(const int DeviceId, void *TgtEntryPtr, void **TgtArgs,
1082aa27cfc1SJohannes Doerfert                           ptrdiff_t *TgtOffsets, const int ArgNum,
1083aa27cfc1SJohannes Doerfert                           const int TeamNum, const int ThreadLimit,
10844031bb98SShilei Tian                           const unsigned int LoopTripCount,
10854031bb98SShilei Tian                           __tgt_async_info *AsyncInfo) const {
10864031bb98SShilei Tian     CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context);
10874031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n"))
10884031bb98SShilei Tian       return OFFLOAD_FAIL;
10894031bb98SShilei Tian 
10903de4cd12SGeorge Rokos     // All args are references.
10914031bb98SShilei Tian     std::vector<void *> Args(ArgNum);
10924031bb98SShilei Tian     std::vector<void *> Ptrs(ArgNum);
10933de4cd12SGeorge Rokos 
10944031bb98SShilei Tian     for (int I = 0; I < ArgNum; ++I) {
10954031bb98SShilei Tian       Ptrs[I] = (void *)((intptr_t)TgtArgs[I] + TgtOffsets[I]);
10964031bb98SShilei Tian       Args[I] = &Ptrs[I];
10971546d319SGeorge Rokos     }
10983de4cd12SGeorge Rokos 
1099aa27cfc1SJohannes Doerfert     KernelTy *KernelInfo = reinterpret_cast<KernelTy *>(TgtEntryPtr);
11003de4cd12SGeorge Rokos 
1101aa27cfc1SJohannes Doerfert     int CudaThreadsPerBlock;
11024031bb98SShilei Tian     if (ThreadLimit > 0) {
11034031bb98SShilei Tian       DP("Setting CUDA threads per block to requested %d\n", ThreadLimit);
11044031bb98SShilei Tian       CudaThreadsPerBlock = ThreadLimit;
11053de4cd12SGeorge Rokos       // Add master warp if necessary
11063de4cd12SGeorge Rokos       if (KernelInfo->ExecutionMode == GENERIC) {
11074031bb98SShilei Tian         DP("Adding master warp: +%d threads\n", DeviceData[DeviceId].WarpSize);
11084031bb98SShilei Tian         CudaThreadsPerBlock += DeviceData[DeviceId].WarpSize;
11093de4cd12SGeorge Rokos       }
1110c13df8e5SGeorge Rokos     } else {
1111c13df8e5SGeorge Rokos       DP("Setting CUDA threads per block to default %d\n",
11124031bb98SShilei Tian          DeviceData[DeviceId].NumThreads);
11134031bb98SShilei Tian       CudaThreadsPerBlock = DeviceData[DeviceId].NumThreads;
1114c13df8e5SGeorge Rokos     }
11153de4cd12SGeorge Rokos 
11164031bb98SShilei Tian     if (CudaThreadsPerBlock > DeviceData[DeviceId].ThreadsPerBlock) {
11173de4cd12SGeorge Rokos       DP("Threads per block capped at device limit %d\n",
11184031bb98SShilei Tian          DeviceData[DeviceId].ThreadsPerBlock);
11194031bb98SShilei Tian       CudaThreadsPerBlock = DeviceData[DeviceId].ThreadsPerBlock;
11203de4cd12SGeorge Rokos     }
11213de4cd12SGeorge Rokos 
1122aa27cfc1SJohannes Doerfert     if (!KernelInfo->MaxThreadsPerBlock) {
1123aa27cfc1SJohannes Doerfert       Err = cuFuncGetAttribute(&KernelInfo->MaxThreadsPerBlock,
11244031bb98SShilei Tian                                CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK,
11254031bb98SShilei Tian                                KernelInfo->Func);
1126aa27cfc1SJohannes Doerfert       if (!checkResult(Err, "Error returned from cuFuncGetAttribute\n"))
1127aa27cfc1SJohannes Doerfert         return OFFLOAD_FAIL;
1128aa27cfc1SJohannes Doerfert     }
1129aa27cfc1SJohannes Doerfert 
1130aa27cfc1SJohannes Doerfert     if (KernelInfo->MaxThreadsPerBlock < CudaThreadsPerBlock) {
1131aa27cfc1SJohannes Doerfert       DP("Threads per block capped at kernel limit %d\n",
1132aa27cfc1SJohannes Doerfert          KernelInfo->MaxThreadsPerBlock);
1133aa27cfc1SJohannes Doerfert       CudaThreadsPerBlock = KernelInfo->MaxThreadsPerBlock;
11343de4cd12SGeorge Rokos     }
11353de4cd12SGeorge Rokos 
11364031bb98SShilei Tian     unsigned int CudaBlocksPerGrid;
11374031bb98SShilei Tian     if (TeamNum <= 0) {
11384031bb98SShilei Tian       if (LoopTripCount > 0 && EnvNumTeams < 0) {
1139c13df8e5SGeorge Rokos         if (KernelInfo->ExecutionMode == SPMD) {
11404031bb98SShilei Tian           // We have a combined construct, i.e. `target teams distribute
11414031bb98SShilei Tian           // parallel for [simd]`. We launch so many teams so that each thread
11424031bb98SShilei Tian           // will execute one iteration of the loop. round up to the nearest
11434031bb98SShilei Tian           // integer
11444031bb98SShilei Tian           CudaBlocksPerGrid = ((LoopTripCount - 1) / CudaThreadsPerBlock) + 1;
11457d576392SJoseph Huber         } else if (KernelInfo->ExecutionMode == GENERIC) {
1146c13df8e5SGeorge Rokos           // If we reach this point, then we have a non-combined construct, i.e.
1147c13df8e5SGeorge Rokos           // `teams distribute` with a nested `parallel for` and each team is
1148c13df8e5SGeorge Rokos           // assigned one iteration of the `distribute` loop. E.g.:
1149c13df8e5SGeorge Rokos           //
1150c13df8e5SGeorge Rokos           // #pragma omp target teams distribute
1151c13df8e5SGeorge Rokos           // for(...loop_tripcount...) {
1152c13df8e5SGeorge Rokos           //   #pragma omp parallel for
1153c13df8e5SGeorge Rokos           //   for(...) {}
1154c13df8e5SGeorge Rokos           // }
1155c13df8e5SGeorge Rokos           //
1156c13df8e5SGeorge Rokos           // Threads within a team will execute the iterations of the `parallel`
1157c13df8e5SGeorge Rokos           // loop.
11584031bb98SShilei Tian           CudaBlocksPerGrid = LoopTripCount;
11597d576392SJoseph Huber         } else if (KernelInfo->ExecutionMode == SPMD_GENERIC) {
11607d576392SJoseph Huber           // If we reach this point, then we are executing a kernel that was
11617d576392SJoseph Huber           // transformed from Generic-mode to SPMD-mode. This kernel has
11627d576392SJoseph Huber           // SPMD-mode execution, but needs its blocks to be scheduled
11637d576392SJoseph Huber           // differently because the current loop trip count only applies to the
11647d576392SJoseph Huber           // `teams distribute` region and will create var too few blocks using
11657d576392SJoseph Huber           // the regular SPMD-mode method.
11667d576392SJoseph Huber           CudaBlocksPerGrid = LoopTripCount;
1167a158d366SJoseph Huber         } else {
1168a158d366SJoseph Huber           REPORT("Unknown execution mode: %d\n", KernelInfo->ExecutionMode);
1169a158d366SJoseph Huber           return OFFLOAD_FAIL;
1170c13df8e5SGeorge Rokos         }
117145bb073dSYe Luo         DP("Using %d teams due to loop trip count %" PRIu32
11724031bb98SShilei Tian            " and number of threads per block %d\n",
11734031bb98SShilei Tian            CudaBlocksPerGrid, LoopTripCount, CudaThreadsPerBlock);
11743de4cd12SGeorge Rokos       } else {
11754031bb98SShilei Tian         DP("Using default number of teams %d\n", DeviceData[DeviceId].NumTeams);
11764031bb98SShilei Tian         CudaBlocksPerGrid = DeviceData[DeviceId].NumTeams;
11773de4cd12SGeorge Rokos       }
11784031bb98SShilei Tian     } else if (TeamNum > DeviceData[DeviceId].BlocksPerGrid) {
11793de4cd12SGeorge Rokos       DP("Capping number of teams to team limit %d\n",
11804031bb98SShilei Tian          DeviceData[DeviceId].BlocksPerGrid);
11814031bb98SShilei Tian       CudaBlocksPerGrid = DeviceData[DeviceId].BlocksPerGrid;
11823de4cd12SGeorge Rokos     } else {
11834031bb98SShilei Tian       DP("Using requested number of teams %d\n", TeamNum);
11844031bb98SShilei Tian       CudaBlocksPerGrid = TeamNum;
11853de4cd12SGeorge Rokos     }
11863de4cd12SGeorge Rokos 
1187119a9ea1SJoseph Huber     INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId,
1188ae209397SJoseph Huber          "Launching kernel %s with %d blocks and %d threads in %s "
1189ae209397SJoseph Huber          "mode\n",
1190ae209397SJoseph Huber          (getOffloadEntry(DeviceId, TgtEntryPtr))
1191ae209397SJoseph Huber              ? getOffloadEntry(DeviceId, TgtEntryPtr)->name
1192ae209397SJoseph Huber              : "(null)",
1193ae209397SJoseph Huber          CudaBlocksPerGrid, CudaThreadsPerBlock,
11947d576392SJoseph Huber          (KernelInfo->ExecutionMode != SPMD
119576c0c0caSJoseph Huber               ? (KernelInfo->ExecutionMode == GENERIC ? "Generic"
119676c0c0caSJoseph Huber                                                       : "SPMD-Generic")
11977d576392SJoseph Huber               : "SPMD"));
11983de4cd12SGeorge Rokos 
11994031bb98SShilei Tian     CUstream Stream = getStream(DeviceId, AsyncInfo);
12004031bb98SShilei Tian     Err = cuLaunchKernel(KernelInfo->Func, CudaBlocksPerGrid, /* gridDimY */ 1,
12014031bb98SShilei Tian                          /* gridDimZ */ 1, CudaThreadsPerBlock,
12024031bb98SShilei Tian                          /* blockDimY */ 1, /* blockDimZ */ 1,
1203f1c821faSJoseph Huber                          DynamicMemorySize, Stream, &Args[0], nullptr);
12044031bb98SShilei Tian     if (!checkResult(Err, "Error returned from cuLaunchKernel\n"))
12054031bb98SShilei Tian       return OFFLOAD_FAIL;
12064031bb98SShilei Tian 
12074031bb98SShilei Tian     DP("Launch of entry point at " DPxMOD " successful!\n",
12084031bb98SShilei Tian        DPxPTR(TgtEntryPtr));
12094031bb98SShilei Tian 
12104031bb98SShilei Tian     return OFFLOAD_SUCCESS;
12114031bb98SShilei Tian   }
12124031bb98SShilei Tian 
12135449fbb5SJohannes Doerfert   int synchronize(const int DeviceId, __tgt_async_info *AsyncInfo) const {
12145449fbb5SJohannes Doerfert     CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue);
12154031bb98SShilei Tian     CUresult Err = cuStreamSynchronize(Stream);
12163de4cd12SGeorge Rokos 
12174031bb98SShilei Tian     // Once the stream is synchronized, return it to stream pool and reset
12185449fbb5SJohannes Doerfert     // AsyncInfo. This is to make sure the synchronization only works for its
12194031bb98SShilei Tian     // own tasks.
12205449fbb5SJohannes Doerfert     StreamManager->returnStream(DeviceId,
12215449fbb5SJohannes Doerfert                                 reinterpret_cast<CUstream>(AsyncInfo->Queue));
12225449fbb5SJohannes Doerfert     AsyncInfo->Queue = nullptr;
12233de4cd12SGeorge Rokos 
122416064e71SJoseph Huber     if (Err != CUDA_SUCCESS) {
122529338459SJoseph Huber       DP("Error when synchronizing stream. stream = " DPxMOD
122616064e71SJoseph Huber          ", async info ptr = " DPxMOD "\n",
122716064e71SJoseph Huber          DPxPTR(Stream), DPxPTR(AsyncInfo));
122816064e71SJoseph Huber       CUDA_ERR_STRING(Err);
122916064e71SJoseph Huber     }
123016064e71SJoseph Huber     return (Err == CUDA_SUCCESS) ? OFFLOAD_SUCCESS : OFFLOAD_FAIL;
12313de4cd12SGeorge Rokos   }
1232d2f85d09SJose M Monsalve Diaz 
1233d2f85d09SJose M Monsalve Diaz   void printDeviceInfo(int32_t device_id) {
1234d2f85d09SJose M Monsalve Diaz     char TmpChar[1000];
1235d2f85d09SJose M Monsalve Diaz     std::string TmpStr;
1236d2f85d09SJose M Monsalve Diaz     size_t TmpSt;
1237d2f85d09SJose M Monsalve Diaz     int TmpInt, TmpInt2, TmpInt3;
1238d2f85d09SJose M Monsalve Diaz 
1239d2f85d09SJose M Monsalve Diaz     CUdevice Device;
1240d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGet(&Device, device_id),
1241d2f85d09SJose M Monsalve Diaz                 "Error returned from cuCtxGetDevice\n");
1242d2f85d09SJose M Monsalve Diaz 
1243d2f85d09SJose M Monsalve Diaz     cuDriverGetVersion(&TmpInt);
1244d2f85d09SJose M Monsalve Diaz     printf("    CUDA Driver Version: \t\t%d \n", TmpInt);
1245d2f85d09SJose M Monsalve Diaz     printf("    CUDA Device Number: \t\t%d \n", device_id);
1246d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetName(TmpChar, 1000, Device),
1247d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetName\n");
1248d2f85d09SJose M Monsalve Diaz     printf("    Device Name: \t\t\t%s \n", TmpChar);
1249d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceTotalMem(&TmpSt, Device),
1250d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceTotalMem\n");
1251d2f85d09SJose M Monsalve Diaz     printf("    Global Memory Size: \t\t%zu bytes \n", TmpSt);
1252d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1253d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT, Device),
1254d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1255d2f85d09SJose M Monsalve Diaz     printf("    Number of Multiprocessors: \t\t%d \n", TmpInt);
1256d2f85d09SJose M Monsalve Diaz     checkResult(
1257d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_GPU_OVERLAP, Device),
1258d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1259d2f85d09SJose M Monsalve Diaz     printf("    Concurrent Copy and Execution: \t%s \n", BOOL2TEXT(TmpInt));
1260d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1261d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY, Device),
1262d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1263d2f85d09SJose M Monsalve Diaz     printf("    Total Constant Memory: \t\t%d bytes\n", TmpInt);
1264d2f85d09SJose M Monsalve Diaz     checkResult(
1265d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(
1266d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK, Device),
1267d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1268d2f85d09SJose M Monsalve Diaz     printf("    Max Shared Memory per Block: \t%d bytes \n", TmpInt);
126928939b6aSShilei Tian     checkResult(
127028939b6aSShilei Tian         cuDeviceGetAttribute(
1271d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK, Device),
1272d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1273d2f85d09SJose M Monsalve Diaz     printf("    Registers per Block: \t\t%d \n", TmpInt);
1274d2f85d09SJose M Monsalve Diaz     checkResult(
1275d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_WARP_SIZE, Device),
1276d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1277d2f85d09SJose M Monsalve Diaz     printf("    Warp Size: \t\t\t\t%d Threads \n", TmpInt);
1278d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1279d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK, Device),
1280d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1281d2f85d09SJose M Monsalve Diaz     printf("    Maximum Threads per Block: \t\t%d \n", TmpInt);
128228939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
128328939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X, Device),
1284d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
128528939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
128628939b6aSShilei Tian                     &TmpInt2, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y, Device),
1287d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
128828939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
128928939b6aSShilei Tian                     &TmpInt3, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z, Device),
1290d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1291d2f85d09SJose M Monsalve Diaz     printf("    Maximum Block Dimensions: \t\t%d, %d, %d \n", TmpInt, TmpInt2,
1292d2f85d09SJose M Monsalve Diaz            TmpInt3);
129328939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
129428939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X, Device),
1295d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
129628939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
129728939b6aSShilei Tian                     &TmpInt2, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y, Device),
1298d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
129928939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
130028939b6aSShilei Tian                     &TmpInt3, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z, Device),
1301d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1302d2f85d09SJose M Monsalve Diaz     printf("    Maximum Grid Dimensions: \t\t%d x %d x %d \n", TmpInt, TmpInt2,
1303d2f85d09SJose M Monsalve Diaz            TmpInt3);
1304d2f85d09SJose M Monsalve Diaz     checkResult(
1305d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_MAX_PITCH, Device),
1306d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1307d2f85d09SJose M Monsalve Diaz     printf("    Maximum Memory Pitch: \t\t%d bytes \n", TmpInt);
1308d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1309d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT, Device),
1310d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1311d2f85d09SJose M Monsalve Diaz     printf("    Texture Alignment: \t\t\t%d bytes \n", TmpInt);
1312d2f85d09SJose M Monsalve Diaz     checkResult(
1313d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_CLOCK_RATE, Device),
1314d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1315d2f85d09SJose M Monsalve Diaz     printf("    Clock Rate: \t\t\t%d kHz\n", TmpInt);
1316d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1317d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT, Device),
1318d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1319d2f85d09SJose M Monsalve Diaz     printf("    Execution Timeout: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1320d2f85d09SJose M Monsalve Diaz     checkResult(
1321d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_INTEGRATED, Device),
1322d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1323d2f85d09SJose M Monsalve Diaz     printf("    Integrated Device: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1324d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1325d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY, Device),
1326d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1327d2f85d09SJose M Monsalve Diaz     printf("    Can Map Host Memory: \t\t%s \n", BOOL2TEXT(TmpInt));
1328d2f85d09SJose M Monsalve Diaz     checkResult(
1329d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_MODE, Device),
1330d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1331d2f85d09SJose M Monsalve Diaz     if (TmpInt == CU_COMPUTEMODE_DEFAULT)
1332d2f85d09SJose M Monsalve Diaz       TmpStr = "DEFAULT";
1333d2f85d09SJose M Monsalve Diaz     else if (TmpInt == CU_COMPUTEMODE_PROHIBITED)
1334d2f85d09SJose M Monsalve Diaz       TmpStr = "PROHIBITED";
1335d2f85d09SJose M Monsalve Diaz     else if (TmpInt == CU_COMPUTEMODE_EXCLUSIVE_PROCESS)
1336d2f85d09SJose M Monsalve Diaz       TmpStr = "EXCLUSIVE PROCESS";
1337d2f85d09SJose M Monsalve Diaz     else
1338d2f85d09SJose M Monsalve Diaz       TmpStr = "unknown";
1339d2f85d09SJose M Monsalve Diaz     printf("    Compute Mode: \t\t\t%s \n", TmpStr.c_str());
1340d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1341d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS, Device),
1342d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1343d2f85d09SJose M Monsalve Diaz     printf("    Concurrent Kernels: \t\t%s \n", BOOL2TEXT(TmpInt));
1344d2f85d09SJose M Monsalve Diaz     checkResult(
1345d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_ECC_ENABLED, Device),
1346d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1347d2f85d09SJose M Monsalve Diaz     printf("    ECC Enabled: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1348d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1349d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE, Device),
1350d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1351d2f85d09SJose M Monsalve Diaz     printf("    Memory Clock Rate: \t\t\t%d kHz\n", TmpInt);
135228939b6aSShilei Tian     checkResult(
135328939b6aSShilei Tian         cuDeviceGetAttribute(
1354d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH, Device),
1355d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1356d2f85d09SJose M Monsalve Diaz     printf("    Memory Bus Width: \t\t\t%d bits\n", TmpInt);
135728939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE,
135828939b6aSShilei Tian                                      Device),
1359d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1360d2f85d09SJose M Monsalve Diaz     printf("    L2 Cache Size: \t\t\t%d bytes \n", TmpInt);
136128939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
136228939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR,
136328939b6aSShilei Tian                     Device),
1364d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1365d2f85d09SJose M Monsalve Diaz     printf("    Max Threads Per SMP: \t\t%d \n", TmpInt);
1366d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1367d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT, Device),
1368d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1369d2f85d09SJose M Monsalve Diaz     printf("    Async Engines: \t\t\t%s (%d) \n", BOOL2TEXT(TmpInt), TmpInt);
1370d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1371d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING, Device),
1372d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1373d2f85d09SJose M Monsalve Diaz     printf("    Unified Addressing: \t\t%s \n", BOOL2TEXT(TmpInt));
137428939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
137528939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY, Device),
1376d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1377d2f85d09SJose M Monsalve Diaz     printf("    Managed Memory: \t\t\t%s \n", BOOL2TEXT(TmpInt));
1378d2f85d09SJose M Monsalve Diaz     checkResult(
1379d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(
1380d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS, Device),
1381d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1382d2f85d09SJose M Monsalve Diaz     printf("    Concurrent Managed Memory: \t\t%s \n", BOOL2TEXT(TmpInt));
1383d2f85d09SJose M Monsalve Diaz     checkResult(
1384d2f85d09SJose M Monsalve Diaz         cuDeviceGetAttribute(
1385d2f85d09SJose M Monsalve Diaz             &TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED, Device),
1386d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1387d2f85d09SJose M Monsalve Diaz     printf("    Preemption Supported: \t\t%s \n", BOOL2TEXT(TmpInt));
1388d2f85d09SJose M Monsalve Diaz     checkResult(cuDeviceGetAttribute(
1389d2f85d09SJose M Monsalve Diaz                     &TmpInt, CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH, Device),
1390d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1391d2f85d09SJose M Monsalve Diaz     printf("    Cooperative Launch: \t\t%s \n", BOOL2TEXT(TmpInt));
139228939b6aSShilei Tian     checkResult(cuDeviceGetAttribute(
139328939b6aSShilei Tian                     &TmpInt, CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD, Device),
1394d2f85d09SJose M Monsalve Diaz                 "Error returned from cuDeviceGetAttribute\n");
1395d2f85d09SJose M Monsalve Diaz     printf("    Multi-Device Boars: \t\t%s \n", BOOL2TEXT(TmpInt));
139628939b6aSShilei Tian     checkResult(
139728939b6aSShilei Tian         cuDeviceGetAttribute(
139828939b6aSShilei Tian             &TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, Device),
1399d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
140028939b6aSShilei Tian     checkResult(
140128939b6aSShilei Tian         cuDeviceGetAttribute(
140228939b6aSShilei Tian             &TmpInt2, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, Device),
1403d2f85d09SJose M Monsalve Diaz         "Error returned from cuDeviceGetAttribute\n");
1404d2f85d09SJose M Monsalve Diaz     printf("    Compute Capabilities: \t\t%d%d \n", TmpInt, TmpInt2);
1405d2f85d09SJose M Monsalve Diaz   }
140629df4ab3SShilei Tian 
140729df4ab3SShilei Tian   int waitEvent(const int DeviceId, __tgt_async_info *AsyncInfo,
140829df4ab3SShilei Tian                 void *EventPtr) const {
140929df4ab3SShilei Tian     CUstream Stream = getStream(DeviceId, AsyncInfo);
141029df4ab3SShilei Tian     CUevent Event = reinterpret_cast<CUevent>(EventPtr);
141129df4ab3SShilei Tian 
141229df4ab3SShilei Tian     // We don't use CU_EVENT_WAIT_DEFAULT here as it is only available from
141329df4ab3SShilei Tian     // specific CUDA version, and defined as 0x0. In previous version, per CUDA
141429df4ab3SShilei Tian     // API document, that argument has to be 0x0.
141529df4ab3SShilei Tian     CUresult Err = cuStreamWaitEvent(Stream, Event, 0);
141629df4ab3SShilei Tian     if (Err != CUDA_SUCCESS) {
141729df4ab3SShilei Tian       DP("Error when waiting event. stream = " DPxMOD ", event = " DPxMOD "\n",
141829df4ab3SShilei Tian          DPxPTR(Stream), DPxPTR(Event));
141929df4ab3SShilei Tian       CUDA_ERR_STRING(Err);
142029df4ab3SShilei Tian       return OFFLOAD_FAIL;
142129df4ab3SShilei Tian     }
142229df4ab3SShilei Tian 
142329df4ab3SShilei Tian     return OFFLOAD_SUCCESS;
142429df4ab3SShilei Tian   }
14254031bb98SShilei Tian };
14264031bb98SShilei Tian 
14274031bb98SShilei Tian DeviceRTLTy DeviceRTL;
14284031bb98SShilei Tian } // namespace
14294031bb98SShilei Tian 
14304031bb98SShilei Tian // Exposed library API function
14314031bb98SShilei Tian #ifdef __cplusplus
14324031bb98SShilei Tian extern "C" {
14334031bb98SShilei Tian #endif
14344031bb98SShilei Tian 
14354031bb98SShilei Tian int32_t __tgt_rtl_is_valid_binary(__tgt_device_image *image) {
14364031bb98SShilei Tian   return elf_check_machine(image, /* EM_CUDA */ 190);
14374031bb98SShilei Tian }
14384031bb98SShilei Tian 
14394031bb98SShilei Tian int32_t __tgt_rtl_number_of_devices() { return DeviceRTL.getNumOfDevices(); }
14404031bb98SShilei Tian 
14414031bb98SShilei Tian int64_t __tgt_rtl_init_requires(int64_t RequiresFlags) {
144240470eb2SGeorge Rokos   DP("Init requires flags to %" PRId64 "\n", RequiresFlags);
14434031bb98SShilei Tian   DeviceRTL.setRequiresFlag(RequiresFlags);
14444031bb98SShilei Tian   return RequiresFlags;
14454031bb98SShilei Tian }
14464031bb98SShilei Tian 
1447a014fbbcSShilei Tian int32_t __tgt_rtl_is_data_exchangable(int32_t src_dev_id, int dst_dev_id) {
1448a014fbbcSShilei Tian   if (DeviceRTL.isValidDeviceId(src_dev_id) &&
1449a014fbbcSShilei Tian       DeviceRTL.isValidDeviceId(dst_dev_id))
1450a014fbbcSShilei Tian     return 1;
1451a014fbbcSShilei Tian 
1452a014fbbcSShilei Tian   return 0;
1453a014fbbcSShilei Tian }
1454a014fbbcSShilei Tian 
14554031bb98SShilei Tian int32_t __tgt_rtl_init_device(int32_t device_id) {
14564031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14574031bb98SShilei Tian 
14584031bb98SShilei Tian   return DeviceRTL.initDevice(device_id);
14594031bb98SShilei Tian }
14604031bb98SShilei Tian 
14614031bb98SShilei Tian __tgt_target_table *__tgt_rtl_load_binary(int32_t device_id,
14624031bb98SShilei Tian                                           __tgt_device_image *image) {
14634031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14644031bb98SShilei Tian 
14654031bb98SShilei Tian   return DeviceRTL.loadBinary(device_id, image);
14664031bb98SShilei Tian }
14674031bb98SShilei Tian 
14682468fdd9SGeorge Rokos void *__tgt_rtl_data_alloc(int32_t device_id, int64_t size, void *,
14692468fdd9SGeorge Rokos                            int32_t kind) {
14704031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14714031bb98SShilei Tian 
1472a15f8589SJoseph Huber   return DeviceRTL.dataAlloc(device_id, size, (TargetAllocTy)kind);
14734031bb98SShilei Tian }
14744031bb98SShilei Tian 
14754031bb98SShilei Tian int32_t __tgt_rtl_data_submit(int32_t device_id, void *tgt_ptr, void *hst_ptr,
14764031bb98SShilei Tian                               int64_t size) {
14774031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14784031bb98SShilei Tian 
14795449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
14804031bb98SShilei Tian   const int32_t rc = __tgt_rtl_data_submit_async(device_id, tgt_ptr, hst_ptr,
14815449fbb5SJohannes Doerfert                                                  size, &AsyncInfo);
14824031bb98SShilei Tian   if (rc != OFFLOAD_SUCCESS)
14834031bb98SShilei Tian     return OFFLOAD_FAIL;
14844031bb98SShilei Tian 
14855449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
14864031bb98SShilei Tian }
14874031bb98SShilei Tian 
14884031bb98SShilei Tian int32_t __tgt_rtl_data_submit_async(int32_t device_id, void *tgt_ptr,
14894031bb98SShilei Tian                                     void *hst_ptr, int64_t size,
14904031bb98SShilei Tian                                     __tgt_async_info *async_info_ptr) {
14914031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
14924031bb98SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
14934031bb98SShilei Tian 
14944031bb98SShilei Tian   return DeviceRTL.dataSubmit(device_id, tgt_ptr, hst_ptr, size,
14954031bb98SShilei Tian                               async_info_ptr);
14964031bb98SShilei Tian }
14974031bb98SShilei Tian 
14984031bb98SShilei Tian int32_t __tgt_rtl_data_retrieve(int32_t device_id, void *hst_ptr, void *tgt_ptr,
14994031bb98SShilei Tian                                 int64_t size) {
15004031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15014031bb98SShilei Tian 
15025449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
15034031bb98SShilei Tian   const int32_t rc = __tgt_rtl_data_retrieve_async(device_id, hst_ptr, tgt_ptr,
15045449fbb5SJohannes Doerfert                                                    size, &AsyncInfo);
15054031bb98SShilei Tian   if (rc != OFFLOAD_SUCCESS)
15064031bb98SShilei Tian     return OFFLOAD_FAIL;
15074031bb98SShilei Tian 
15085449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
15094031bb98SShilei Tian }
15104031bb98SShilei Tian 
15114031bb98SShilei Tian int32_t __tgt_rtl_data_retrieve_async(int32_t device_id, void *hst_ptr,
15124031bb98SShilei Tian                                       void *tgt_ptr, int64_t size,
15134031bb98SShilei Tian                                       __tgt_async_info *async_info_ptr) {
15144031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15154031bb98SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
15164031bb98SShilei Tian 
15174031bb98SShilei Tian   return DeviceRTL.dataRetrieve(device_id, hst_ptr, tgt_ptr, size,
15184031bb98SShilei Tian                                 async_info_ptr);
15194031bb98SShilei Tian }
15204031bb98SShilei Tian 
1521a014fbbcSShilei Tian int32_t __tgt_rtl_data_exchange_async(int32_t src_dev_id, void *src_ptr,
1522a014fbbcSShilei Tian                                       int dst_dev_id, void *dst_ptr,
1523a014fbbcSShilei Tian                                       int64_t size,
15245449fbb5SJohannes Doerfert                                       __tgt_async_info *AsyncInfo) {
1525a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(src_dev_id) && "src_dev_id is invalid");
1526a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(dst_dev_id) && "dst_dev_id is invalid");
15275449fbb5SJohannes Doerfert   assert(AsyncInfo && "AsyncInfo is nullptr");
1528a014fbbcSShilei Tian 
1529a014fbbcSShilei Tian   return DeviceRTL.dataExchange(src_dev_id, src_ptr, dst_dev_id, dst_ptr, size,
15305449fbb5SJohannes Doerfert                                 AsyncInfo);
1531a014fbbcSShilei Tian }
1532a014fbbcSShilei Tian 
1533a014fbbcSShilei Tian int32_t __tgt_rtl_data_exchange(int32_t src_dev_id, void *src_ptr,
1534a014fbbcSShilei Tian                                 int32_t dst_dev_id, void *dst_ptr,
1535a014fbbcSShilei Tian                                 int64_t size) {
1536a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(src_dev_id) && "src_dev_id is invalid");
1537a014fbbcSShilei Tian   assert(DeviceRTL.isValidDeviceId(dst_dev_id) && "dst_dev_id is invalid");
1538a014fbbcSShilei Tian 
15395449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
1540a014fbbcSShilei Tian   const int32_t rc = __tgt_rtl_data_exchange_async(
15415449fbb5SJohannes Doerfert       src_dev_id, src_ptr, dst_dev_id, dst_ptr, size, &AsyncInfo);
1542a014fbbcSShilei Tian   if (rc != OFFLOAD_SUCCESS)
1543a014fbbcSShilei Tian     return OFFLOAD_FAIL;
1544a014fbbcSShilei Tian 
15455449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(src_dev_id, &AsyncInfo);
1546a014fbbcSShilei Tian }
1547a014fbbcSShilei Tian 
15484031bb98SShilei Tian int32_t __tgt_rtl_data_delete(int32_t device_id, void *tgt_ptr) {
15494031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15504031bb98SShilei Tian 
15514031bb98SShilei Tian   return DeviceRTL.dataDelete(device_id, tgt_ptr);
15524031bb98SShilei Tian }
15534031bb98SShilei Tian 
15544031bb98SShilei Tian int32_t __tgt_rtl_run_target_team_region(int32_t device_id, void *tgt_entry_ptr,
15554031bb98SShilei Tian                                          void **tgt_args,
15564031bb98SShilei Tian                                          ptrdiff_t *tgt_offsets,
15574031bb98SShilei Tian                                          int32_t arg_num, int32_t team_num,
15584031bb98SShilei Tian                                          int32_t thread_limit,
15594031bb98SShilei Tian                                          uint64_t loop_tripcount) {
15604031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15614031bb98SShilei Tian 
15625449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
15634031bb98SShilei Tian   const int32_t rc = __tgt_rtl_run_target_team_region_async(
15644031bb98SShilei Tian       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num,
15655449fbb5SJohannes Doerfert       thread_limit, loop_tripcount, &AsyncInfo);
15664031bb98SShilei Tian   if (rc != OFFLOAD_SUCCESS)
15674031bb98SShilei Tian     return OFFLOAD_FAIL;
15684031bb98SShilei Tian 
15695449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
15704031bb98SShilei Tian }
15714031bb98SShilei Tian 
15724031bb98SShilei Tian int32_t __tgt_rtl_run_target_team_region_async(
15734031bb98SShilei Tian     int32_t device_id, void *tgt_entry_ptr, void **tgt_args,
15744031bb98SShilei Tian     ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t team_num,
15754031bb98SShilei Tian     int32_t thread_limit, uint64_t loop_tripcount,
15764031bb98SShilei Tian     __tgt_async_info *async_info_ptr) {
15774031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15784031bb98SShilei Tian 
15794031bb98SShilei Tian   return DeviceRTL.runTargetTeamRegion(
15804031bb98SShilei Tian       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num,
15814031bb98SShilei Tian       thread_limit, loop_tripcount, async_info_ptr);
15824031bb98SShilei Tian }
15833de4cd12SGeorge Rokos 
15843de4cd12SGeorge Rokos int32_t __tgt_rtl_run_target_region(int32_t device_id, void *tgt_entry_ptr,
158532ed2927SShilei Tian                                     void **tgt_args, ptrdiff_t *tgt_offsets,
158603ff643dSShilei Tian                                     int32_t arg_num) {
15874031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
15884031bb98SShilei Tian 
15895449fbb5SJohannes Doerfert   __tgt_async_info AsyncInfo;
15904031bb98SShilei Tian   const int32_t rc = __tgt_rtl_run_target_region_async(
15915449fbb5SJohannes Doerfert       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, &AsyncInfo);
159203ff643dSShilei Tian   if (rc != OFFLOAD_SUCCESS)
159303ff643dSShilei Tian     return OFFLOAD_FAIL;
159403ff643dSShilei Tian 
15955449fbb5SJohannes Doerfert   return __tgt_rtl_synchronize(device_id, &AsyncInfo);
159603ff643dSShilei Tian }
159703ff643dSShilei Tian 
159803ff643dSShilei Tian int32_t __tgt_rtl_run_target_region_async(int32_t device_id,
159903ff643dSShilei Tian                                           void *tgt_entry_ptr, void **tgt_args,
160003ff643dSShilei Tian                                           ptrdiff_t *tgt_offsets,
160132ed2927SShilei Tian                                           int32_t arg_num,
16024031bb98SShilei Tian                                           __tgt_async_info *async_info_ptr) {
16034031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
16044031bb98SShilei Tian 
160503ff643dSShilei Tian   return __tgt_rtl_run_target_team_region_async(
16064031bb98SShilei Tian       device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num,
16074031bb98SShilei Tian       /* team num*/ 1, /* thread_limit */ 1, /* loop_tripcount */ 0,
16084031bb98SShilei Tian       async_info_ptr);
160932ed2927SShilei Tian }
161032ed2927SShilei Tian 
16114031bb98SShilei Tian int32_t __tgt_rtl_synchronize(int32_t device_id,
16124031bb98SShilei Tian                               __tgt_async_info *async_info_ptr) {
16134031bb98SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
16144031bb98SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
16154031bb98SShilei Tian   assert(async_info_ptr->Queue && "async_info_ptr->Queue is nullptr");
161632ed2927SShilei Tian 
16174031bb98SShilei Tian   return DeviceRTL.synchronize(device_id, async_info_ptr);
16183de4cd12SGeorge Rokos }
16193de4cd12SGeorge Rokos 
16202b6f2008SJoseph Huber void __tgt_rtl_set_info_flag(uint32_t NewInfoLevel) {
162159b68490SJoseph Huber   std::atomic<uint32_t> &InfoLevel = getInfoLevelInternal();
16222b6f2008SJoseph Huber   InfoLevel.store(NewInfoLevel);
16232b6f2008SJoseph Huber }
16242b6f2008SJoseph Huber 
1625d2f85d09SJose M Monsalve Diaz void __tgt_rtl_print_device_info(int32_t device_id) {
1626d2f85d09SJose M Monsalve Diaz   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
1627d2f85d09SJose M Monsalve Diaz   DeviceRTL.printDeviceInfo(device_id);
1628d2f85d09SJose M Monsalve Diaz }
1629d2f85d09SJose M Monsalve Diaz 
163029df4ab3SShilei Tian int32_t __tgt_rtl_create_event(int32_t device_id, void **event) {
163129df4ab3SShilei Tian   assert(event && "event is nullptr");
163229df4ab3SShilei Tian   return createEvent(event);
163329df4ab3SShilei Tian }
163429df4ab3SShilei Tian 
163529df4ab3SShilei Tian int32_t __tgt_rtl_record_event(int32_t device_id, void *event_ptr,
163629df4ab3SShilei Tian                                __tgt_async_info *async_info_ptr) {
163729df4ab3SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
163829df4ab3SShilei Tian   assert(async_info_ptr->Queue && "async_info_ptr->Queue is nullptr");
163929df4ab3SShilei Tian   assert(event_ptr && "event_ptr is nullptr");
164029df4ab3SShilei Tian 
164129df4ab3SShilei Tian   return recordEvent(event_ptr, async_info_ptr);
164229df4ab3SShilei Tian }
164329df4ab3SShilei Tian 
164429df4ab3SShilei Tian int32_t __tgt_rtl_wait_event(int32_t device_id, void *event_ptr,
164529df4ab3SShilei Tian                              __tgt_async_info *async_info_ptr) {
164629df4ab3SShilei Tian   assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid");
164729df4ab3SShilei Tian   assert(async_info_ptr && "async_info_ptr is nullptr");
164829df4ab3SShilei Tian   assert(event_ptr && "event is nullptr");
164929df4ab3SShilei Tian 
165029df4ab3SShilei Tian   return DeviceRTL.waitEvent(device_id, async_info_ptr, event_ptr);
165129df4ab3SShilei Tian }
165229df4ab3SShilei Tian 
165329df4ab3SShilei Tian int32_t __tgt_rtl_sync_event(int32_t device_id, void *event_ptr) {
165429df4ab3SShilei Tian   assert(event_ptr && "event is nullptr");
165529df4ab3SShilei Tian 
165629df4ab3SShilei Tian   return syncEvent(event_ptr);
165729df4ab3SShilei Tian }
165829df4ab3SShilei Tian 
165929df4ab3SShilei Tian int32_t __tgt_rtl_destroy_event(int32_t device_id, void *event_ptr) {
166029df4ab3SShilei Tian   assert(event_ptr && "event is nullptr");
166129df4ab3SShilei Tian 
166229df4ab3SShilei Tian   return destroyEvent(event_ptr);
166329df4ab3SShilei Tian }
166429df4ab3SShilei Tian 
16653de4cd12SGeorge Rokos #ifdef __cplusplus
16663de4cd12SGeorge Rokos }
16673de4cd12SGeorge Rokos #endif
1668