13de4cd12SGeorge Rokos //===----RTLs/cuda/src/rtl.cpp - Target RTLs Implementation ------- C++ -*-===// 23de4cd12SGeorge Rokos // 357b08b09SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 457b08b09SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 557b08b09SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 63de4cd12SGeorge Rokos // 73de4cd12SGeorge Rokos //===----------------------------------------------------------------------===// 83de4cd12SGeorge Rokos // 93de4cd12SGeorge Rokos // RTL for CUDA machine 103de4cd12SGeorge Rokos // 113de4cd12SGeorge Rokos //===----------------------------------------------------------------------===// 123de4cd12SGeorge Rokos 133de4cd12SGeorge Rokos #include <cassert> 143de4cd12SGeorge Rokos #include <cstddef> 153de4cd12SGeorge Rokos #include <cuda.h> 163de4cd12SGeorge Rokos #include <list> 17a5153dbcSJohannes Doerfert #include <memory> 18feed674dSShilei Tian #include <mutex> 193de4cd12SGeorge Rokos #include <string> 20a15f8589SJoseph Huber #include <unordered_map> 213de4cd12SGeorge Rokos #include <vector> 223de4cd12SGeorge Rokos 23ae95ceebSJoseph Huber #include "Debug.h" 241546d319SGeorge Rokos #include "omptargetplugin.h" 253de4cd12SGeorge Rokos 263de4cd12SGeorge Rokos #define TARGET_NAME CUDA 27ae95ceebSJoseph Huber #define DEBUG_PREFIX "Target " GETNAME(TARGET_NAME) " RTL" 2878769ec4SMichael Kruse 290871d6d5SShilei Tian #include "MemoryManager.h" 300871d6d5SShilei Tian 3178769ec4SMichael Kruse // Utility for retrieving and printing CUDA error string. 32ae95ceebSJoseph Huber #ifdef OMPTARGET_DEBUG 3378769ec4SMichael Kruse #define CUDA_ERR_STRING(err) \ 3478769ec4SMichael Kruse do { \ 35ae95ceebSJoseph Huber if (getDebugLevel() > 0) { \ 3603111e5eSYe Luo const char *errStr = nullptr; \ 3703111e5eSYe Luo CUresult errStr_status = cuGetErrorString(err, &errStr); \ 3803111e5eSYe Luo if (errStr_status == CUDA_ERROR_INVALID_VALUE) \ 392ce16810SJoseph Huber REPORT("Unrecognized CUDA error code: %d\n", err); \ 4003111e5eSYe Luo else if (errStr_status == CUDA_SUCCESS) \ 412ce16810SJoseph Huber REPORT("CUDA error is: %s\n", errStr); \ 4203111e5eSYe Luo else { \ 432ce16810SJoseph Huber REPORT("Unresolved CUDA error code: %d\n", err); \ 442ce16810SJoseph Huber REPORT("Unsuccessful cuGetErrorString return status: %d\n", \ 4503111e5eSYe Luo errStr_status); \ 4603111e5eSYe Luo } \ 472ce16810SJoseph Huber } else { \ 482ce16810SJoseph Huber const char *errStr = nullptr; \ 492ce16810SJoseph Huber CUresult errStr_status = cuGetErrorString(err, &errStr); \ 502ce16810SJoseph Huber if (errStr_status == CUDA_SUCCESS) \ 512ce16810SJoseph Huber REPORT("%s \n", errStr); \ 5278769ec4SMichael Kruse } \ 5378769ec4SMichael Kruse } while (false) 54b305d26bSSergey Dmitriev #else // OMPTARGET_DEBUG 552ce16810SJoseph Huber #define CUDA_ERR_STRING(err) \ 562ce16810SJoseph Huber do { \ 572ce16810SJoseph Huber const char *errStr = nullptr; \ 582ce16810SJoseph Huber CUresult errStr_status = cuGetErrorString(err, &errStr); \ 592ce16810SJoseph Huber if (errStr_status == CUDA_SUCCESS) \ 602ce16810SJoseph Huber REPORT("%s \n", errStr); \ 612ce16810SJoseph Huber } while (false) 62b305d26bSSergey Dmitriev #endif // OMPTARGET_DEBUG 633de4cd12SGeorge Rokos 64d2f85d09SJose M Monsalve Diaz #define BOOL2TEXT(b) ((b) ? "Yes" : "No") 65d2f85d09SJose M Monsalve Diaz 66a81c68aeSShilei Tian #include "elf_common.h" 673de4cd12SGeorge Rokos 683de4cd12SGeorge Rokos /// Keep entries table per device. 693de4cd12SGeorge Rokos struct FuncOrGblEntryTy { 703de4cd12SGeorge Rokos __tgt_target_table Table; 713de4cd12SGeorge Rokos std::vector<__tgt_offload_entry> Entries; 723de4cd12SGeorge Rokos }; 733de4cd12SGeorge Rokos 743de4cd12SGeorge Rokos enum ExecutionModeType { 75c13df8e5SGeorge Rokos SPMD, // constructors, destructors, 76c13df8e5SGeorge Rokos // combined constructs (`teams distribute parallel for [simd]`) 77c13df8e5SGeorge Rokos GENERIC, // everything else 787d576392SJoseph Huber SPMD_GENERIC, // Generic kernel with SPMD execution 793de4cd12SGeorge Rokos NONE 803de4cd12SGeorge Rokos }; 813de4cd12SGeorge Rokos 82dc34b1c9SRon Lieberman /// Use a single entity to encode a kernel and a set of flags. 833de4cd12SGeorge Rokos struct KernelTy { 843de4cd12SGeorge Rokos CUfunction Func; 853de4cd12SGeorge Rokos 863de4cd12SGeorge Rokos // execution mode of kernel 873de4cd12SGeorge Rokos // 0 - SPMD mode (without master warp) 883de4cd12SGeorge Rokos // 1 - Generic mode (with master warp) 897d576392SJoseph Huber // 2 - SPMD mode execution with Generic mode semantics. 903de4cd12SGeorge Rokos int8_t ExecutionMode; 913de4cd12SGeorge Rokos 92aa27cfc1SJohannes Doerfert /// Maximal number of threads per block for this kernel. 93aa27cfc1SJohannes Doerfert int MaxThreadsPerBlock = 0; 94aa27cfc1SJohannes Doerfert 953de4cd12SGeorge Rokos KernelTy(CUfunction _Func, int8_t _ExecutionMode) 963de4cd12SGeorge Rokos : Func(_Func), ExecutionMode(_ExecutionMode) {} 973de4cd12SGeorge Rokos }; 983de4cd12SGeorge Rokos 994c6a098aSKazuaki Ishizaki /// Device environment data 100feed674dSShilei Tian /// Manually sync with the deviceRTL side for now, move to a dedicated header 101feed674dSShilei Tian /// file later. 102e1c7a46dSGuansong Zhang struct omptarget_device_environmentTy { 103e1c7a46dSGuansong Zhang int32_t debug_level; 104ec02c34bSJoseph Huber uint32_t num_devices; 105ec02c34bSJoseph Huber uint32_t device_num; 106f1c821faSJoseph Huber uint64_t dynamic_shared_size; 107e1c7a46dSGuansong Zhang }; 108e1c7a46dSGuansong Zhang 109feed674dSShilei Tian namespace { 110feed674dSShilei Tian bool checkResult(CUresult Err, const char *ErrMsg) { 111feed674dSShilei Tian if (Err == CUDA_SUCCESS) 112feed674dSShilei Tian return true; 113feed674dSShilei Tian 1142ce16810SJoseph Huber REPORT("%s", ErrMsg); 115feed674dSShilei Tian CUDA_ERR_STRING(Err); 116feed674dSShilei Tian return false; 117feed674dSShilei Tian } 1184031bb98SShilei Tian 119a014fbbcSShilei Tian int memcpyDtoD(const void *SrcPtr, void *DstPtr, int64_t Size, 120a014fbbcSShilei Tian CUstream Stream) { 12195f0d1edSJon Chesterfield CUresult Err = 12295f0d1edSJon Chesterfield cuMemcpyDtoDAsync((CUdeviceptr)DstPtr, (CUdeviceptr)SrcPtr, Size, Stream); 123a014fbbcSShilei Tian 124a014fbbcSShilei Tian if (Err != CUDA_SUCCESS) { 12529338459SJoseph Huber DP("Error when copying data from device to device. Pointers: src " 126a014fbbcSShilei Tian "= " DPxMOD ", dst = " DPxMOD ", size = %" PRId64 "\n", 127a014fbbcSShilei Tian DPxPTR(SrcPtr), DPxPTR(DstPtr), Size); 128a014fbbcSShilei Tian CUDA_ERR_STRING(Err); 129a014fbbcSShilei Tian return OFFLOAD_FAIL; 130a014fbbcSShilei Tian } 131a014fbbcSShilei Tian 132a014fbbcSShilei Tian return OFFLOAD_SUCCESS; 133a014fbbcSShilei Tian } 134a014fbbcSShilei Tian 13529df4ab3SShilei Tian int createEvent(void **P) { 13629df4ab3SShilei Tian CUevent Event = nullptr; 13729df4ab3SShilei Tian 13829df4ab3SShilei Tian CUresult Err = cuEventCreate(&Event, CU_EVENT_DEFAULT); 13929df4ab3SShilei Tian if (Err != CUDA_SUCCESS) { 14029df4ab3SShilei Tian DP("Error when creating event event = " DPxMOD "\n", DPxPTR(Event)); 14129df4ab3SShilei Tian CUDA_ERR_STRING(Err); 14229df4ab3SShilei Tian return OFFLOAD_FAIL; 14329df4ab3SShilei Tian } 14429df4ab3SShilei Tian 14529df4ab3SShilei Tian *P = Event; 14629df4ab3SShilei Tian 14729df4ab3SShilei Tian return OFFLOAD_SUCCESS; 14829df4ab3SShilei Tian } 14929df4ab3SShilei Tian 15029df4ab3SShilei Tian int recordEvent(void *EventPtr, __tgt_async_info *AsyncInfo) { 15129df4ab3SShilei Tian CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue); 15229df4ab3SShilei Tian CUevent Event = reinterpret_cast<CUevent>(EventPtr); 15329df4ab3SShilei Tian 15429df4ab3SShilei Tian CUresult Err = cuEventRecord(Event, Stream); 15529df4ab3SShilei Tian if (Err != CUDA_SUCCESS) { 15629df4ab3SShilei Tian DP("Error when recording event. stream = " DPxMOD ", event = " DPxMOD "\n", 15729df4ab3SShilei Tian DPxPTR(Stream), DPxPTR(Event)); 15829df4ab3SShilei Tian CUDA_ERR_STRING(Err); 15929df4ab3SShilei Tian return OFFLOAD_FAIL; 16029df4ab3SShilei Tian } 16129df4ab3SShilei Tian 16229df4ab3SShilei Tian return OFFLOAD_SUCCESS; 16329df4ab3SShilei Tian } 16429df4ab3SShilei Tian 16529df4ab3SShilei Tian int syncEvent(void *EventPtr) { 16629df4ab3SShilei Tian CUevent Event = reinterpret_cast<CUevent>(EventPtr); 16729df4ab3SShilei Tian 16829df4ab3SShilei Tian CUresult Err = cuEventSynchronize(Event); 16929df4ab3SShilei Tian if (Err != CUDA_SUCCESS) { 17029df4ab3SShilei Tian DP("Error when syncing event = " DPxMOD "\n", DPxPTR(Event)); 17129df4ab3SShilei Tian CUDA_ERR_STRING(Err); 17229df4ab3SShilei Tian return OFFLOAD_FAIL; 17329df4ab3SShilei Tian } 17429df4ab3SShilei Tian 17529df4ab3SShilei Tian return OFFLOAD_SUCCESS; 17629df4ab3SShilei Tian } 17729df4ab3SShilei Tian 17829df4ab3SShilei Tian int destroyEvent(void *EventPtr) { 17929df4ab3SShilei Tian CUevent Event = reinterpret_cast<CUevent>(EventPtr); 18029df4ab3SShilei Tian 18129df4ab3SShilei Tian CUresult Err = cuEventDestroy(Event); 18229df4ab3SShilei Tian if (Err != CUDA_SUCCESS) { 18329df4ab3SShilei Tian DP("Error when destroying event = " DPxMOD "\n", DPxPTR(Event)); 18429df4ab3SShilei Tian CUDA_ERR_STRING(Err); 18529df4ab3SShilei Tian return OFFLOAD_FAIL; 18629df4ab3SShilei Tian } 18729df4ab3SShilei Tian 18829df4ab3SShilei Tian return OFFLOAD_SUCCESS; 18929df4ab3SShilei Tian } 19029df4ab3SShilei Tian 1914031bb98SShilei Tian // Structure contains per-device data 1924031bb98SShilei Tian struct DeviceDataTy { 1935272d29eSJohannes Doerfert /// List that contains all the kernels. 1945272d29eSJohannes Doerfert std::list<KernelTy> KernelsList; 1955272d29eSJohannes Doerfert 1964031bb98SShilei Tian std::list<FuncOrGblEntryTy> FuncGblEntries; 1975272d29eSJohannes Doerfert 1984031bb98SShilei Tian CUcontext Context = nullptr; 1994031bb98SShilei Tian // Device properties 2004031bb98SShilei Tian int ThreadsPerBlock = 0; 2014031bb98SShilei Tian int BlocksPerGrid = 0; 2024031bb98SShilei Tian int WarpSize = 0; 2034031bb98SShilei Tian // OpenMP properties 2044031bb98SShilei Tian int NumTeams = 0; 2054031bb98SShilei Tian int NumThreads = 0; 2064031bb98SShilei Tian }; 207feed674dSShilei Tian 208feed674dSShilei Tian class StreamManagerTy { 209feed674dSShilei Tian int NumberOfDevices; 2104031bb98SShilei Tian // The initial size of stream pool 2114031bb98SShilei Tian int EnvNumInitialStreams; 212feed674dSShilei Tian // Per-device stream mutex 213feed674dSShilei Tian std::vector<std::unique_ptr<std::mutex>> StreamMtx; 214feed674dSShilei Tian // Per-device stream Id indicates the next available stream in the pool 215feed674dSShilei Tian std::vector<int> NextStreamId; 216feed674dSShilei Tian // Per-device stream pool 217feed674dSShilei Tian std::vector<std::vector<CUstream>> StreamPool; 2184031bb98SShilei Tian // Reference to per-device data 2194031bb98SShilei Tian std::vector<DeviceDataTy> &DeviceData; 220feed674dSShilei Tian 221feed674dSShilei Tian // If there is no CUstream left in the pool, we will resize the pool to 222feed674dSShilei Tian // allocate more CUstream. This function should be called with device mutex, 223feed674dSShilei Tian // and we do not resize to smaller one. 224feed674dSShilei Tian void resizeStreamPool(const int DeviceId, const size_t NewSize) { 225feed674dSShilei Tian std::vector<CUstream> &Pool = StreamPool[DeviceId]; 226feed674dSShilei Tian const size_t CurrentSize = Pool.size(); 227feed674dSShilei Tian assert(NewSize > CurrentSize && "new size is not larger than current size"); 228feed674dSShilei Tian 2294031bb98SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 2304031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) { 231feed674dSShilei Tian // We will return if cannot switch to the right context in case of 232feed674dSShilei Tian // creating bunch of streams that are not corresponding to the right 233feed674dSShilei Tian // device. The offloading will fail later because selected CUstream is 234feed674dSShilei Tian // nullptr. 235feed674dSShilei Tian return; 236feed674dSShilei Tian } 237feed674dSShilei Tian 2384031bb98SShilei Tian Pool.resize(NewSize, nullptr); 2394031bb98SShilei Tian 240feed674dSShilei Tian for (size_t I = CurrentSize; I < NewSize; ++I) { 2414031bb98SShilei Tian checkResult(cuStreamCreate(&Pool[I], CU_STREAM_NON_BLOCKING), 2424031bb98SShilei Tian "Error returned from cuStreamCreate\n"); 243feed674dSShilei Tian } 244feed674dSShilei Tian } 245feed674dSShilei Tian 246feed674dSShilei Tian public: 2474031bb98SShilei Tian StreamManagerTy(const int NumberOfDevices, 2484031bb98SShilei Tian std::vector<DeviceDataTy> &DeviceData) 2494031bb98SShilei Tian : NumberOfDevices(NumberOfDevices), EnvNumInitialStreams(32), 2504031bb98SShilei Tian DeviceData(DeviceData) { 251feed674dSShilei Tian StreamPool.resize(NumberOfDevices); 252feed674dSShilei Tian NextStreamId.resize(NumberOfDevices); 253feed674dSShilei Tian StreamMtx.resize(NumberOfDevices); 254feed674dSShilei Tian 2554031bb98SShilei Tian if (const char *EnvStr = getenv("LIBOMPTARGET_NUM_INITIAL_STREAMS")) 2564031bb98SShilei Tian EnvNumInitialStreams = std::stoi(EnvStr); 257feed674dSShilei Tian 258feed674dSShilei Tian // Initialize the next stream id 259feed674dSShilei Tian std::fill(NextStreamId.begin(), NextStreamId.end(), 0); 260feed674dSShilei Tian 261feed674dSShilei Tian // Initialize stream mutex 262feed674dSShilei Tian for (std::unique_ptr<std::mutex> &Ptr : StreamMtx) 263feed674dSShilei Tian Ptr = std::make_unique<std::mutex>(); 264feed674dSShilei Tian } 265feed674dSShilei Tian 266feed674dSShilei Tian ~StreamManagerTy() { 267feed674dSShilei Tian // Destroy streams 268feed674dSShilei Tian for (int I = 0; I < NumberOfDevices; ++I) { 2694031bb98SShilei Tian checkResult(cuCtxSetCurrent(DeviceData[I].Context), 2704031bb98SShilei Tian "Error returned from cuCtxSetCurrent\n"); 271feed674dSShilei Tian 272feed674dSShilei Tian for (CUstream &S : StreamPool[I]) { 2734031bb98SShilei Tian if (S) 27495f0d1edSJon Chesterfield checkResult(cuStreamDestroy(S), 27595f0d1edSJon Chesterfield "Error returned from cuStreamDestroy\n"); 276feed674dSShilei Tian } 277feed674dSShilei Tian } 278feed674dSShilei Tian } 279feed674dSShilei Tian 280feed674dSShilei Tian // Get a CUstream from pool. Per-device next stream id always points to the 281feed674dSShilei Tian // next available CUstream. That means, CUstreams [0, id-1] have been 282feed674dSShilei Tian // assigned, and [id,] are still available. If there is no CUstream left, we 283feed674dSShilei Tian // will ask more CUstreams from CUDA RT. Each time a CUstream is assigned, 284feed674dSShilei Tian // the id will increase one. 285feed674dSShilei Tian // xxxxxs+++++++++ 286feed674dSShilei Tian // ^ 287feed674dSShilei Tian // id 288feed674dSShilei Tian // After assignment, the pool becomes the following and s is assigned. 289feed674dSShilei Tian // xxxxxs+++++++++ 290feed674dSShilei Tian // ^ 291feed674dSShilei Tian // id 292feed674dSShilei Tian CUstream getStream(const int DeviceId) { 293feed674dSShilei Tian const std::lock_guard<std::mutex> Lock(*StreamMtx[DeviceId]); 294feed674dSShilei Tian int &Id = NextStreamId[DeviceId]; 295feed674dSShilei Tian // No CUstream left in the pool, we need to request from CUDA RT 296ad94fce8SJohannes Doerfert if (Id == static_cast<int>(StreamPool[DeviceId].size())) { 297feed674dSShilei Tian // By default we double the stream pool every time 298feed674dSShilei Tian resizeStreamPool(DeviceId, Id * 2); 299feed674dSShilei Tian } 300feed674dSShilei Tian return StreamPool[DeviceId][Id++]; 301feed674dSShilei Tian } 302feed674dSShilei Tian 303feed674dSShilei Tian // Return a CUstream back to pool. As mentioned above, per-device next 304feed674dSShilei Tian // stream is always points to the next available CUstream, so when we return 305feed674dSShilei Tian // a CUstream, we need to first decrease the id, and then copy the CUstream 306feed674dSShilei Tian // back. 307feed674dSShilei Tian // It is worth noting that, the order of streams return might be different 308feed674dSShilei Tian // from that they're assigned, that saying, at some point, there might be 309feed674dSShilei Tian // two identical CUstreams. 310feed674dSShilei Tian // xxax+a+++++ 311feed674dSShilei Tian // ^ 312feed674dSShilei Tian // id 313feed674dSShilei Tian // However, it doesn't matter, because they're always on the two sides of 314feed674dSShilei Tian // id. The left one will in the end be overwritten by another CUstream. 315feed674dSShilei Tian // Therefore, after several execution, the order of pool might be different 316feed674dSShilei Tian // from its initial state. 317feed674dSShilei Tian void returnStream(const int DeviceId, CUstream Stream) { 318feed674dSShilei Tian const std::lock_guard<std::mutex> Lock(*StreamMtx[DeviceId]); 319feed674dSShilei Tian int &Id = NextStreamId[DeviceId]; 320feed674dSShilei Tian assert(Id > 0 && "Wrong stream ID"); 321feed674dSShilei Tian StreamPool[DeviceId][--Id] = Stream; 322feed674dSShilei Tian } 323feed674dSShilei Tian 3244031bb98SShilei Tian bool initializeDeviceStreamPool(const int DeviceId) { 3254031bb98SShilei Tian assert(StreamPool[DeviceId].empty() && "stream pool has been initialized"); 3264031bb98SShilei Tian 3274031bb98SShilei Tian resizeStreamPool(DeviceId, EnvNumInitialStreams); 3284031bb98SShilei Tian 3294031bb98SShilei Tian // Check the size of stream pool 330ad94fce8SJohannes Doerfert if (static_cast<int>(StreamPool[DeviceId].size()) != EnvNumInitialStreams) 3314031bb98SShilei Tian return false; 3324031bb98SShilei Tian 3334031bb98SShilei Tian // Check whether each stream is valid 3344031bb98SShilei Tian for (CUstream &S : StreamPool[DeviceId]) 3354031bb98SShilei Tian if (!S) 3364031bb98SShilei Tian return false; 3374031bb98SShilei Tian 3384031bb98SShilei Tian return true; 339feed674dSShilei Tian } 340feed674dSShilei Tian }; 341feed674dSShilei Tian 3424031bb98SShilei Tian class DeviceRTLTy { 3433de4cd12SGeorge Rokos int NumberOfDevices; 3444031bb98SShilei Tian // OpenMP environment properties 3453de4cd12SGeorge Rokos int EnvNumTeams; 3463de4cd12SGeorge Rokos int EnvTeamLimit; 347422adaa8SJoseph Huber int EnvTeamThreadLimit; 3484031bb98SShilei Tian // OpenMP requires flags 349c5fe030cSGheorghe-Teodor Bercea int64_t RequiresFlags; 350f1c821faSJoseph Huber // Amount of dynamic shared memory to use at launch. 351f1c821faSJoseph Huber uint64_t DynamicMemorySize; 352c5fe030cSGheorghe-Teodor Bercea 3534031bb98SShilei Tian static constexpr const int HardTeamLimit = 1U << 16U; // 64k 3544031bb98SShilei Tian static constexpr const int HardThreadLimit = 1024; 3554031bb98SShilei Tian static constexpr const int DefaultNumTeams = 128; 3564031bb98SShilei Tian static constexpr const int DefaultNumThreads = 128; 3573de4cd12SGeorge Rokos 3584031bb98SShilei Tian std::unique_ptr<StreamManagerTy> StreamManager; 3594031bb98SShilei Tian std::vector<DeviceDataTy> DeviceData; 3604031bb98SShilei Tian std::vector<CUmodule> Modules; 361feed674dSShilei Tian 3620871d6d5SShilei Tian /// A class responsible for interacting with device native runtime library to 3630871d6d5SShilei Tian /// allocate and free memory. 3640871d6d5SShilei Tian class CUDADeviceAllocatorTy : public DeviceAllocatorTy { 3650871d6d5SShilei Tian const int DeviceId; 3660871d6d5SShilei Tian const std::vector<DeviceDataTy> &DeviceData; 367a15f8589SJoseph Huber std::unordered_map<void *, TargetAllocTy> HostPinnedAllocs; 3680871d6d5SShilei Tian 3690871d6d5SShilei Tian public: 3700871d6d5SShilei Tian CUDADeviceAllocatorTy(int DeviceId, std::vector<DeviceDataTy> &DeviceData) 3710871d6d5SShilei Tian : DeviceId(DeviceId), DeviceData(DeviceData) {} 3720871d6d5SShilei Tian 373a15f8589SJoseph Huber void *allocate(size_t Size, void *, TargetAllocTy Kind) override { 3740871d6d5SShilei Tian if (Size == 0) 3750871d6d5SShilei Tian return nullptr; 3760871d6d5SShilei Tian 3770871d6d5SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 3780871d6d5SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 3790871d6d5SShilei Tian return nullptr; 3800871d6d5SShilei Tian 381a15f8589SJoseph Huber void *MemAlloc = nullptr; 382a15f8589SJoseph Huber switch (Kind) { 383a15f8589SJoseph Huber case TARGET_ALLOC_DEFAULT: 384a15f8589SJoseph Huber case TARGET_ALLOC_DEVICE: 3850871d6d5SShilei Tian CUdeviceptr DevicePtr; 38695f0d1edSJon Chesterfield Err = cuMemAlloc(&DevicePtr, Size); 387a15f8589SJoseph Huber MemAlloc = (void *)DevicePtr; 38895f0d1edSJon Chesterfield if (!checkResult(Err, "Error returned from cuMemAlloc\n")) 3890871d6d5SShilei Tian return nullptr; 390a15f8589SJoseph Huber break; 391a15f8589SJoseph Huber case TARGET_ALLOC_HOST: 392a15f8589SJoseph Huber void *HostPtr; 393a15f8589SJoseph Huber Err = cuMemAllocHost(&HostPtr, Size); 394a15f8589SJoseph Huber MemAlloc = HostPtr; 395a15f8589SJoseph Huber if (!checkResult(Err, "Error returned from cuMemAllocHost\n")) 396a15f8589SJoseph Huber return nullptr; 397a15f8589SJoseph Huber HostPinnedAllocs[MemAlloc] = Kind; 398a15f8589SJoseph Huber break; 399a15f8589SJoseph Huber case TARGET_ALLOC_SHARED: 400a15f8589SJoseph Huber CUdeviceptr SharedPtr; 401a15f8589SJoseph Huber Err = cuMemAllocManaged(&SharedPtr, Size, CU_MEM_ATTACH_GLOBAL); 402a15f8589SJoseph Huber MemAlloc = (void *)SharedPtr; 403a15f8589SJoseph Huber if (!checkResult(Err, "Error returned from cuMemAllocManaged\n")) 404a15f8589SJoseph Huber return nullptr; 405a15f8589SJoseph Huber break; 406a15f8589SJoseph Huber } 4070871d6d5SShilei Tian 408a15f8589SJoseph Huber return MemAlloc; 4090871d6d5SShilei Tian } 4100871d6d5SShilei Tian 4110871d6d5SShilei Tian int free(void *TgtPtr) override { 4120871d6d5SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 4130871d6d5SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 4140871d6d5SShilei Tian return OFFLOAD_FAIL; 4150871d6d5SShilei Tian 416a15f8589SJoseph Huber // Host pinned memory must be freed differently. 417a15f8589SJoseph Huber TargetAllocTy Kind = 418a15f8589SJoseph Huber (HostPinnedAllocs.find(TgtPtr) == HostPinnedAllocs.end()) 419a15f8589SJoseph Huber ? TARGET_ALLOC_DEFAULT 420a15f8589SJoseph Huber : TARGET_ALLOC_HOST; 421a15f8589SJoseph Huber switch (Kind) { 422a15f8589SJoseph Huber case TARGET_ALLOC_DEFAULT: 423a15f8589SJoseph Huber case TARGET_ALLOC_DEVICE: 424a15f8589SJoseph Huber case TARGET_ALLOC_SHARED: 42595f0d1edSJon Chesterfield Err = cuMemFree((CUdeviceptr)TgtPtr); 42695f0d1edSJon Chesterfield if (!checkResult(Err, "Error returned from cuMemFree\n")) 4270871d6d5SShilei Tian return OFFLOAD_FAIL; 428a15f8589SJoseph Huber break; 429a15f8589SJoseph Huber case TARGET_ALLOC_HOST: 430a15f8589SJoseph Huber Err = cuMemFreeHost(TgtPtr); 431a15f8589SJoseph Huber if (!checkResult(Err, "Error returned from cuMemFreeHost\n")) 432a15f8589SJoseph Huber return OFFLOAD_FAIL; 433a15f8589SJoseph Huber break; 434a15f8589SJoseph Huber } 4350871d6d5SShilei Tian 4360871d6d5SShilei Tian return OFFLOAD_SUCCESS; 4370871d6d5SShilei Tian } 4380871d6d5SShilei Tian }; 4390871d6d5SShilei Tian 4400871d6d5SShilei Tian /// A vector of device allocators 4410871d6d5SShilei Tian std::vector<CUDADeviceAllocatorTy> DeviceAllocators; 4420871d6d5SShilei Tian 4430871d6d5SShilei Tian /// A vector of memory managers. Since the memory manager is non-copyable and 4440871d6d5SShilei Tian // non-removable, we wrap them into std::unique_ptr. 4450871d6d5SShilei Tian std::vector<std::unique_ptr<MemoryManagerTy>> MemoryManagers; 4460871d6d5SShilei Tian 4470871d6d5SShilei Tian /// Whether use memory manager 4480871d6d5SShilei Tian bool UseMemoryManager = true; 4490871d6d5SShilei Tian 4503de4cd12SGeorge Rokos // Record entry point associated with device 4514031bb98SShilei Tian void addOffloadEntry(const int DeviceId, const __tgt_offload_entry entry) { 4524031bb98SShilei Tian FuncOrGblEntryTy &E = DeviceData[DeviceId].FuncGblEntries.back(); 4533de4cd12SGeorge Rokos E.Entries.push_back(entry); 4543de4cd12SGeorge Rokos } 4553de4cd12SGeorge Rokos 456ae209397SJoseph Huber // Return a pointer to the entry associated with the pointer 457ae209397SJoseph Huber const __tgt_offload_entry *getOffloadEntry(const int DeviceId, 458ae209397SJoseph Huber const void *Addr) const { 4594031bb98SShilei Tian for (const __tgt_offload_entry &Itr : 4604031bb98SShilei Tian DeviceData[DeviceId].FuncGblEntries.back().Entries) 4614031bb98SShilei Tian if (Itr.addr == Addr) 462ae209397SJoseph Huber return &Itr; 4633de4cd12SGeorge Rokos 464ae209397SJoseph Huber return nullptr; 4653de4cd12SGeorge Rokos } 4663de4cd12SGeorge Rokos 4673de4cd12SGeorge Rokos // Return the pointer to the target entries table 4684031bb98SShilei Tian __tgt_target_table *getOffloadEntriesTable(const int DeviceId) { 4694031bb98SShilei Tian FuncOrGblEntryTy &E = DeviceData[DeviceId].FuncGblEntries.back(); 4703de4cd12SGeorge Rokos 4714031bb98SShilei Tian if (E.Entries.empty()) 4724031bb98SShilei Tian return nullptr; 4733de4cd12SGeorge Rokos 4743de4cd12SGeorge Rokos // Update table info according to the entries and return the pointer 4754031bb98SShilei Tian E.Table.EntriesBegin = E.Entries.data(); 4764031bb98SShilei Tian E.Table.EntriesEnd = E.Entries.data() + E.Entries.size(); 4773de4cd12SGeorge Rokos 4783de4cd12SGeorge Rokos return &E.Table; 4793de4cd12SGeorge Rokos } 4803de4cd12SGeorge Rokos 4813de4cd12SGeorge Rokos // Clear entries table for a device 4824031bb98SShilei Tian void clearOffloadEntriesTable(const int DeviceId) { 4834031bb98SShilei Tian DeviceData[DeviceId].FuncGblEntries.emplace_back(); 4844031bb98SShilei Tian FuncOrGblEntryTy &E = DeviceData[DeviceId].FuncGblEntries.back(); 4853de4cd12SGeorge Rokos E.Entries.clear(); 4864031bb98SShilei Tian E.Table.EntriesBegin = E.Table.EntriesEnd = nullptr; 4873de4cd12SGeorge Rokos } 4883de4cd12SGeorge Rokos 4895449fbb5SJohannes Doerfert CUstream getStream(const int DeviceId, __tgt_async_info *AsyncInfo) const { 4905449fbb5SJohannes Doerfert assert(AsyncInfo && "AsyncInfo is nullptr"); 4914031bb98SShilei Tian 4925449fbb5SJohannes Doerfert if (!AsyncInfo->Queue) 4935449fbb5SJohannes Doerfert AsyncInfo->Queue = StreamManager->getStream(DeviceId); 4944031bb98SShilei Tian 4955449fbb5SJohannes Doerfert return reinterpret_cast<CUstream>(AsyncInfo->Queue); 496b305d26bSSergey Dmitriev } 4974031bb98SShilei Tian 4984031bb98SShilei Tian public: 4994031bb98SShilei Tian // This class should not be copied 5004031bb98SShilei Tian DeviceRTLTy(const DeviceRTLTy &) = delete; 5014031bb98SShilei Tian DeviceRTLTy(DeviceRTLTy &&) = delete; 5024031bb98SShilei Tian 5034031bb98SShilei Tian DeviceRTLTy() 5044031bb98SShilei Tian : NumberOfDevices(0), EnvNumTeams(-1), EnvTeamLimit(-1), 505f1c821faSJoseph Huber EnvTeamThreadLimit(-1), RequiresFlags(OMP_REQ_UNDEFINED), 506f1c821faSJoseph Huber DynamicMemorySize(0) { 507b305d26bSSergey Dmitriev 5083de4cd12SGeorge Rokos DP("Start initializing CUDA\n"); 5093de4cd12SGeorge Rokos 5104031bb98SShilei Tian CUresult Err = cuInit(0); 5117baff00eSJon Chesterfield if (Err == CUDA_ERROR_INVALID_HANDLE) { 5127baff00eSJon Chesterfield // Can't call cuGetErrorString if dlsym failed 5137baff00eSJon Chesterfield DP("Failed to load CUDA shared library\n"); 5147baff00eSJon Chesterfield return; 5157baff00eSJon Chesterfield } 5164031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuInit\n")) { 5173de4cd12SGeorge Rokos return; 5183de4cd12SGeorge Rokos } 5193de4cd12SGeorge Rokos 5204031bb98SShilei Tian Err = cuDeviceGetCount(&NumberOfDevices); 5214031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuDeviceGetCount\n")) 5223de4cd12SGeorge Rokos return; 5233de4cd12SGeorge Rokos 5243de4cd12SGeorge Rokos if (NumberOfDevices == 0) { 5253de4cd12SGeorge Rokos DP("There are no devices supporting CUDA.\n"); 5263de4cd12SGeorge Rokos return; 5273de4cd12SGeorge Rokos } 5283de4cd12SGeorge Rokos 5294031bb98SShilei Tian DeviceData.resize(NumberOfDevices); 5303de4cd12SGeorge Rokos 5313de4cd12SGeorge Rokos // Get environment variables regarding teams 5324031bb98SShilei Tian if (const char *EnvStr = getenv("OMP_TEAM_LIMIT")) { 5333de4cd12SGeorge Rokos // OMP_TEAM_LIMIT has been set 5344031bb98SShilei Tian EnvTeamLimit = std::stoi(EnvStr); 5353de4cd12SGeorge Rokos DP("Parsed OMP_TEAM_LIMIT=%d\n", EnvTeamLimit); 5363de4cd12SGeorge Rokos } 537422adaa8SJoseph Huber if (const char *EnvStr = getenv("OMP_TEAMS_THREAD_LIMIT")) { 538422adaa8SJoseph Huber // OMP_TEAMS_THREAD_LIMIT has been set 539422adaa8SJoseph Huber EnvTeamThreadLimit = std::stoi(EnvStr); 540422adaa8SJoseph Huber DP("Parsed OMP_TEAMS_THREAD_LIMIT=%d\n", EnvTeamThreadLimit); 541422adaa8SJoseph Huber } 5424031bb98SShilei Tian if (const char *EnvStr = getenv("OMP_NUM_TEAMS")) { 5433de4cd12SGeorge Rokos // OMP_NUM_TEAMS has been set 5444031bb98SShilei Tian EnvNumTeams = std::stoi(EnvStr); 5453de4cd12SGeorge Rokos DP("Parsed OMP_NUM_TEAMS=%d\n", EnvNumTeams); 5463de4cd12SGeorge Rokos } 547f1c821faSJoseph Huber if (const char *EnvStr = getenv("LIBOMPTARGET_SHARED_MEMORY_SIZE")) { 548f1c821faSJoseph Huber // LIBOMPTARGET_SHARED_MEMORY_SIZE has been set 549f1c821faSJoseph Huber DynamicMemorySize = std::stoi(EnvStr); 550*49e976c9SShilei Tian DP("Parsed LIBOMPTARGET_SHARED_MEMORY_SIZE = %" PRIu64 "\n", 551*49e976c9SShilei Tian DynamicMemorySize); 552f1c821faSJoseph Huber } 553c5fe030cSGheorghe-Teodor Bercea 554feed674dSShilei Tian StreamManager = 5554031bb98SShilei Tian std::make_unique<StreamManagerTy>(NumberOfDevices, DeviceData); 5560871d6d5SShilei Tian 5570871d6d5SShilei Tian for (int I = 0; I < NumberOfDevices; ++I) 5580871d6d5SShilei Tian DeviceAllocators.emplace_back(I, DeviceData); 5590871d6d5SShilei Tian 5600871d6d5SShilei Tian // Get the size threshold from environment variable 5610871d6d5SShilei Tian std::pair<size_t, bool> Res = MemoryManagerTy::getSizeThresholdFromEnv(); 5620871d6d5SShilei Tian UseMemoryManager = Res.second; 5630871d6d5SShilei Tian size_t MemoryManagerThreshold = Res.first; 5640871d6d5SShilei Tian 5650871d6d5SShilei Tian if (UseMemoryManager) 5660871d6d5SShilei Tian for (int I = 0; I < NumberOfDevices; ++I) 5670871d6d5SShilei Tian MemoryManagers.emplace_back(std::make_unique<MemoryManagerTy>( 5680871d6d5SShilei Tian DeviceAllocators[I], MemoryManagerThreshold)); 5693de4cd12SGeorge Rokos } 5703de4cd12SGeorge Rokos 5714031bb98SShilei Tian ~DeviceRTLTy() { 5720871d6d5SShilei Tian // We first destruct memory managers in case that its dependent data are 5730871d6d5SShilei Tian // destroyed before it. 5740871d6d5SShilei Tian for (auto &M : MemoryManagers) 5750871d6d5SShilei Tian M.release(); 5760871d6d5SShilei Tian 577feed674dSShilei Tian StreamManager = nullptr; 578feed674dSShilei Tian 5794031bb98SShilei Tian for (CUmodule &M : Modules) 5804031bb98SShilei Tian // Close module 5814031bb98SShilei Tian if (M) 5824031bb98SShilei Tian checkResult(cuModuleUnload(M), "Error returned from cuModuleUnload\n"); 5834031bb98SShilei Tian 5844031bb98SShilei Tian for (DeviceDataTy &D : DeviceData) { 5854031bb98SShilei Tian // Destroy context 586c5348aecSYe Luo if (D.Context) { 587c5348aecSYe Luo checkResult(cuCtxSetCurrent(D.Context), 588c5348aecSYe Luo "Error returned from cuCtxSetCurrent\n"); 589c5348aecSYe Luo CUdevice Device; 590c5348aecSYe Luo checkResult(cuCtxGetDevice(&Device), 591c5348aecSYe Luo "Error returned from cuCtxGetDevice\n"); 59295f0d1edSJon Chesterfield checkResult(cuDevicePrimaryCtxRelease(Device), 59395f0d1edSJon Chesterfield "Error returned from cuDevicePrimaryCtxRelease\n"); 594c5348aecSYe Luo } 5953de4cd12SGeorge Rokos } 5963de4cd12SGeorge Rokos } 5973de4cd12SGeorge Rokos 5984031bb98SShilei Tian // Check whether a given DeviceId is valid 5994031bb98SShilei Tian bool isValidDeviceId(const int DeviceId) const { 6004031bb98SShilei Tian return DeviceId >= 0 && DeviceId < NumberOfDevices; 60132ed2927SShilei Tian } 60232ed2927SShilei Tian 603cb038927SShilei Tian int getNumOfDevices() const { return NumberOfDevices; } 6044031bb98SShilei Tian 6054031bb98SShilei Tian void setRequiresFlag(const int64_t Flags) { this->RequiresFlags = Flags; } 6064031bb98SShilei Tian 6074031bb98SShilei Tian int initDevice(const int DeviceId) { 6084031bb98SShilei Tian CUdevice Device; 6094031bb98SShilei Tian 6104031bb98SShilei Tian DP("Getting device %d\n", DeviceId); 6114031bb98SShilei Tian CUresult Err = cuDeviceGet(&Device, DeviceId); 6124031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuDeviceGet\n")) 61332ed2927SShilei Tian return OFFLOAD_FAIL; 6143de4cd12SGeorge Rokos 615c5348aecSYe Luo // Query the current flags of the primary context and set its flags if 616c5348aecSYe Luo // it is inactive 617c5348aecSYe Luo unsigned int FormerPrimaryCtxFlags = 0; 618c5348aecSYe Luo int FormerPrimaryCtxIsActive = 0; 619c5348aecSYe Luo Err = cuDevicePrimaryCtxGetState(Device, &FormerPrimaryCtxFlags, 620c5348aecSYe Luo &FormerPrimaryCtxIsActive); 621c5348aecSYe Luo if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxGetState\n")) 622c5348aecSYe Luo return OFFLOAD_FAIL; 623c5348aecSYe Luo 624c5348aecSYe Luo if (FormerPrimaryCtxIsActive) { 625c5348aecSYe Luo DP("The primary context is active, no change to its flags\n"); 626c5348aecSYe Luo if ((FormerPrimaryCtxFlags & CU_CTX_SCHED_MASK) != 627c5348aecSYe Luo CU_CTX_SCHED_BLOCKING_SYNC) 628c5348aecSYe Luo DP("Warning the current flags are not CU_CTX_SCHED_BLOCKING_SYNC\n"); 629c5348aecSYe Luo } else { 630c5348aecSYe Luo DP("The primary context is inactive, set its flags to " 631c5348aecSYe Luo "CU_CTX_SCHED_BLOCKING_SYNC\n"); 63295f0d1edSJon Chesterfield Err = cuDevicePrimaryCtxSetFlags(Device, CU_CTX_SCHED_BLOCKING_SYNC); 63395f0d1edSJon Chesterfield if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxSetFlags\n")) 634c5348aecSYe Luo return OFFLOAD_FAIL; 635c5348aecSYe Luo } 636c5348aecSYe Luo 637c5348aecSYe Luo // Retain the per device primary context and save it to use whenever this 638c5348aecSYe Luo // device is selected. 639c5348aecSYe Luo Err = cuDevicePrimaryCtxRetain(&DeviceData[DeviceId].Context, Device); 640c5348aecSYe Luo if (!checkResult(Err, "Error returned from cuDevicePrimaryCtxRetain\n")) 6413de4cd12SGeorge Rokos return OFFLOAD_FAIL; 6423de4cd12SGeorge Rokos 6434031bb98SShilei Tian Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 6444031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 6454031bb98SShilei Tian return OFFLOAD_FAIL; 646a5153dbcSJohannes Doerfert 647feed674dSShilei Tian // Initialize stream pool 6484031bb98SShilei Tian if (!StreamManager->initializeDeviceStreamPool(DeviceId)) 6494031bb98SShilei Tian return OFFLOAD_FAIL; 650a5153dbcSJohannes Doerfert 651bb51d398SJonas Hahnfeld // Query attributes to determine number of threads/block and blocks/grid. 6524031bb98SShilei Tian int MaxGridDimX; 6534031bb98SShilei Tian Err = cuDeviceGetAttribute(&MaxGridDimX, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X, 6544031bb98SShilei Tian Device); 6554031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 6564031bb98SShilei Tian DP("Error getting max grid dimension, use default value %d\n", 6574031bb98SShilei Tian DeviceRTLTy::DefaultNumTeams); 6584031bb98SShilei Tian DeviceData[DeviceId].BlocksPerGrid = DeviceRTLTy::DefaultNumTeams; 6594031bb98SShilei Tian } else if (MaxGridDimX <= DeviceRTLTy::HardTeamLimit) { 6604031bb98SShilei Tian DP("Using %d CUDA blocks per grid\n", MaxGridDimX); 6614031bb98SShilei Tian DeviceData[DeviceId].BlocksPerGrid = MaxGridDimX; 6623de4cd12SGeorge Rokos } else { 6633de4cd12SGeorge Rokos DP("Max CUDA blocks per grid %d exceeds the hard team limit %d, capping " 664bb51d398SJonas Hahnfeld "at the hard limit\n", 6654031bb98SShilei Tian MaxGridDimX, DeviceRTLTy::HardTeamLimit); 6664031bb98SShilei Tian DeviceData[DeviceId].BlocksPerGrid = DeviceRTLTy::HardTeamLimit; 6673de4cd12SGeorge Rokos } 6683de4cd12SGeorge Rokos 669bb51d398SJonas Hahnfeld // We are only exploiting threads along the x axis. 6704031bb98SShilei Tian int MaxBlockDimX; 6714031bb98SShilei Tian Err = cuDeviceGetAttribute(&MaxBlockDimX, 6724031bb98SShilei Tian CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X, Device); 6734031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 6744031bb98SShilei Tian DP("Error getting max block dimension, use default value %d\n", 6754031bb98SShilei Tian DeviceRTLTy::DefaultNumThreads); 6764031bb98SShilei Tian DeviceData[DeviceId].ThreadsPerBlock = DeviceRTLTy::DefaultNumThreads; 677422adaa8SJoseph Huber } else { 6784031bb98SShilei Tian DP("Using %d CUDA threads per block\n", MaxBlockDimX); 6794031bb98SShilei Tian DeviceData[DeviceId].ThreadsPerBlock = MaxBlockDimX; 680422adaa8SJoseph Huber 681422adaa8SJoseph Huber if (EnvTeamThreadLimit > 0 && 682422adaa8SJoseph Huber DeviceData[DeviceId].ThreadsPerBlock > EnvTeamThreadLimit) { 683422adaa8SJoseph Huber DP("Max CUDA threads per block %d exceeds the thread limit %d set by " 684422adaa8SJoseph Huber "OMP_TEAMS_THREAD_LIMIT, capping at the limit\n", 685422adaa8SJoseph Huber DeviceData[DeviceId].ThreadsPerBlock, EnvTeamThreadLimit); 686422adaa8SJoseph Huber DeviceData[DeviceId].ThreadsPerBlock = EnvTeamThreadLimit; 687422adaa8SJoseph Huber } 688422adaa8SJoseph Huber if (DeviceData[DeviceId].ThreadsPerBlock > DeviceRTLTy::HardThreadLimit) { 6894031bb98SShilei Tian DP("Max CUDA threads per block %d exceeds the hard thread limit %d, " 6904031bb98SShilei Tian "capping at the hard limit\n", 691422adaa8SJoseph Huber DeviceData[DeviceId].ThreadsPerBlock, DeviceRTLTy::HardThreadLimit); 6924031bb98SShilei Tian DeviceData[DeviceId].ThreadsPerBlock = DeviceRTLTy::HardThreadLimit; 6933de4cd12SGeorge Rokos } 694422adaa8SJoseph Huber } 6953de4cd12SGeorge Rokos 6964031bb98SShilei Tian // Get and set warp size 6974031bb98SShilei Tian int WarpSize; 6984031bb98SShilei Tian Err = 6994031bb98SShilei Tian cuDeviceGetAttribute(&WarpSize, CU_DEVICE_ATTRIBUTE_WARP_SIZE, Device); 7004031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 7014031bb98SShilei Tian DP("Error getting warp size, assume default value 32\n"); 7024031bb98SShilei Tian DeviceData[DeviceId].WarpSize = 32; 703bb51d398SJonas Hahnfeld } else { 7044031bb98SShilei Tian DP("Using warp size %d\n", WarpSize); 7054031bb98SShilei Tian DeviceData[DeviceId].WarpSize = WarpSize; 7063de4cd12SGeorge Rokos } 7073de4cd12SGeorge Rokos 7083de4cd12SGeorge Rokos // Adjust teams to the env variables 7094031bb98SShilei Tian if (EnvTeamLimit > 0 && DeviceData[DeviceId].BlocksPerGrid > EnvTeamLimit) { 7103de4cd12SGeorge Rokos DP("Capping max CUDA blocks per grid to OMP_TEAM_LIMIT=%d\n", 7114031bb98SShilei Tian EnvTeamLimit); 7124031bb98SShilei Tian DeviceData[DeviceId].BlocksPerGrid = EnvTeamLimit; 7133de4cd12SGeorge Rokos } 7143de4cd12SGeorge Rokos 7153817ba13SJoseph Huber size_t StackLimit; 7163817ba13SJoseph Huber size_t HeapLimit; 7173817ba13SJoseph Huber if (const char *EnvStr = getenv("LIBOMPTARGET_STACK_SIZE")) { 7183817ba13SJoseph Huber StackLimit = std::stol(EnvStr); 7193817ba13SJoseph Huber if (cuCtxSetLimit(CU_LIMIT_STACK_SIZE, StackLimit) != CUDA_SUCCESS) 7203817ba13SJoseph Huber return OFFLOAD_FAIL; 7213817ba13SJoseph Huber } else { 7223817ba13SJoseph Huber if (cuCtxGetLimit(&StackLimit, CU_LIMIT_STACK_SIZE) != CUDA_SUCCESS) 7233817ba13SJoseph Huber return OFFLOAD_FAIL; 7243817ba13SJoseph Huber } 7253817ba13SJoseph Huber if (const char *EnvStr = getenv("LIBOMPTARGET_HEAP_SIZE")) { 7263817ba13SJoseph Huber HeapLimit = std::stol(EnvStr); 7273817ba13SJoseph Huber if (cuCtxSetLimit(CU_LIMIT_MALLOC_HEAP_SIZE, HeapLimit) != CUDA_SUCCESS) 7283817ba13SJoseph Huber return OFFLOAD_FAIL; 7293817ba13SJoseph Huber } else { 7303817ba13SJoseph Huber if (cuCtxGetLimit(&HeapLimit, CU_LIMIT_MALLOC_HEAP_SIZE) != CUDA_SUCCESS) 7313817ba13SJoseph Huber return OFFLOAD_FAIL; 7323817ba13SJoseph Huber } 7333817ba13SJoseph Huber 734119a9ea1SJoseph Huber INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId, 735ae209397SJoseph Huber "Device supports up to %d CUDA blocks and %d threads with a " 736ae209397SJoseph Huber "warp size of %d\n", 737ae209397SJoseph Huber DeviceData[DeviceId].BlocksPerGrid, 738ae209397SJoseph Huber DeviceData[DeviceId].ThreadsPerBlock, DeviceData[DeviceId].WarpSize); 7393817ba13SJoseph Huber INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId, 7403817ba13SJoseph Huber "Device heap size is %d Bytes, device stack size is %d Bytes per " 7413817ba13SJoseph Huber "thread\n", 7423817ba13SJoseph Huber (int)HeapLimit, (int)StackLimit); 7433de4cd12SGeorge Rokos 7443de4cd12SGeorge Rokos // Set default number of teams 7454031bb98SShilei Tian if (EnvNumTeams > 0) { 7463de4cd12SGeorge Rokos DP("Default number of teams set according to environment %d\n", 7474031bb98SShilei Tian EnvNumTeams); 7484031bb98SShilei Tian DeviceData[DeviceId].NumTeams = EnvNumTeams; 7493de4cd12SGeorge Rokos } else { 7504031bb98SShilei Tian DeviceData[DeviceId].NumTeams = DeviceRTLTy::DefaultNumTeams; 7513de4cd12SGeorge Rokos DP("Default number of teams set according to library's default %d\n", 7524031bb98SShilei Tian DeviceRTLTy::DefaultNumTeams); 7533de4cd12SGeorge Rokos } 7544031bb98SShilei Tian 7554031bb98SShilei Tian if (DeviceData[DeviceId].NumTeams > DeviceData[DeviceId].BlocksPerGrid) { 7563de4cd12SGeorge Rokos DP("Default number of teams exceeds device limit, capping at %d\n", 7574031bb98SShilei Tian DeviceData[DeviceId].BlocksPerGrid); 7584031bb98SShilei Tian DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].BlocksPerGrid; 7593de4cd12SGeorge Rokos } 7603de4cd12SGeorge Rokos 7613de4cd12SGeorge Rokos // Set default number of threads 7624031bb98SShilei Tian DeviceData[DeviceId].NumThreads = DeviceRTLTy::DefaultNumThreads; 7633de4cd12SGeorge Rokos DP("Default number of threads set according to library's default %d\n", 7644031bb98SShilei Tian DeviceRTLTy::DefaultNumThreads); 7654031bb98SShilei Tian if (DeviceData[DeviceId].NumThreads > 7664031bb98SShilei Tian DeviceData[DeviceId].ThreadsPerBlock) { 7673de4cd12SGeorge Rokos DP("Default number of threads exceeds device limit, capping at %d\n", 7684031bb98SShilei Tian DeviceData[DeviceId].ThreadsPerBlock); 7694031bb98SShilei Tian DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].ThreadsPerBlock; 7703de4cd12SGeorge Rokos } 7713de4cd12SGeorge Rokos 7723de4cd12SGeorge Rokos return OFFLOAD_SUCCESS; 7733de4cd12SGeorge Rokos } 7743de4cd12SGeorge Rokos 7754031bb98SShilei Tian __tgt_target_table *loadBinary(const int DeviceId, 7764031bb98SShilei Tian const __tgt_device_image *Image) { 7774031bb98SShilei Tian // Set the context we are using 7784031bb98SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 7794031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 7804031bb98SShilei Tian return nullptr; 7813de4cd12SGeorge Rokos 7823de4cd12SGeorge Rokos // Clear the offload table as we are going to create a new one. 7834031bb98SShilei Tian clearOffloadEntriesTable(DeviceId); 7843de4cd12SGeorge Rokos 7853de4cd12SGeorge Rokos // Create the module and extract the function pointers. 7864031bb98SShilei Tian CUmodule Module; 7874031bb98SShilei Tian DP("Load data from image " DPxMOD "\n", DPxPTR(Image->ImageStart)); 7884031bb98SShilei Tian Err = cuModuleLoadDataEx(&Module, Image->ImageStart, 0, nullptr, nullptr); 7894031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuModuleLoadDataEx\n")) 7904031bb98SShilei Tian return nullptr; 7913de4cd12SGeorge Rokos 7923de4cd12SGeorge Rokos DP("CUDA module successfully loaded!\n"); 7934031bb98SShilei Tian 7944031bb98SShilei Tian Modules.push_back(Module); 7953de4cd12SGeorge Rokos 7963de4cd12SGeorge Rokos // Find the symbols in the module by name. 7974031bb98SShilei Tian const __tgt_offload_entry *HostBegin = Image->EntriesBegin; 7984031bb98SShilei Tian const __tgt_offload_entry *HostEnd = Image->EntriesEnd; 7993de4cd12SGeorge Rokos 8005272d29eSJohannes Doerfert std::list<KernelTy> &KernelsList = DeviceData[DeviceId].KernelsList; 8014031bb98SShilei Tian for (const __tgt_offload_entry *E = HostBegin; E != HostEnd; ++E) { 8024031bb98SShilei Tian if (!E->addr) { 8034031bb98SShilei Tian // We return nullptr when something like this happens, the host should 8044031bb98SShilei Tian // have always something in the address to uniquely identify the target 8054031bb98SShilei Tian // region. 8064031bb98SShilei Tian DP("Invalid binary: host entry '<null>' (size = %zd)...\n", E->size); 8074031bb98SShilei Tian return nullptr; 8083de4cd12SGeorge Rokos } 8093de4cd12SGeorge Rokos 8104031bb98SShilei Tian if (E->size) { 8114031bb98SShilei Tian __tgt_offload_entry Entry = *E; 8124031bb98SShilei Tian CUdeviceptr CUPtr; 8134031bb98SShilei Tian size_t CUSize; 81495f0d1edSJon Chesterfield Err = cuModuleGetGlobal(&CUPtr, &CUSize, Module, E->name); 8154031bb98SShilei Tian // We keep this style here because we need the name 8164031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 8172ce16810SJoseph Huber REPORT("Loading global '%s' Failed\n", E->name); 8184031bb98SShilei Tian CUDA_ERR_STRING(Err); 8194031bb98SShilei Tian return nullptr; 8203de4cd12SGeorge Rokos } 8213de4cd12SGeorge Rokos 8224031bb98SShilei Tian if (CUSize != E->size) { 8234031bb98SShilei Tian DP("Loading global '%s' - size mismatch (%zd != %zd)\n", E->name, 8244031bb98SShilei Tian CUSize, E->size); 8254031bb98SShilei Tian return nullptr; 8263de4cd12SGeorge Rokos } 8273de4cd12SGeorge Rokos 8283de4cd12SGeorge Rokos DP("Entry point " DPxMOD " maps to global %s (" DPxMOD ")\n", 8294031bb98SShilei Tian DPxPTR(E - HostBegin), E->name, DPxPTR(CUPtr)); 8304031bb98SShilei Tian 8314031bb98SShilei Tian Entry.addr = (void *)(CUPtr); 8323de4cd12SGeorge Rokos 833aace6d28SGheorghe-Teodor Bercea // Note: In the current implementation declare target variables 834aace6d28SGheorghe-Teodor Bercea // can either be link or to. This means that once unified 835aace6d28SGheorghe-Teodor Bercea // memory is activated via the requires directive, the variable 836aace6d28SGheorghe-Teodor Bercea // can be used directly from the host in both cases. 837aace6d28SGheorghe-Teodor Bercea // TODO: when variables types other than to or link are added, 8384c6a098aSKazuaki Ishizaki // the below condition should be changed to explicitly 839aace6d28SGheorghe-Teodor Bercea // check for to and link variables types: 8404031bb98SShilei Tian // (RequiresFlags & OMP_REQ_UNIFIED_SHARED_MEMORY && (e->flags & 8414031bb98SShilei Tian // OMP_DECLARE_TARGET_LINK || e->flags == OMP_DECLARE_TARGET_TO)) 8424031bb98SShilei Tian if (RequiresFlags & OMP_REQ_UNIFIED_SHARED_MEMORY) { 843aace6d28SGheorghe-Teodor Bercea // If unified memory is present any target link or to variables 844c5fe030cSGheorghe-Teodor Bercea // can access host addresses directly. There is no longer a 845c5fe030cSGheorghe-Teodor Bercea // need for device copies. 84695f0d1edSJon Chesterfield cuMemcpyHtoD(CUPtr, E->addr, sizeof(void *)); 8474031bb98SShilei Tian DP("Copy linked variable host address (" DPxMOD 8484031bb98SShilei Tian ") to device address (" DPxMOD ")\n", 8494031bb98SShilei Tian DPxPTR(*((void **)E->addr)), DPxPTR(CUPtr)); 850c5fe030cSGheorghe-Teodor Bercea } 851c5fe030cSGheorghe-Teodor Bercea 8524031bb98SShilei Tian addOffloadEntry(DeviceId, Entry); 8533de4cd12SGeorge Rokos 8543de4cd12SGeorge Rokos continue; 8553de4cd12SGeorge Rokos } 8563de4cd12SGeorge Rokos 8574031bb98SShilei Tian CUfunction Func; 8584031bb98SShilei Tian Err = cuModuleGetFunction(&Func, Module, E->name); 8594031bb98SShilei Tian // We keep this style here because we need the name 8604031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 8612ce16810SJoseph Huber REPORT("Loading '%s' Failed\n", E->name); 8624031bb98SShilei Tian CUDA_ERR_STRING(Err); 8634031bb98SShilei Tian return nullptr; 8643de4cd12SGeorge Rokos } 8653de4cd12SGeorge Rokos 8663de4cd12SGeorge Rokos DP("Entry point " DPxMOD " maps to %s (" DPxMOD ")\n", 8674031bb98SShilei Tian DPxPTR(E - HostBegin), E->name, DPxPTR(Func)); 8683de4cd12SGeorge Rokos 8693de4cd12SGeorge Rokos // default value GENERIC (in case symbol is missing from cubin file) 8703de4cd12SGeorge Rokos int8_t ExecModeVal = ExecutionModeType::GENERIC; 8714031bb98SShilei Tian std::string ExecModeNameStr(E->name); 8723de4cd12SGeorge Rokos ExecModeNameStr += "_exec_mode"; 8733de4cd12SGeorge Rokos const char *ExecModeName = ExecModeNameStr.c_str(); 8743de4cd12SGeorge Rokos 8753de4cd12SGeorge Rokos CUdeviceptr ExecModePtr; 8764031bb98SShilei Tian size_t CUSize; 87795f0d1edSJon Chesterfield Err = cuModuleGetGlobal(&ExecModePtr, &CUSize, Module, ExecModeName); 8784031bb98SShilei Tian if (Err == CUDA_SUCCESS) { 8794031bb98SShilei Tian if (CUSize != sizeof(int8_t)) { 8803de4cd12SGeorge Rokos DP("Loading global exec_mode '%s' - size mismatch (%zd != %zd)\n", 8814031bb98SShilei Tian ExecModeName, CUSize, sizeof(int8_t)); 8824031bb98SShilei Tian return nullptr; 8833de4cd12SGeorge Rokos } 8843de4cd12SGeorge Rokos 88595f0d1edSJon Chesterfield Err = cuMemcpyDtoH(&ExecModeVal, ExecModePtr, CUSize); 8864031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 8872ce16810SJoseph Huber REPORT("Error when copying data from device to host. Pointers: " 8883de4cd12SGeorge Rokos "host = " DPxMOD ", device = " DPxMOD ", size = %zd\n", 8894031bb98SShilei Tian DPxPTR(&ExecModeVal), DPxPTR(ExecModePtr), CUSize); 8904031bb98SShilei Tian CUDA_ERR_STRING(Err); 8914031bb98SShilei Tian return nullptr; 8923de4cd12SGeorge Rokos } 8933de4cd12SGeorge Rokos 8947d576392SJoseph Huber if (ExecModeVal < 0 || ExecModeVal > 2) { 8953de4cd12SGeorge Rokos DP("Error wrong exec_mode value specified in cubin file: %d\n", 8963de4cd12SGeorge Rokos ExecModeVal); 8974031bb98SShilei Tian return nullptr; 8983de4cd12SGeorge Rokos } 8993de4cd12SGeorge Rokos } else { 900edb8acdcSJoseph Huber DP("Loading global exec_mode '%s' - symbol missing, using default " 9014031bb98SShilei Tian "value GENERIC (1)\n", 9024031bb98SShilei Tian ExecModeName); 9033de4cd12SGeorge Rokos } 9043de4cd12SGeorge Rokos 9054031bb98SShilei Tian KernelsList.emplace_back(Func, ExecModeVal); 9063de4cd12SGeorge Rokos 9074031bb98SShilei Tian __tgt_offload_entry Entry = *E; 9084031bb98SShilei Tian Entry.addr = &KernelsList.back(); 9094031bb98SShilei Tian addOffloadEntry(DeviceId, Entry); 9103de4cd12SGeorge Rokos } 9113de4cd12SGeorge Rokos 912e1c7a46dSGuansong Zhang // send device environment data to the device 913e1c7a46dSGuansong Zhang { 914ec02c34bSJoseph Huber // TODO: The device ID used here is not the real device ID used by OpenMP. 915ec02c34bSJoseph Huber omptarget_device_environmentTy DeviceEnv{ 916ec02c34bSJoseph Huber 0, static_cast<uint32_t>(NumberOfDevices), 917f1c821faSJoseph Huber static_cast<uint32_t>(DeviceId), DynamicMemorySize}; 918e1c7a46dSGuansong Zhang 919e1c7a46dSGuansong Zhang #ifdef OMPTARGET_DEBUG 9204031bb98SShilei Tian if (const char *EnvStr = getenv("LIBOMPTARGET_DEVICE_RTL_DEBUG")) 9214031bb98SShilei Tian DeviceEnv.debug_level = std::stoi(EnvStr); 922e1c7a46dSGuansong Zhang #endif 923e1c7a46dSGuansong Zhang 9244031bb98SShilei Tian const char *DeviceEnvName = "omptarget_device_environment"; 9254031bb98SShilei Tian CUdeviceptr DeviceEnvPtr; 9264031bb98SShilei Tian size_t CUSize; 927e1c7a46dSGuansong Zhang 92895f0d1edSJon Chesterfield Err = cuModuleGetGlobal(&DeviceEnvPtr, &CUSize, Module, DeviceEnvName); 9294031bb98SShilei Tian if (Err == CUDA_SUCCESS) { 9304031bb98SShilei Tian if (CUSize != sizeof(DeviceEnv)) { 9312ce16810SJoseph Huber REPORT( 9322ce16810SJoseph Huber "Global device_environment '%s' - size mismatch (%zu != %zu)\n", 9334031bb98SShilei Tian DeviceEnvName, CUSize, sizeof(int32_t)); 9344031bb98SShilei Tian CUDA_ERR_STRING(Err); 9354031bb98SShilei Tian return nullptr; 936e1c7a46dSGuansong Zhang } 937e1c7a46dSGuansong Zhang 93895f0d1edSJon Chesterfield Err = cuMemcpyHtoD(DeviceEnvPtr, &DeviceEnv, CUSize); 9394031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 9402ce16810SJoseph Huber REPORT("Error when copying data from host to device. Pointers: " 941e1c7a46dSGuansong Zhang "host = " DPxMOD ", device = " DPxMOD ", size = %zu\n", 9424031bb98SShilei Tian DPxPTR(&DeviceEnv), DPxPTR(DeviceEnvPtr), CUSize); 9434031bb98SShilei Tian CUDA_ERR_STRING(Err); 9444031bb98SShilei Tian return nullptr; 945e1c7a46dSGuansong Zhang } 946e1c7a46dSGuansong Zhang 9474031bb98SShilei Tian DP("Sending global device environment data %zu bytes\n", CUSize); 948e1c7a46dSGuansong Zhang } else { 9494031bb98SShilei Tian DP("Finding global device environment '%s' - symbol missing.\n", 9504031bb98SShilei Tian DeviceEnvName); 9514031bb98SShilei Tian DP("Continue, considering this is a device RTL which does not accept " 9524031bb98SShilei Tian "environment setting.\n"); 953e1c7a46dSGuansong Zhang } 954e1c7a46dSGuansong Zhang } 955e1c7a46dSGuansong Zhang 9564031bb98SShilei Tian return getOffloadEntriesTable(DeviceId); 9573de4cd12SGeorge Rokos } 9583de4cd12SGeorge Rokos 959a15f8589SJoseph Huber void *dataAlloc(const int DeviceId, const int64_t Size, 960a15f8589SJoseph Huber const TargetAllocTy Kind) { 961a15f8589SJoseph Huber switch (Kind) { 962a15f8589SJoseph Huber case TARGET_ALLOC_DEFAULT: 963a15f8589SJoseph Huber case TARGET_ALLOC_DEVICE: 9640871d6d5SShilei Tian if (UseMemoryManager) 9650871d6d5SShilei Tian return MemoryManagers[DeviceId]->allocate(Size, nullptr); 966a15f8589SJoseph Huber else 967a15f8589SJoseph Huber return DeviceAllocators[DeviceId].allocate(Size, nullptr, Kind); 968a15f8589SJoseph Huber case TARGET_ALLOC_HOST: 969a15f8589SJoseph Huber case TARGET_ALLOC_SHARED: 970a15f8589SJoseph Huber return DeviceAllocators[DeviceId].allocate(Size, nullptr, Kind); 971a15f8589SJoseph Huber } 9724031bb98SShilei Tian 973a15f8589SJoseph Huber REPORT("Invalid target data allocation kind or requested allocator not " 974a15f8589SJoseph Huber "implemented yet\n"); 975a15f8589SJoseph Huber 976a15f8589SJoseph Huber return nullptr; 9773de4cd12SGeorge Rokos } 9783de4cd12SGeorge Rokos 9794031bb98SShilei Tian int dataSubmit(const int DeviceId, const void *TgtPtr, const void *HstPtr, 9805449fbb5SJohannes Doerfert const int64_t Size, __tgt_async_info *AsyncInfo) const { 9815449fbb5SJohannes Doerfert assert(AsyncInfo && "AsyncInfo is nullptr"); 9823de4cd12SGeorge Rokos 9834031bb98SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 9844031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 9853de4cd12SGeorge Rokos return OFFLOAD_FAIL; 9863de4cd12SGeorge Rokos 9875449fbb5SJohannes Doerfert CUstream Stream = getStream(DeviceId, AsyncInfo); 9883de4cd12SGeorge Rokos 98995f0d1edSJon Chesterfield Err = cuMemcpyHtoDAsync((CUdeviceptr)TgtPtr, HstPtr, Size, Stream); 9904031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 99129338459SJoseph Huber DP("Error when copying data from host to device. Pointers: host " 9922ce16810SJoseph Huber "= " DPxMOD ", device = " DPxMOD ", size = %" PRId64 "\n", 9934031bb98SShilei Tian DPxPTR(HstPtr), DPxPTR(TgtPtr), Size); 9944031bb98SShilei Tian CUDA_ERR_STRING(Err); 9953de4cd12SGeorge Rokos return OFFLOAD_FAIL; 9963de4cd12SGeorge Rokos } 9973de4cd12SGeorge Rokos 9983de4cd12SGeorge Rokos return OFFLOAD_SUCCESS; 9993de4cd12SGeorge Rokos } 10003de4cd12SGeorge Rokos 10014031bb98SShilei Tian int dataRetrieve(const int DeviceId, void *HstPtr, const void *TgtPtr, 10025449fbb5SJohannes Doerfert const int64_t Size, __tgt_async_info *AsyncInfo) const { 10035449fbb5SJohannes Doerfert assert(AsyncInfo && "AsyncInfo is nullptr"); 10044031bb98SShilei Tian 10054031bb98SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 10064031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 100703ff643dSShilei Tian return OFFLOAD_FAIL; 100803ff643dSShilei Tian 10095449fbb5SJohannes Doerfert CUstream Stream = getStream(DeviceId, AsyncInfo); 101003ff643dSShilei Tian 101195f0d1edSJon Chesterfield Err = cuMemcpyDtoHAsync(HstPtr, (CUdeviceptr)TgtPtr, Size, Stream); 10124031bb98SShilei Tian if (Err != CUDA_SUCCESS) { 101329338459SJoseph Huber DP("Error when copying data from device to host. Pointers: host " 10142ce16810SJoseph Huber "= " DPxMOD ", device = " DPxMOD ", size = %" PRId64 "\n", 10154031bb98SShilei Tian DPxPTR(HstPtr), DPxPTR(TgtPtr), Size); 10164031bb98SShilei Tian CUDA_ERR_STRING(Err); 10173de4cd12SGeorge Rokos return OFFLOAD_FAIL; 10183de4cd12SGeorge Rokos } 10193de4cd12SGeorge Rokos 10204031bb98SShilei Tian return OFFLOAD_SUCCESS; 10214031bb98SShilei Tian } 10224031bb98SShilei Tian 1023a014fbbcSShilei Tian int dataExchange(int SrcDevId, const void *SrcPtr, int DstDevId, void *DstPtr, 10245449fbb5SJohannes Doerfert int64_t Size, __tgt_async_info *AsyncInfo) const { 10255449fbb5SJohannes Doerfert assert(AsyncInfo && "AsyncInfo is nullptr"); 1026a014fbbcSShilei Tian 1027a014fbbcSShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[SrcDevId].Context); 1028a014fbbcSShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 1029a014fbbcSShilei Tian return OFFLOAD_FAIL; 1030a014fbbcSShilei Tian 10315449fbb5SJohannes Doerfert CUstream Stream = getStream(SrcDevId, AsyncInfo); 1032a014fbbcSShilei Tian 1033a014fbbcSShilei Tian // If they are two devices, we try peer to peer copy first 1034a014fbbcSShilei Tian if (SrcDevId != DstDevId) { 1035a014fbbcSShilei Tian int CanAccessPeer = 0; 1036a014fbbcSShilei Tian Err = cuDeviceCanAccessPeer(&CanAccessPeer, SrcDevId, DstDevId); 1037a014fbbcSShilei Tian if (Err != CUDA_SUCCESS) { 10382ce16810SJoseph Huber REPORT("Error returned from cuDeviceCanAccessPeer. src = %" PRId32 1039a014fbbcSShilei Tian ", dst = %" PRId32 "\n", 1040a014fbbcSShilei Tian SrcDevId, DstDevId); 1041a014fbbcSShilei Tian CUDA_ERR_STRING(Err); 1042a014fbbcSShilei Tian return memcpyDtoD(SrcPtr, DstPtr, Size, Stream); 1043a014fbbcSShilei Tian } 1044a014fbbcSShilei Tian 1045a014fbbcSShilei Tian if (!CanAccessPeer) { 1046a014fbbcSShilei Tian DP("P2P memcpy not supported so fall back to D2D memcpy"); 1047a014fbbcSShilei Tian return memcpyDtoD(SrcPtr, DstPtr, Size, Stream); 1048a014fbbcSShilei Tian } 1049a014fbbcSShilei Tian 1050a014fbbcSShilei Tian Err = cuCtxEnablePeerAccess(DeviceData[DstDevId].Context, 0); 1051a014fbbcSShilei Tian if (Err != CUDA_SUCCESS) { 10522ce16810SJoseph Huber REPORT("Error returned from cuCtxEnablePeerAccess. src = %" PRId32 1053a014fbbcSShilei Tian ", dst = %" PRId32 "\n", 1054a014fbbcSShilei Tian SrcDevId, DstDevId); 1055a014fbbcSShilei Tian CUDA_ERR_STRING(Err); 1056a014fbbcSShilei Tian return memcpyDtoD(SrcPtr, DstPtr, Size, Stream); 1057a014fbbcSShilei Tian } 1058a014fbbcSShilei Tian 1059a014fbbcSShilei Tian Err = cuMemcpyPeerAsync((CUdeviceptr)DstPtr, DeviceData[DstDevId].Context, 1060a014fbbcSShilei Tian (CUdeviceptr)SrcPtr, DeviceData[SrcDevId].Context, 1061a014fbbcSShilei Tian Size, Stream); 1062a014fbbcSShilei Tian if (Err == CUDA_SUCCESS) 1063a014fbbcSShilei Tian return OFFLOAD_SUCCESS; 1064a014fbbcSShilei Tian 106529338459SJoseph Huber DP("Error returned from cuMemcpyPeerAsync. src_ptr = " DPxMOD 106629338459SJoseph Huber ", src_id =%" PRId32 ", dst_ptr = " DPxMOD ", dst_id =%" PRId32 "\n", 106745bb073dSYe Luo DPxPTR(SrcPtr), SrcDevId, DPxPTR(DstPtr), DstDevId); 1068a014fbbcSShilei Tian CUDA_ERR_STRING(Err); 1069a014fbbcSShilei Tian } 1070a014fbbcSShilei Tian 1071a014fbbcSShilei Tian return memcpyDtoD(SrcPtr, DstPtr, Size, Stream); 1072a014fbbcSShilei Tian } 1073a014fbbcSShilei Tian 10740871d6d5SShilei Tian int dataDelete(const int DeviceId, void *TgtPtr) { 10750871d6d5SShilei Tian if (UseMemoryManager) 10760871d6d5SShilei Tian return MemoryManagers[DeviceId]->free(TgtPtr); 10774031bb98SShilei Tian 10780871d6d5SShilei Tian return DeviceAllocators[DeviceId].free(TgtPtr); 10794031bb98SShilei Tian } 10804031bb98SShilei Tian 1081aa27cfc1SJohannes Doerfert int runTargetTeamRegion(const int DeviceId, void *TgtEntryPtr, void **TgtArgs, 1082aa27cfc1SJohannes Doerfert ptrdiff_t *TgtOffsets, const int ArgNum, 1083aa27cfc1SJohannes Doerfert const int TeamNum, const int ThreadLimit, 10844031bb98SShilei Tian const unsigned int LoopTripCount, 10854031bb98SShilei Tian __tgt_async_info *AsyncInfo) const { 10864031bb98SShilei Tian CUresult Err = cuCtxSetCurrent(DeviceData[DeviceId].Context); 10874031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuCtxSetCurrent\n")) 10884031bb98SShilei Tian return OFFLOAD_FAIL; 10894031bb98SShilei Tian 10903de4cd12SGeorge Rokos // All args are references. 10914031bb98SShilei Tian std::vector<void *> Args(ArgNum); 10924031bb98SShilei Tian std::vector<void *> Ptrs(ArgNum); 10933de4cd12SGeorge Rokos 10944031bb98SShilei Tian for (int I = 0; I < ArgNum; ++I) { 10954031bb98SShilei Tian Ptrs[I] = (void *)((intptr_t)TgtArgs[I] + TgtOffsets[I]); 10964031bb98SShilei Tian Args[I] = &Ptrs[I]; 10971546d319SGeorge Rokos } 10983de4cd12SGeorge Rokos 1099aa27cfc1SJohannes Doerfert KernelTy *KernelInfo = reinterpret_cast<KernelTy *>(TgtEntryPtr); 11003de4cd12SGeorge Rokos 1101aa27cfc1SJohannes Doerfert int CudaThreadsPerBlock; 11024031bb98SShilei Tian if (ThreadLimit > 0) { 11034031bb98SShilei Tian DP("Setting CUDA threads per block to requested %d\n", ThreadLimit); 11044031bb98SShilei Tian CudaThreadsPerBlock = ThreadLimit; 11053de4cd12SGeorge Rokos // Add master warp if necessary 11063de4cd12SGeorge Rokos if (KernelInfo->ExecutionMode == GENERIC) { 11074031bb98SShilei Tian DP("Adding master warp: +%d threads\n", DeviceData[DeviceId].WarpSize); 11084031bb98SShilei Tian CudaThreadsPerBlock += DeviceData[DeviceId].WarpSize; 11093de4cd12SGeorge Rokos } 1110c13df8e5SGeorge Rokos } else { 1111c13df8e5SGeorge Rokos DP("Setting CUDA threads per block to default %d\n", 11124031bb98SShilei Tian DeviceData[DeviceId].NumThreads); 11134031bb98SShilei Tian CudaThreadsPerBlock = DeviceData[DeviceId].NumThreads; 1114c13df8e5SGeorge Rokos } 11153de4cd12SGeorge Rokos 11164031bb98SShilei Tian if (CudaThreadsPerBlock > DeviceData[DeviceId].ThreadsPerBlock) { 11173de4cd12SGeorge Rokos DP("Threads per block capped at device limit %d\n", 11184031bb98SShilei Tian DeviceData[DeviceId].ThreadsPerBlock); 11194031bb98SShilei Tian CudaThreadsPerBlock = DeviceData[DeviceId].ThreadsPerBlock; 11203de4cd12SGeorge Rokos } 11213de4cd12SGeorge Rokos 1122aa27cfc1SJohannes Doerfert if (!KernelInfo->MaxThreadsPerBlock) { 1123aa27cfc1SJohannes Doerfert Err = cuFuncGetAttribute(&KernelInfo->MaxThreadsPerBlock, 11244031bb98SShilei Tian CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK, 11254031bb98SShilei Tian KernelInfo->Func); 1126aa27cfc1SJohannes Doerfert if (!checkResult(Err, "Error returned from cuFuncGetAttribute\n")) 1127aa27cfc1SJohannes Doerfert return OFFLOAD_FAIL; 1128aa27cfc1SJohannes Doerfert } 1129aa27cfc1SJohannes Doerfert 1130aa27cfc1SJohannes Doerfert if (KernelInfo->MaxThreadsPerBlock < CudaThreadsPerBlock) { 1131aa27cfc1SJohannes Doerfert DP("Threads per block capped at kernel limit %d\n", 1132aa27cfc1SJohannes Doerfert KernelInfo->MaxThreadsPerBlock); 1133aa27cfc1SJohannes Doerfert CudaThreadsPerBlock = KernelInfo->MaxThreadsPerBlock; 11343de4cd12SGeorge Rokos } 11353de4cd12SGeorge Rokos 11364031bb98SShilei Tian unsigned int CudaBlocksPerGrid; 11374031bb98SShilei Tian if (TeamNum <= 0) { 11384031bb98SShilei Tian if (LoopTripCount > 0 && EnvNumTeams < 0) { 1139c13df8e5SGeorge Rokos if (KernelInfo->ExecutionMode == SPMD) { 11404031bb98SShilei Tian // We have a combined construct, i.e. `target teams distribute 11414031bb98SShilei Tian // parallel for [simd]`. We launch so many teams so that each thread 11424031bb98SShilei Tian // will execute one iteration of the loop. round up to the nearest 11434031bb98SShilei Tian // integer 11444031bb98SShilei Tian CudaBlocksPerGrid = ((LoopTripCount - 1) / CudaThreadsPerBlock) + 1; 11457d576392SJoseph Huber } else if (KernelInfo->ExecutionMode == GENERIC) { 1146c13df8e5SGeorge Rokos // If we reach this point, then we have a non-combined construct, i.e. 1147c13df8e5SGeorge Rokos // `teams distribute` with a nested `parallel for` and each team is 1148c13df8e5SGeorge Rokos // assigned one iteration of the `distribute` loop. E.g.: 1149c13df8e5SGeorge Rokos // 1150c13df8e5SGeorge Rokos // #pragma omp target teams distribute 1151c13df8e5SGeorge Rokos // for(...loop_tripcount...) { 1152c13df8e5SGeorge Rokos // #pragma omp parallel for 1153c13df8e5SGeorge Rokos // for(...) {} 1154c13df8e5SGeorge Rokos // } 1155c13df8e5SGeorge Rokos // 1156c13df8e5SGeorge Rokos // Threads within a team will execute the iterations of the `parallel` 1157c13df8e5SGeorge Rokos // loop. 11584031bb98SShilei Tian CudaBlocksPerGrid = LoopTripCount; 11597d576392SJoseph Huber } else if (KernelInfo->ExecutionMode == SPMD_GENERIC) { 11607d576392SJoseph Huber // If we reach this point, then we are executing a kernel that was 11617d576392SJoseph Huber // transformed from Generic-mode to SPMD-mode. This kernel has 11627d576392SJoseph Huber // SPMD-mode execution, but needs its blocks to be scheduled 11637d576392SJoseph Huber // differently because the current loop trip count only applies to the 11647d576392SJoseph Huber // `teams distribute` region and will create var too few blocks using 11657d576392SJoseph Huber // the regular SPMD-mode method. 11667d576392SJoseph Huber CudaBlocksPerGrid = LoopTripCount; 1167a158d366SJoseph Huber } else { 1168a158d366SJoseph Huber REPORT("Unknown execution mode: %d\n", KernelInfo->ExecutionMode); 1169a158d366SJoseph Huber return OFFLOAD_FAIL; 1170c13df8e5SGeorge Rokos } 117145bb073dSYe Luo DP("Using %d teams due to loop trip count %" PRIu32 11724031bb98SShilei Tian " and number of threads per block %d\n", 11734031bb98SShilei Tian CudaBlocksPerGrid, LoopTripCount, CudaThreadsPerBlock); 11743de4cd12SGeorge Rokos } else { 11754031bb98SShilei Tian DP("Using default number of teams %d\n", DeviceData[DeviceId].NumTeams); 11764031bb98SShilei Tian CudaBlocksPerGrid = DeviceData[DeviceId].NumTeams; 11773de4cd12SGeorge Rokos } 11784031bb98SShilei Tian } else if (TeamNum > DeviceData[DeviceId].BlocksPerGrid) { 11793de4cd12SGeorge Rokos DP("Capping number of teams to team limit %d\n", 11804031bb98SShilei Tian DeviceData[DeviceId].BlocksPerGrid); 11814031bb98SShilei Tian CudaBlocksPerGrid = DeviceData[DeviceId].BlocksPerGrid; 11823de4cd12SGeorge Rokos } else { 11834031bb98SShilei Tian DP("Using requested number of teams %d\n", TeamNum); 11844031bb98SShilei Tian CudaBlocksPerGrid = TeamNum; 11853de4cd12SGeorge Rokos } 11863de4cd12SGeorge Rokos 1187119a9ea1SJoseph Huber INFO(OMP_INFOTYPE_PLUGIN_KERNEL, DeviceId, 1188ae209397SJoseph Huber "Launching kernel %s with %d blocks and %d threads in %s " 1189ae209397SJoseph Huber "mode\n", 1190ae209397SJoseph Huber (getOffloadEntry(DeviceId, TgtEntryPtr)) 1191ae209397SJoseph Huber ? getOffloadEntry(DeviceId, TgtEntryPtr)->name 1192ae209397SJoseph Huber : "(null)", 1193ae209397SJoseph Huber CudaBlocksPerGrid, CudaThreadsPerBlock, 11947d576392SJoseph Huber (KernelInfo->ExecutionMode != SPMD 119576c0c0caSJoseph Huber ? (KernelInfo->ExecutionMode == GENERIC ? "Generic" 119676c0c0caSJoseph Huber : "SPMD-Generic") 11977d576392SJoseph Huber : "SPMD")); 11983de4cd12SGeorge Rokos 11994031bb98SShilei Tian CUstream Stream = getStream(DeviceId, AsyncInfo); 12004031bb98SShilei Tian Err = cuLaunchKernel(KernelInfo->Func, CudaBlocksPerGrid, /* gridDimY */ 1, 12014031bb98SShilei Tian /* gridDimZ */ 1, CudaThreadsPerBlock, 12024031bb98SShilei Tian /* blockDimY */ 1, /* blockDimZ */ 1, 1203f1c821faSJoseph Huber DynamicMemorySize, Stream, &Args[0], nullptr); 12044031bb98SShilei Tian if (!checkResult(Err, "Error returned from cuLaunchKernel\n")) 12054031bb98SShilei Tian return OFFLOAD_FAIL; 12064031bb98SShilei Tian 12074031bb98SShilei Tian DP("Launch of entry point at " DPxMOD " successful!\n", 12084031bb98SShilei Tian DPxPTR(TgtEntryPtr)); 12094031bb98SShilei Tian 12104031bb98SShilei Tian return OFFLOAD_SUCCESS; 12114031bb98SShilei Tian } 12124031bb98SShilei Tian 12135449fbb5SJohannes Doerfert int synchronize(const int DeviceId, __tgt_async_info *AsyncInfo) const { 12145449fbb5SJohannes Doerfert CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue); 12154031bb98SShilei Tian CUresult Err = cuStreamSynchronize(Stream); 12163de4cd12SGeorge Rokos 12174031bb98SShilei Tian // Once the stream is synchronized, return it to stream pool and reset 12185449fbb5SJohannes Doerfert // AsyncInfo. This is to make sure the synchronization only works for its 12194031bb98SShilei Tian // own tasks. 12205449fbb5SJohannes Doerfert StreamManager->returnStream(DeviceId, 12215449fbb5SJohannes Doerfert reinterpret_cast<CUstream>(AsyncInfo->Queue)); 12225449fbb5SJohannes Doerfert AsyncInfo->Queue = nullptr; 12233de4cd12SGeorge Rokos 122416064e71SJoseph Huber if (Err != CUDA_SUCCESS) { 122529338459SJoseph Huber DP("Error when synchronizing stream. stream = " DPxMOD 122616064e71SJoseph Huber ", async info ptr = " DPxMOD "\n", 122716064e71SJoseph Huber DPxPTR(Stream), DPxPTR(AsyncInfo)); 122816064e71SJoseph Huber CUDA_ERR_STRING(Err); 122916064e71SJoseph Huber } 123016064e71SJoseph Huber return (Err == CUDA_SUCCESS) ? OFFLOAD_SUCCESS : OFFLOAD_FAIL; 12313de4cd12SGeorge Rokos } 1232d2f85d09SJose M Monsalve Diaz 1233d2f85d09SJose M Monsalve Diaz void printDeviceInfo(int32_t device_id) { 1234d2f85d09SJose M Monsalve Diaz char TmpChar[1000]; 1235d2f85d09SJose M Monsalve Diaz std::string TmpStr; 1236d2f85d09SJose M Monsalve Diaz size_t TmpSt; 1237d2f85d09SJose M Monsalve Diaz int TmpInt, TmpInt2, TmpInt3; 1238d2f85d09SJose M Monsalve Diaz 1239d2f85d09SJose M Monsalve Diaz CUdevice Device; 1240d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGet(&Device, device_id), 1241d2f85d09SJose M Monsalve Diaz "Error returned from cuCtxGetDevice\n"); 1242d2f85d09SJose M Monsalve Diaz 1243d2f85d09SJose M Monsalve Diaz cuDriverGetVersion(&TmpInt); 1244d2f85d09SJose M Monsalve Diaz printf(" CUDA Driver Version: \t\t%d \n", TmpInt); 1245d2f85d09SJose M Monsalve Diaz printf(" CUDA Device Number: \t\t%d \n", device_id); 1246d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetName(TmpChar, 1000, Device), 1247d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetName\n"); 1248d2f85d09SJose M Monsalve Diaz printf(" Device Name: \t\t\t%s \n", TmpChar); 1249d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceTotalMem(&TmpSt, Device), 1250d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceTotalMem\n"); 1251d2f85d09SJose M Monsalve Diaz printf(" Global Memory Size: \t\t%zu bytes \n", TmpSt); 1252d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1253d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT, Device), 1254d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1255d2f85d09SJose M Monsalve Diaz printf(" Number of Multiprocessors: \t\t%d \n", TmpInt); 1256d2f85d09SJose M Monsalve Diaz checkResult( 1257d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_GPU_OVERLAP, Device), 1258d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1259d2f85d09SJose M Monsalve Diaz printf(" Concurrent Copy and Execution: \t%s \n", BOOL2TEXT(TmpInt)); 1260d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1261d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY, Device), 1262d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1263d2f85d09SJose M Monsalve Diaz printf(" Total Constant Memory: \t\t%d bytes\n", TmpInt); 1264d2f85d09SJose M Monsalve Diaz checkResult( 1265d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute( 1266d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK, Device), 1267d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1268d2f85d09SJose M Monsalve Diaz printf(" Max Shared Memory per Block: \t%d bytes \n", TmpInt); 126928939b6aSShilei Tian checkResult( 127028939b6aSShilei Tian cuDeviceGetAttribute( 1271d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK, Device), 1272d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1273d2f85d09SJose M Monsalve Diaz printf(" Registers per Block: \t\t%d \n", TmpInt); 1274d2f85d09SJose M Monsalve Diaz checkResult( 1275d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_WARP_SIZE, Device), 1276d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1277d2f85d09SJose M Monsalve Diaz printf(" Warp Size: \t\t\t\t%d Threads \n", TmpInt); 1278d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1279d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK, Device), 1280d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1281d2f85d09SJose M Monsalve Diaz printf(" Maximum Threads per Block: \t\t%d \n", TmpInt); 128228939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 128328939b6aSShilei Tian &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X, Device), 1284d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 128528939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 128628939b6aSShilei Tian &TmpInt2, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y, Device), 1287d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 128828939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 128928939b6aSShilei Tian &TmpInt3, CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z, Device), 1290d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1291d2f85d09SJose M Monsalve Diaz printf(" Maximum Block Dimensions: \t\t%d, %d, %d \n", TmpInt, TmpInt2, 1292d2f85d09SJose M Monsalve Diaz TmpInt3); 129328939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 129428939b6aSShilei Tian &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X, Device), 1295d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 129628939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 129728939b6aSShilei Tian &TmpInt2, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y, Device), 1298d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 129928939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 130028939b6aSShilei Tian &TmpInt3, CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z, Device), 1301d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1302d2f85d09SJose M Monsalve Diaz printf(" Maximum Grid Dimensions: \t\t%d x %d x %d \n", TmpInt, TmpInt2, 1303d2f85d09SJose M Monsalve Diaz TmpInt3); 1304d2f85d09SJose M Monsalve Diaz checkResult( 1305d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_MAX_PITCH, Device), 1306d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1307d2f85d09SJose M Monsalve Diaz printf(" Maximum Memory Pitch: \t\t%d bytes \n", TmpInt); 1308d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1309d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT, Device), 1310d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1311d2f85d09SJose M Monsalve Diaz printf(" Texture Alignment: \t\t\t%d bytes \n", TmpInt); 1312d2f85d09SJose M Monsalve Diaz checkResult( 1313d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_CLOCK_RATE, Device), 1314d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1315d2f85d09SJose M Monsalve Diaz printf(" Clock Rate: \t\t\t%d kHz\n", TmpInt); 1316d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1317d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT, Device), 1318d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1319d2f85d09SJose M Monsalve Diaz printf(" Execution Timeout: \t\t\t%s \n", BOOL2TEXT(TmpInt)); 1320d2f85d09SJose M Monsalve Diaz checkResult( 1321d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_INTEGRATED, Device), 1322d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1323d2f85d09SJose M Monsalve Diaz printf(" Integrated Device: \t\t\t%s \n", BOOL2TEXT(TmpInt)); 1324d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1325d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY, Device), 1326d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1327d2f85d09SJose M Monsalve Diaz printf(" Can Map Host Memory: \t\t%s \n", BOOL2TEXT(TmpInt)); 1328d2f85d09SJose M Monsalve Diaz checkResult( 1329d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_MODE, Device), 1330d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1331d2f85d09SJose M Monsalve Diaz if (TmpInt == CU_COMPUTEMODE_DEFAULT) 1332d2f85d09SJose M Monsalve Diaz TmpStr = "DEFAULT"; 1333d2f85d09SJose M Monsalve Diaz else if (TmpInt == CU_COMPUTEMODE_PROHIBITED) 1334d2f85d09SJose M Monsalve Diaz TmpStr = "PROHIBITED"; 1335d2f85d09SJose M Monsalve Diaz else if (TmpInt == CU_COMPUTEMODE_EXCLUSIVE_PROCESS) 1336d2f85d09SJose M Monsalve Diaz TmpStr = "EXCLUSIVE PROCESS"; 1337d2f85d09SJose M Monsalve Diaz else 1338d2f85d09SJose M Monsalve Diaz TmpStr = "unknown"; 1339d2f85d09SJose M Monsalve Diaz printf(" Compute Mode: \t\t\t%s \n", TmpStr.c_str()); 1340d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1341d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS, Device), 1342d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1343d2f85d09SJose M Monsalve Diaz printf(" Concurrent Kernels: \t\t%s \n", BOOL2TEXT(TmpInt)); 1344d2f85d09SJose M Monsalve Diaz checkResult( 1345d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_ECC_ENABLED, Device), 1346d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1347d2f85d09SJose M Monsalve Diaz printf(" ECC Enabled: \t\t\t%s \n", BOOL2TEXT(TmpInt)); 1348d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1349d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE, Device), 1350d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1351d2f85d09SJose M Monsalve Diaz printf(" Memory Clock Rate: \t\t\t%d kHz\n", TmpInt); 135228939b6aSShilei Tian checkResult( 135328939b6aSShilei Tian cuDeviceGetAttribute( 1354d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH, Device), 1355d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1356d2f85d09SJose M Monsalve Diaz printf(" Memory Bus Width: \t\t\t%d bits\n", TmpInt); 135728939b6aSShilei Tian checkResult(cuDeviceGetAttribute(&TmpInt, CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE, 135828939b6aSShilei Tian Device), 1359d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1360d2f85d09SJose M Monsalve Diaz printf(" L2 Cache Size: \t\t\t%d bytes \n", TmpInt); 136128939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 136228939b6aSShilei Tian &TmpInt, CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR, 136328939b6aSShilei Tian Device), 1364d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1365d2f85d09SJose M Monsalve Diaz printf(" Max Threads Per SMP: \t\t%d \n", TmpInt); 1366d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1367d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT, Device), 1368d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1369d2f85d09SJose M Monsalve Diaz printf(" Async Engines: \t\t\t%s (%d) \n", BOOL2TEXT(TmpInt), TmpInt); 1370d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1371d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING, Device), 1372d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1373d2f85d09SJose M Monsalve Diaz printf(" Unified Addressing: \t\t%s \n", BOOL2TEXT(TmpInt)); 137428939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 137528939b6aSShilei Tian &TmpInt, CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY, Device), 1376d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1377d2f85d09SJose M Monsalve Diaz printf(" Managed Memory: \t\t\t%s \n", BOOL2TEXT(TmpInt)); 1378d2f85d09SJose M Monsalve Diaz checkResult( 1379d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute( 1380d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS, Device), 1381d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1382d2f85d09SJose M Monsalve Diaz printf(" Concurrent Managed Memory: \t\t%s \n", BOOL2TEXT(TmpInt)); 1383d2f85d09SJose M Monsalve Diaz checkResult( 1384d2f85d09SJose M Monsalve Diaz cuDeviceGetAttribute( 1385d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED, Device), 1386d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1387d2f85d09SJose M Monsalve Diaz printf(" Preemption Supported: \t\t%s \n", BOOL2TEXT(TmpInt)); 1388d2f85d09SJose M Monsalve Diaz checkResult(cuDeviceGetAttribute( 1389d2f85d09SJose M Monsalve Diaz &TmpInt, CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH, Device), 1390d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1391d2f85d09SJose M Monsalve Diaz printf(" Cooperative Launch: \t\t%s \n", BOOL2TEXT(TmpInt)); 139228939b6aSShilei Tian checkResult(cuDeviceGetAttribute( 139328939b6aSShilei Tian &TmpInt, CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD, Device), 1394d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1395d2f85d09SJose M Monsalve Diaz printf(" Multi-Device Boars: \t\t%s \n", BOOL2TEXT(TmpInt)); 139628939b6aSShilei Tian checkResult( 139728939b6aSShilei Tian cuDeviceGetAttribute( 139828939b6aSShilei Tian &TmpInt, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, Device), 1399d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 140028939b6aSShilei Tian checkResult( 140128939b6aSShilei Tian cuDeviceGetAttribute( 140228939b6aSShilei Tian &TmpInt2, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, Device), 1403d2f85d09SJose M Monsalve Diaz "Error returned from cuDeviceGetAttribute\n"); 1404d2f85d09SJose M Monsalve Diaz printf(" Compute Capabilities: \t\t%d%d \n", TmpInt, TmpInt2); 1405d2f85d09SJose M Monsalve Diaz } 140629df4ab3SShilei Tian 140729df4ab3SShilei Tian int waitEvent(const int DeviceId, __tgt_async_info *AsyncInfo, 140829df4ab3SShilei Tian void *EventPtr) const { 140929df4ab3SShilei Tian CUstream Stream = getStream(DeviceId, AsyncInfo); 141029df4ab3SShilei Tian CUevent Event = reinterpret_cast<CUevent>(EventPtr); 141129df4ab3SShilei Tian 141229df4ab3SShilei Tian // We don't use CU_EVENT_WAIT_DEFAULT here as it is only available from 141329df4ab3SShilei Tian // specific CUDA version, and defined as 0x0. In previous version, per CUDA 141429df4ab3SShilei Tian // API document, that argument has to be 0x0. 141529df4ab3SShilei Tian CUresult Err = cuStreamWaitEvent(Stream, Event, 0); 141629df4ab3SShilei Tian if (Err != CUDA_SUCCESS) { 141729df4ab3SShilei Tian DP("Error when waiting event. stream = " DPxMOD ", event = " DPxMOD "\n", 141829df4ab3SShilei Tian DPxPTR(Stream), DPxPTR(Event)); 141929df4ab3SShilei Tian CUDA_ERR_STRING(Err); 142029df4ab3SShilei Tian return OFFLOAD_FAIL; 142129df4ab3SShilei Tian } 142229df4ab3SShilei Tian 142329df4ab3SShilei Tian return OFFLOAD_SUCCESS; 142429df4ab3SShilei Tian } 14254031bb98SShilei Tian }; 14264031bb98SShilei Tian 14274031bb98SShilei Tian DeviceRTLTy DeviceRTL; 14284031bb98SShilei Tian } // namespace 14294031bb98SShilei Tian 14304031bb98SShilei Tian // Exposed library API function 14314031bb98SShilei Tian #ifdef __cplusplus 14324031bb98SShilei Tian extern "C" { 14334031bb98SShilei Tian #endif 14344031bb98SShilei Tian 14354031bb98SShilei Tian int32_t __tgt_rtl_is_valid_binary(__tgt_device_image *image) { 14364031bb98SShilei Tian return elf_check_machine(image, /* EM_CUDA */ 190); 14374031bb98SShilei Tian } 14384031bb98SShilei Tian 14394031bb98SShilei Tian int32_t __tgt_rtl_number_of_devices() { return DeviceRTL.getNumOfDevices(); } 14404031bb98SShilei Tian 14414031bb98SShilei Tian int64_t __tgt_rtl_init_requires(int64_t RequiresFlags) { 144240470eb2SGeorge Rokos DP("Init requires flags to %" PRId64 "\n", RequiresFlags); 14434031bb98SShilei Tian DeviceRTL.setRequiresFlag(RequiresFlags); 14444031bb98SShilei Tian return RequiresFlags; 14454031bb98SShilei Tian } 14464031bb98SShilei Tian 1447a014fbbcSShilei Tian int32_t __tgt_rtl_is_data_exchangable(int32_t src_dev_id, int dst_dev_id) { 1448a014fbbcSShilei Tian if (DeviceRTL.isValidDeviceId(src_dev_id) && 1449a014fbbcSShilei Tian DeviceRTL.isValidDeviceId(dst_dev_id)) 1450a014fbbcSShilei Tian return 1; 1451a014fbbcSShilei Tian 1452a014fbbcSShilei Tian return 0; 1453a014fbbcSShilei Tian } 1454a014fbbcSShilei Tian 14554031bb98SShilei Tian int32_t __tgt_rtl_init_device(int32_t device_id) { 14564031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 14574031bb98SShilei Tian 14584031bb98SShilei Tian return DeviceRTL.initDevice(device_id); 14594031bb98SShilei Tian } 14604031bb98SShilei Tian 14614031bb98SShilei Tian __tgt_target_table *__tgt_rtl_load_binary(int32_t device_id, 14624031bb98SShilei Tian __tgt_device_image *image) { 14634031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 14644031bb98SShilei Tian 14654031bb98SShilei Tian return DeviceRTL.loadBinary(device_id, image); 14664031bb98SShilei Tian } 14674031bb98SShilei Tian 14682468fdd9SGeorge Rokos void *__tgt_rtl_data_alloc(int32_t device_id, int64_t size, void *, 14692468fdd9SGeorge Rokos int32_t kind) { 14704031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 14714031bb98SShilei Tian 1472a15f8589SJoseph Huber return DeviceRTL.dataAlloc(device_id, size, (TargetAllocTy)kind); 14734031bb98SShilei Tian } 14744031bb98SShilei Tian 14754031bb98SShilei Tian int32_t __tgt_rtl_data_submit(int32_t device_id, void *tgt_ptr, void *hst_ptr, 14764031bb98SShilei Tian int64_t size) { 14774031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 14784031bb98SShilei Tian 14795449fbb5SJohannes Doerfert __tgt_async_info AsyncInfo; 14804031bb98SShilei Tian const int32_t rc = __tgt_rtl_data_submit_async(device_id, tgt_ptr, hst_ptr, 14815449fbb5SJohannes Doerfert size, &AsyncInfo); 14824031bb98SShilei Tian if (rc != OFFLOAD_SUCCESS) 14834031bb98SShilei Tian return OFFLOAD_FAIL; 14844031bb98SShilei Tian 14855449fbb5SJohannes Doerfert return __tgt_rtl_synchronize(device_id, &AsyncInfo); 14864031bb98SShilei Tian } 14874031bb98SShilei Tian 14884031bb98SShilei Tian int32_t __tgt_rtl_data_submit_async(int32_t device_id, void *tgt_ptr, 14894031bb98SShilei Tian void *hst_ptr, int64_t size, 14904031bb98SShilei Tian __tgt_async_info *async_info_ptr) { 14914031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 14924031bb98SShilei Tian assert(async_info_ptr && "async_info_ptr is nullptr"); 14934031bb98SShilei Tian 14944031bb98SShilei Tian return DeviceRTL.dataSubmit(device_id, tgt_ptr, hst_ptr, size, 14954031bb98SShilei Tian async_info_ptr); 14964031bb98SShilei Tian } 14974031bb98SShilei Tian 14984031bb98SShilei Tian int32_t __tgt_rtl_data_retrieve(int32_t device_id, void *hst_ptr, void *tgt_ptr, 14994031bb98SShilei Tian int64_t size) { 15004031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 15014031bb98SShilei Tian 15025449fbb5SJohannes Doerfert __tgt_async_info AsyncInfo; 15034031bb98SShilei Tian const int32_t rc = __tgt_rtl_data_retrieve_async(device_id, hst_ptr, tgt_ptr, 15045449fbb5SJohannes Doerfert size, &AsyncInfo); 15054031bb98SShilei Tian if (rc != OFFLOAD_SUCCESS) 15064031bb98SShilei Tian return OFFLOAD_FAIL; 15074031bb98SShilei Tian 15085449fbb5SJohannes Doerfert return __tgt_rtl_synchronize(device_id, &AsyncInfo); 15094031bb98SShilei Tian } 15104031bb98SShilei Tian 15114031bb98SShilei Tian int32_t __tgt_rtl_data_retrieve_async(int32_t device_id, void *hst_ptr, 15124031bb98SShilei Tian void *tgt_ptr, int64_t size, 15134031bb98SShilei Tian __tgt_async_info *async_info_ptr) { 15144031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 15154031bb98SShilei Tian assert(async_info_ptr && "async_info_ptr is nullptr"); 15164031bb98SShilei Tian 15174031bb98SShilei Tian return DeviceRTL.dataRetrieve(device_id, hst_ptr, tgt_ptr, size, 15184031bb98SShilei Tian async_info_ptr); 15194031bb98SShilei Tian } 15204031bb98SShilei Tian 1521a014fbbcSShilei Tian int32_t __tgt_rtl_data_exchange_async(int32_t src_dev_id, void *src_ptr, 1522a014fbbcSShilei Tian int dst_dev_id, void *dst_ptr, 1523a014fbbcSShilei Tian int64_t size, 15245449fbb5SJohannes Doerfert __tgt_async_info *AsyncInfo) { 1525a014fbbcSShilei Tian assert(DeviceRTL.isValidDeviceId(src_dev_id) && "src_dev_id is invalid"); 1526a014fbbcSShilei Tian assert(DeviceRTL.isValidDeviceId(dst_dev_id) && "dst_dev_id is invalid"); 15275449fbb5SJohannes Doerfert assert(AsyncInfo && "AsyncInfo is nullptr"); 1528a014fbbcSShilei Tian 1529a014fbbcSShilei Tian return DeviceRTL.dataExchange(src_dev_id, src_ptr, dst_dev_id, dst_ptr, size, 15305449fbb5SJohannes Doerfert AsyncInfo); 1531a014fbbcSShilei Tian } 1532a014fbbcSShilei Tian 1533a014fbbcSShilei Tian int32_t __tgt_rtl_data_exchange(int32_t src_dev_id, void *src_ptr, 1534a014fbbcSShilei Tian int32_t dst_dev_id, void *dst_ptr, 1535a014fbbcSShilei Tian int64_t size) { 1536a014fbbcSShilei Tian assert(DeviceRTL.isValidDeviceId(src_dev_id) && "src_dev_id is invalid"); 1537a014fbbcSShilei Tian assert(DeviceRTL.isValidDeviceId(dst_dev_id) && "dst_dev_id is invalid"); 1538a014fbbcSShilei Tian 15395449fbb5SJohannes Doerfert __tgt_async_info AsyncInfo; 1540a014fbbcSShilei Tian const int32_t rc = __tgt_rtl_data_exchange_async( 15415449fbb5SJohannes Doerfert src_dev_id, src_ptr, dst_dev_id, dst_ptr, size, &AsyncInfo); 1542a014fbbcSShilei Tian if (rc != OFFLOAD_SUCCESS) 1543a014fbbcSShilei Tian return OFFLOAD_FAIL; 1544a014fbbcSShilei Tian 15455449fbb5SJohannes Doerfert return __tgt_rtl_synchronize(src_dev_id, &AsyncInfo); 1546a014fbbcSShilei Tian } 1547a014fbbcSShilei Tian 15484031bb98SShilei Tian int32_t __tgt_rtl_data_delete(int32_t device_id, void *tgt_ptr) { 15494031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 15504031bb98SShilei Tian 15514031bb98SShilei Tian return DeviceRTL.dataDelete(device_id, tgt_ptr); 15524031bb98SShilei Tian } 15534031bb98SShilei Tian 15544031bb98SShilei Tian int32_t __tgt_rtl_run_target_team_region(int32_t device_id, void *tgt_entry_ptr, 15554031bb98SShilei Tian void **tgt_args, 15564031bb98SShilei Tian ptrdiff_t *tgt_offsets, 15574031bb98SShilei Tian int32_t arg_num, int32_t team_num, 15584031bb98SShilei Tian int32_t thread_limit, 15594031bb98SShilei Tian uint64_t loop_tripcount) { 15604031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 15614031bb98SShilei Tian 15625449fbb5SJohannes Doerfert __tgt_async_info AsyncInfo; 15634031bb98SShilei Tian const int32_t rc = __tgt_rtl_run_target_team_region_async( 15644031bb98SShilei Tian device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num, 15655449fbb5SJohannes Doerfert thread_limit, loop_tripcount, &AsyncInfo); 15664031bb98SShilei Tian if (rc != OFFLOAD_SUCCESS) 15674031bb98SShilei Tian return OFFLOAD_FAIL; 15684031bb98SShilei Tian 15695449fbb5SJohannes Doerfert return __tgt_rtl_synchronize(device_id, &AsyncInfo); 15704031bb98SShilei Tian } 15714031bb98SShilei Tian 15724031bb98SShilei Tian int32_t __tgt_rtl_run_target_team_region_async( 15734031bb98SShilei Tian int32_t device_id, void *tgt_entry_ptr, void **tgt_args, 15744031bb98SShilei Tian ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t team_num, 15754031bb98SShilei Tian int32_t thread_limit, uint64_t loop_tripcount, 15764031bb98SShilei Tian __tgt_async_info *async_info_ptr) { 15774031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 15784031bb98SShilei Tian 15794031bb98SShilei Tian return DeviceRTL.runTargetTeamRegion( 15804031bb98SShilei Tian device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num, 15814031bb98SShilei Tian thread_limit, loop_tripcount, async_info_ptr); 15824031bb98SShilei Tian } 15833de4cd12SGeorge Rokos 15843de4cd12SGeorge Rokos int32_t __tgt_rtl_run_target_region(int32_t device_id, void *tgt_entry_ptr, 158532ed2927SShilei Tian void **tgt_args, ptrdiff_t *tgt_offsets, 158603ff643dSShilei Tian int32_t arg_num) { 15874031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 15884031bb98SShilei Tian 15895449fbb5SJohannes Doerfert __tgt_async_info AsyncInfo; 15904031bb98SShilei Tian const int32_t rc = __tgt_rtl_run_target_region_async( 15915449fbb5SJohannes Doerfert device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, &AsyncInfo); 159203ff643dSShilei Tian if (rc != OFFLOAD_SUCCESS) 159303ff643dSShilei Tian return OFFLOAD_FAIL; 159403ff643dSShilei Tian 15955449fbb5SJohannes Doerfert return __tgt_rtl_synchronize(device_id, &AsyncInfo); 159603ff643dSShilei Tian } 159703ff643dSShilei Tian 159803ff643dSShilei Tian int32_t __tgt_rtl_run_target_region_async(int32_t device_id, 159903ff643dSShilei Tian void *tgt_entry_ptr, void **tgt_args, 160003ff643dSShilei Tian ptrdiff_t *tgt_offsets, 160132ed2927SShilei Tian int32_t arg_num, 16024031bb98SShilei Tian __tgt_async_info *async_info_ptr) { 16034031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 16044031bb98SShilei Tian 160503ff643dSShilei Tian return __tgt_rtl_run_target_team_region_async( 16064031bb98SShilei Tian device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, 16074031bb98SShilei Tian /* team num*/ 1, /* thread_limit */ 1, /* loop_tripcount */ 0, 16084031bb98SShilei Tian async_info_ptr); 160932ed2927SShilei Tian } 161032ed2927SShilei Tian 16114031bb98SShilei Tian int32_t __tgt_rtl_synchronize(int32_t device_id, 16124031bb98SShilei Tian __tgt_async_info *async_info_ptr) { 16134031bb98SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 16144031bb98SShilei Tian assert(async_info_ptr && "async_info_ptr is nullptr"); 16154031bb98SShilei Tian assert(async_info_ptr->Queue && "async_info_ptr->Queue is nullptr"); 161632ed2927SShilei Tian 16174031bb98SShilei Tian return DeviceRTL.synchronize(device_id, async_info_ptr); 16183de4cd12SGeorge Rokos } 16193de4cd12SGeorge Rokos 16202b6f2008SJoseph Huber void __tgt_rtl_set_info_flag(uint32_t NewInfoLevel) { 162159b68490SJoseph Huber std::atomic<uint32_t> &InfoLevel = getInfoLevelInternal(); 16222b6f2008SJoseph Huber InfoLevel.store(NewInfoLevel); 16232b6f2008SJoseph Huber } 16242b6f2008SJoseph Huber 1625d2f85d09SJose M Monsalve Diaz void __tgt_rtl_print_device_info(int32_t device_id) { 1626d2f85d09SJose M Monsalve Diaz assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 1627d2f85d09SJose M Monsalve Diaz DeviceRTL.printDeviceInfo(device_id); 1628d2f85d09SJose M Monsalve Diaz } 1629d2f85d09SJose M Monsalve Diaz 163029df4ab3SShilei Tian int32_t __tgt_rtl_create_event(int32_t device_id, void **event) { 163129df4ab3SShilei Tian assert(event && "event is nullptr"); 163229df4ab3SShilei Tian return createEvent(event); 163329df4ab3SShilei Tian } 163429df4ab3SShilei Tian 163529df4ab3SShilei Tian int32_t __tgt_rtl_record_event(int32_t device_id, void *event_ptr, 163629df4ab3SShilei Tian __tgt_async_info *async_info_ptr) { 163729df4ab3SShilei Tian assert(async_info_ptr && "async_info_ptr is nullptr"); 163829df4ab3SShilei Tian assert(async_info_ptr->Queue && "async_info_ptr->Queue is nullptr"); 163929df4ab3SShilei Tian assert(event_ptr && "event_ptr is nullptr"); 164029df4ab3SShilei Tian 164129df4ab3SShilei Tian return recordEvent(event_ptr, async_info_ptr); 164229df4ab3SShilei Tian } 164329df4ab3SShilei Tian 164429df4ab3SShilei Tian int32_t __tgt_rtl_wait_event(int32_t device_id, void *event_ptr, 164529df4ab3SShilei Tian __tgt_async_info *async_info_ptr) { 164629df4ab3SShilei Tian assert(DeviceRTL.isValidDeviceId(device_id) && "device_id is invalid"); 164729df4ab3SShilei Tian assert(async_info_ptr && "async_info_ptr is nullptr"); 164829df4ab3SShilei Tian assert(event_ptr && "event is nullptr"); 164929df4ab3SShilei Tian 165029df4ab3SShilei Tian return DeviceRTL.waitEvent(device_id, async_info_ptr, event_ptr); 165129df4ab3SShilei Tian } 165229df4ab3SShilei Tian 165329df4ab3SShilei Tian int32_t __tgt_rtl_sync_event(int32_t device_id, void *event_ptr) { 165429df4ab3SShilei Tian assert(event_ptr && "event is nullptr"); 165529df4ab3SShilei Tian 165629df4ab3SShilei Tian return syncEvent(event_ptr); 165729df4ab3SShilei Tian } 165829df4ab3SShilei Tian 165929df4ab3SShilei Tian int32_t __tgt_rtl_destroy_event(int32_t device_id, void *event_ptr) { 166029df4ab3SShilei Tian assert(event_ptr && "event is nullptr"); 166129df4ab3SShilei Tian 166229df4ab3SShilei Tian return destroyEvent(event_ptr); 166329df4ab3SShilei Tian } 166429df4ab3SShilei Tian 16653de4cd12SGeorge Rokos #ifdef __cplusplus 16663de4cd12SGeorge Rokos } 16673de4cd12SGeorge Rokos #endif 1668