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