Lines Matching defs:DeviceNum

42 void *targetAllocExplicit(size_t Size, int DeviceNum, int Kind,
44 void targetFreeExplicit(void *DevicePtr, int DeviceNum, int Kind,
46 void *targetLockExplicit(void *HostPtr, size_t Size, int DeviceNum,
48 void targetUnlockExplicit(void *HostPtr, int DeviceNum, const char *Name);
101 EXTERN void *omp_target_alloc(size_t Size, int DeviceNum) {
102   TIMESCOPE_WITH_DETAILS("dst_dev=" + std::to_string(DeviceNum) +
105   return targetAllocExplicit(Size, DeviceNum, TARGET_ALLOC_DEFAULT, __func__);
108 EXTERN void *llvm_omp_target_alloc_device(size_t Size, int DeviceNum) {
110   return targetAllocExplicit(Size, DeviceNum, TARGET_ALLOC_DEVICE, __func__);
113 EXTERN void *llvm_omp_target_alloc_host(size_t Size, int DeviceNum) {
115   return targetAllocExplicit(Size, DeviceNum, TARGET_ALLOC_HOST, __func__);
118 EXTERN void *llvm_omp_target_alloc_shared(size_t Size, int DeviceNum) {
120   return targetAllocExplicit(Size, DeviceNum, TARGET_ALLOC_SHARED, __func__);
123 EXTERN void omp_target_free(void *Ptr, int DeviceNum) {
126   return targetFreeExplicit(Ptr, DeviceNum, TARGET_ALLOC_DEFAULT, __func__);
129 EXTERN void llvm_omp_target_free_device(void *Ptr, int DeviceNum) {
131   return targetFreeExplicit(Ptr, DeviceNum, TARGET_ALLOC_DEVICE, __func__);
134 EXTERN void llvm_omp_target_free_host(void *Ptr, int DeviceNum) {
136   return targetFreeExplicit(Ptr, DeviceNum, TARGET_ALLOC_HOST, __func__);
139 EXTERN void llvm_omp_target_free_shared(void *Ptre, int DeviceNum) {
141   return targetFreeExplicit(Ptre, DeviceNum, TARGET_ALLOC_SHARED, __func__);
155                                                     int DeviceNum) {
157   return targetLockExplicit(Ptr, Size, DeviceNum, __func__);
160 EXTERN void llvm_omp_target_unlock_mem(void *Ptr, int DeviceNum) {
162   targetUnlockExplicit(Ptr, DeviceNum, __func__);
165 EXTERN int omp_target_is_present(const void *Ptr, int DeviceNum) {
169      DeviceNum, DPxPTR(Ptr));
176   if (DeviceNum == omp_get_initial_device()) {
181   auto DeviceOrErr = PM->getDevice(DeviceNum);
183     FATAL_MESSAGE(DeviceNum, "%s", toString(DeviceOrErr.takeError()).c_str());
327   omp_target_memset(Args->Ptr, Args->C, Args->N, Args->DeviceNum);
379                                int DeviceNum) {
383      DeviceNum, Ptr, NumBytes);
391   if (DeviceNum == omp_get_initial_device()) {
405       (void)omp_target_memcpy(Ptr, Shadow, NumBytes, 0, 0, DeviceNum,
422                                      int DeviceNum, int DepObjCount,
426      DeviceNum, Ptr, NumBytes);
434   auto *Args = new TargetMemsetArgsTy{Ptr, ByteVal, NumBytes, DeviceNum};
578                                     int DeviceNum) {
583      DPxPTR(HostPtr), DPxPTR(DevicePtr), Size, DeviceOffset, DeviceNum);
590   if (DeviceNum == omp_get_initial_device()) {
595   auto DeviceOrErr = PM->getDevice(DeviceNum);
597     FATAL_MESSAGE(DeviceNum, "%s", toString(DeviceOrErr.takeError()).c_str());
602       RegionInterface.getCallbacks<ompt_target_data_associate>(), DeviceNum,
612 EXTERN int omp_target_disassociate_ptr(const void *HostPtr, int DeviceNum) {
617      DPxPTR(HostPtr), DeviceNum);
624   if (DeviceNum == omp_get_initial_device()) {
630   auto DeviceOrErr = PM->getDevice(DeviceNum);
632     FATAL_MESSAGE(DeviceNum, "%s", toString(DeviceOrErr.takeError()).c_str());
635       RegionInterface.getCallbacks<ompt_target_data_disassociate>(), DeviceNum,
645 EXTERN void *omp_get_mapped_ptr(const void *Ptr, int DeviceNum) {
649      DPxPTR(Ptr), DeviceNum);
657   if (DeviceNum == NumDevices) {
659            DeviceNum, DPxPTR(Ptr));
663   if (NumDevices <= DeviceNum) {
664     DP("DeviceNum %d is invalid, returning nullptr.\n", DeviceNum);
668   auto DeviceOrErr = PM->getDevice(DeviceNum);
670     FATAL_MESSAGE(DeviceNum, "%s", toString(DeviceOrErr.takeError()).c_str());
678        DPxPTR(Ptr), DeviceNum);