| /external/llvm-project/openmp/libomptarget/src/ |
| D | MemoryManager.cpp | 104 void *MemoryManagerTy::allocateOnDevice(size_t Size, void *HstPtr) const { in allocateOnDevice()
105 return Device.RTL->data_alloc(Device.RTLDeviceID, Size, HstPtr); in allocateOnDevice()
112 void *MemoryManagerTy::freeAndAllocate(size_t Size, void *HstPtr) { in freeAndAllocate() argument
136 return allocateOnDevice(Size, HstPtr); in freeAndAllocate()
140 void *HstPtr) { in allocateOrFreeAndAllocateOnDevice() argument
141 void *TgtPtr = allocateOnDevice(Size, HstPtr); in allocateOrFreeAndAllocateOnDevice()
147 TgtPtr = freeAndAllocate(Size, HstPtr); in allocateOrFreeAndAllocateOnDevice()
159 void *MemoryManagerTy::allocate(size_t Size, void *HstPtr) { in allocate() argument
166 Size, DPxPTR(HstPtr)); in allocate()
174 void *TgtPtr = allocateOrFreeAndAllocateOnDevice(Size, HstPtr); in allocate()
[all …]
|
| D | MemoryManager.h | 63 void *allocateOnDevice(size_t Size, void *HstPtr) const;
71 void *freeAndAllocate(size_t Size, void *HstPtr);
77 void *allocateOrFreeAndAllocateOnDevice(size_t Size, void *HstPtr);
89 void *allocate(size_t Size, void *HstPtr);
|
| D | omptarget.cpp | 439 DeallocTgtPtrInfo(void *HstPtr, int64_t Size, bool ForceDelete, in DeallocTgtPtrInfo()
441 : HstPtrBegin(HstPtr), DataSize(Size), ForceDelete(ForceDelete), in DeallocTgtPtrInfo()
904 FirstPrivateArgInfoTy(int Index, const void *HstPtr, int64_t Size, in FirstPrivateArgInfoTy()
906 : Index(Index), HstPtrBegin(reinterpret_cast<const char *>(HstPtr)), in FirstPrivateArgInfoTy()
937 int addArg(void *HstPtr, int64_t ArgSize, int64_t ArgOffset, in addArg() argument
944 TgtPtr = Device.allocData(ArgSize, HstPtr); in addArg()
947 (IsFirstPrivate ? "first-" : ""), DPxPTR(HstPtr)); in addArg()
956 DPxPTR(HstPtr), DPxPTR(TgtPtrBase)); in addArg()
960 int Ret = Device.submitData(TgtPtr, HstPtr, ArgSize, AsyncInfo); in addArg()
969 DPxPTR(HstPtr), ArgSize); in addArg()
[all …]
|
| D | device.cpp | 414 void *DeviceTy::allocData(int64_t Size, void *HstPtr) { in allocData() argument
417 return MemoryManager->allocate(Size, HstPtr); in allocData()
419 return RTL->data_alloc(RTLDeviceID, Size, HstPtr); in allocData()
|
| D | device.h | 202 void *allocData(int64_t Size, void *HstPtr = nullptr);
|
| /external/llvm-project/openmp/libomptarget/plugins/amdgpu/src/ |
| D | rtl.cpp | 568 int32_t dataRetrieve(int32_t DeviceId, void *HstPtr, void *TgtPtr, int64_t Size, in dataRetrieve() argument
573 if (!HstPtr) in dataRetrieve()
578 (long long unsigned)(Elf64_Addr)HstPtr); in dataRetrieve()
580 err = DeviceInfo.freesignalpool_memcpy_d2h(HstPtr, TgtPtr, (size_t)Size, in dataRetrieve()
586 (Elf64_Addr)HstPtr, (Elf64_Addr)TgtPtr, (unsigned long long)Size); in dataRetrieve()
591 (long long unsigned)(Elf64_Addr)HstPtr); in dataRetrieve()
595 int32_t dataSubmit(int32_t DeviceId, void *TgtPtr, void *HstPtr, int64_t Size, in dataSubmit() argument
601 if (!HstPtr) in dataSubmit()
605 (long long unsigned)(Elf64_Addr)HstPtr, in dataSubmit()
607 err = DeviceInfo.freesignalpool_memcpy_h2d(TgtPtr, HstPtr, (size_t)Size, in dataSubmit()
[all …]
|
| /external/llvm-project/openmp/libomptarget/plugins/cuda/src/ |
| D | rtl.cpp | 738 int dataSubmit(const int DeviceId, const void *TgtPtr, const void *HstPtr, in dataSubmit() argument
748 Err = cuMemcpyHtoDAsync((CUdeviceptr)TgtPtr, HstPtr, Size, Stream); in dataSubmit()
752 DPxPTR(HstPtr), DPxPTR(TgtPtr), Size); in dataSubmit()
760 int dataRetrieve(const int DeviceId, void *HstPtr, const void *TgtPtr, in dataRetrieve() argument
770 Err = cuMemcpyDtoHAsync(HstPtr, (CUdeviceptr)TgtPtr, Size, Stream); in dataRetrieve()
774 DPxPTR(HstPtr), DPxPTR(TgtPtr), Size); in dataRetrieve()
|