Home
last modified time | relevance | path

Searched refs:DeviceId (Results 1 – 16 of 16) sorted by relevance

/llvm-project-15.0.7/openmp/libomptarget/plugins/cuda/src/
H A Drtl.cpp585 for (int DeviceId = 0; DeviceId < NumberOfDevices; ++DeviceId) in ~DeviceRTLTy() local
586 deinitDevice(DeviceId); in ~DeviceRTLTy()
591 return DeviceId >= 0 && DeviceId < NumberOfDevices; in isValidDeviceId()
642 if (!StreamPool[DeviceId]) in initDevice()
647 if (!EventPool[DeviceId]) in initDevice()
648 EventPool[DeviceId] = in initDevice()
733 DeviceData[DeviceId].ThreadsPerBlock, DeviceData[DeviceId].WarpSize); in initDevice()
750 if (DeviceData[DeviceId].NumTeams > DeviceData[DeviceId].BlocksPerGrid) { in initDevice()
753 DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].BlocksPerGrid; in initDevice()
764 DeviceData[DeviceId].NumThreads = DeviceData[DeviceId].ThreadsPerBlock; in initDevice()
[all …]
/llvm-project-15.0.7/openmp/libomptarget/src/
H A DLegacyAPI.cpp16 EXTERN void __tgt_target_data_begin(int64_t DeviceId, int32_t ArgNum, in __tgt_target_data_begin() argument
36 EXTERN void __tgt_target_data_end(int64_t DeviceId, int32_t ArgNum, in __tgt_target_data_end() argument
44 EXTERN void __tgt_target_data_update(int64_t DeviceId, int32_t ArgNum, in __tgt_target_data_update() argument
53 int64_t DeviceId, int32_t ArgNum, void **ArgsBase, void **Args, in __tgt_target_data_update_nowait() argument
103 ident_t *Loc, int64_t DeviceId, void *HostPtr, int32_t ArgNum, in __tgt_target_nowait_mapper() argument
113 EXTERN int __tgt_target_teams_mapper(ident_t *Loc, int64_t DeviceId, in __tgt_target_teams_mapper() argument
138 EXTERN int __tgt_target_teams_nowait(int64_t DeviceId, void *HostPtr, in __tgt_target_teams_nowait() argument
153 ident_t *Loc, int64_t DeviceId, void *HostPtr, int32_t ArgNum, in __tgt_target_teams_nowait_mapper() argument
168 if (checkDeviceAndCtors(DeviceId, Loc)) { in __kmpc_push_target_tripcount_mapper()
169 DP("Not offloading to device %" PRId64 "\n", DeviceId); in __kmpc_push_target_tripcount_mapper()
[all …]
H A Dinterface.cpp75 DeviceId, ArgNum); in __tgt_target_data_begin_mapper()
76 if (checkDeviceAndCtors(DeviceId, Loc)) { in __tgt_target_data_begin_mapper()
81 DeviceTy &Device = *PM->Devices[DeviceId]; in __tgt_target_data_begin_mapper()
125 if (checkDeviceAndCtors(DeviceId, Loc)) { in __tgt_target_data_end_mapper()
130 DeviceTy &Device = *PM->Devices[DeviceId]; in __tgt_target_data_end_mapper()
171 if (checkDeviceAndCtors(DeviceId, Loc)) { in __tgt_target_data_update_mapper()
180 DeviceTy &Device = *PM->Devices[DeviceId]; in __tgt_target_data_update_mapper()
217 DPxPTR(HostPtr), DeviceId); in __tgt_target_kernel()
221 if (checkDeviceAndCtors(DeviceId, Loc)) { in __tgt_target_kernel()
245 DeviceTy &Device = *PM->Devices[DeviceId]; in __tgt_target_kernel()
[all …]
H A Dinterop.cpp193 if (DeviceId == -1) { in __tgt_interop_init()
194 DeviceId = omp_get_default_device(); in __tgt_interop_init()
203 if (!deviceIsReady(DeviceId)) { in __tgt_interop_init()
232 if (DeviceId == -1) { in __tgt_interop_use()
233 DeviceId = omp_get_default_device(); in __tgt_interop_use()
237 assert((DeviceId == -1 || InteropVal->device_id == DeviceId) && in __tgt_interop_use()
240 if (!deviceIsReady(DeviceId)) { in __tgt_interop_use()
260 if (DeviceId == -1) { in __tgt_interop_destroy()
261 DeviceId = omp_get_default_device(); in __tgt_interop_destroy()
267 assert((DeviceId == -1 || InteropVal->device_id == DeviceId) && in __tgt_interop_destroy()
[all …]
H A Domptarget.cpp77 int32_t DeviceId = Device.DeviceID; in initLibrary() local
116 DeviceId); in initLibrary()
117 TransTable->TargetsImages[DeviceId] = 0; in initLibrary()
131 DeviceId, Hsize, Tsize); in initLibrary()
132 TransTable->TargetsImages[DeviceId] = 0; in initLibrary()
133 TransTable->TargetsTable[DeviceId] = 0; in initLibrary()
1133 uint64_t getLoopTripCount(int64_t DeviceId) { in getLoopTripCount() argument
1134 DeviceTy &Device = *PM->Devices[DeviceId]; in getLoopTripCount()
1331 DeviceTy &Device = *PM->Devices[DeviceId]; in processDataBefore()
1460 DeviceTy &Device = *PM->Devices[DeviceId]; in processDataAfter()
[all …]
H A Drtl.cpp375 for (int32_t DeviceId = 0; DeviceId < R.NumberOfDevices; DeviceId++) { in initRTLonce() local
378 PM->Devices[Start + DeviceId]->DeviceID = Start + DeviceId; in initRTLonce()
380 PM->Devices[Start + DeviceId]->RTLDeviceID = DeviceId; in initRTLonce()
H A Dprivate.h156 printKernelArguments(const ident_t *Loc, const int64_t DeviceId, in printKernelArguments() argument
161 INFO(OMP_INFOTYPE_ALL, DeviceId, "%s at %s:%d:%d with %d arguments:\n", in printKernelArguments()
185 INFO(OMP_INFOTYPE_ALL, DeviceId, "%s(%s)[%" PRId64 "] %s\n", Type, in printKernelArguments()
/llvm-project-15.0.7/openmp/libomptarget/plugins/amdgpu/src/
H A Drtl.cpp63 DeviceId); in hostrpc_assign_buffer()
220 int32_t DeviceId; member
229 DeviceId(DeviceId), CallStackAddr(CallStackAddr), Name(Name) { in KernelTy()
886 for (unsigned int DeviceId = 0; DeviceId < Agents.size(); DeviceId++) { in setupDevicePools() local
951 assert(DeviceId >= 0 && DeviceId < DeviceCoarseGrainedMemoryPools.size() && in getDeviceMemoryPool()
1467 DeviceId); in runRegionLocked()
2038 for (int32_t DeviceId = 0; DeviceId < NumberOfDevices; ++DeviceId) { in __tgt_rtl_is_valid_binary_info() local
2107 DeviceInfo().ComputeUnits[DeviceId], DeviceInfo().GPUName[DeviceId].c_str()); in __tgt_rtl_init_device()
2216 DeviceId, DeviceInfo().GroupsPerDevice[DeviceId], in __tgt_rtl_init_device()
2220 DeviceInfo().WarpSize[DeviceId], DeviceInfo().ThreadsPerGroup[DeviceId], in __tgt_rtl_init_device()
[all …]
/llvm-project-15.0.7/openmp/libomptarget/plugins/remote/src/
H A DClient.cpp172 Request->set_number(DeviceId); in initDevice()
280 Request->set_device_id(DeviceId); in dataAlloc()
381 Request->set_device_id(DeviceId); in dataRetrieve()
417 DeviceId) in dataRetrieve()
467 Request->set_device_id(DeviceId); in dataDelete()
479 DeviceId) in dataDelete()
497 Request->set_device_id(DeviceId); in runTargetRegion()
567 DeviceId) in runTargetTeamRegion()
620 if (DeviceId < Devices[ClientIdx]) in mapDeviceId()
621 return {ClientIdx, DeviceId}; in mapDeviceId()
[all …]
H A Drtl.cpp58 int32_t __tgt_rtl_init_device(int32_t DeviceId) { in __tgt_rtl_init_device() argument
59 return Manager->initDevice(DeviceId); in __tgt_rtl_init_device()
66 __tgt_target_table *__tgt_rtl_load_binary(int32_t DeviceId, in __tgt_rtl_load_binary() argument
68 return Manager->loadBinary(DeviceId, (__tgt_device_image *)Image); in __tgt_rtl_load_binary()
75 void *__tgt_rtl_data_alloc(int32_t DeviceId, int64_t Size, void *HstPtr, in __tgt_rtl_data_alloc() argument
83 return Manager->dataAlloc(DeviceId, Size, HstPtr); in __tgt_rtl_data_alloc()
88 return Manager->dataSubmit(DeviceId, TgtPtr, HstPtr, Size); in __tgt_rtl_data_submit()
93 return Manager->dataRetrieve(DeviceId, HstPtr, TgtPtr, Size); in __tgt_rtl_data_retrieve()
96 int32_t __tgt_rtl_data_delete(int32_t DeviceId, void *TgtPtr) { in __tgt_rtl_data_delete() argument
97 return Manager->dataDelete(DeviceId, TgtPtr); in __tgt_rtl_data_delete()
[all …]
H A DClient.h75 int32_t initDevice(int32_t DeviceId);
80 void *dataAlloc(int32_t DeviceId, int64_t Size, void *HstPtr);
81 int32_t dataDelete(int32_t DeviceId, void *TgtPtr);
83 int32_t dataSubmit(int32_t DeviceId, void *TgtPtr, void *HstPtr,
94 int32_t runTargetTeamRegion(int32_t DeviceId, void *TgtEntryPtr,
105 std::pair<int32_t, int32_t> mapDeviceId(int32_t DeviceId);
130 int32_t initDevice(int32_t DeviceId);
135 void *dataAlloc(int32_t DeviceId, int64_t Size, void *HstPtr);
136 int32_t dataDelete(int32_t DeviceId, void *TgtPtr);
138 int32_t dataSubmit(int32_t DeviceId, void *TgtPtr, void *HstPtr,
[all …]
/llvm-project-15.0.7/openmp/libomptarget/plugins/generic-elf-64bit/src/
H A Drtl.cpp63 assert(DeviceId < (int32_t)FuncGblEntries.size() && in createOffloadTable()
65 FuncGblEntries[DeviceId].emplace_back(); in createOffloadTable()
66 FuncOrGblEntryTy &E = FuncGblEntries[DeviceId].back(); in createOffloadTable()
73 bool findOffloadEntry(int32_t DeviceId, void *Addr) { in findOffloadEntry() argument
74 assert(DeviceId < (int32_t)FuncGblEntries.size() && in findOffloadEntry()
76 FuncOrGblEntryTy &E = FuncGblEntries[DeviceId].back(); in findOffloadEntry()
90 assert(DeviceId < (int32_t)FuncGblEntries.size() && in getOffloadEntriesTable()
92 FuncOrGblEntryTy &E = FuncGblEntries[DeviceId].back(); in getOffloadEntriesTable()
135 assert(DeviceId >= 0 && DeviceId < NUMBER_OF_DEVICES && "bad dev id"); in __tgt_rtl_load_binary()
251 return DeviceInfo.getOffloadEntriesTable(DeviceId); in __tgt_rtl_load_binary()
[all …]
/llvm-project-15.0.7/openmp/libomptarget/include/
H A Domptarget.h270 void __tgt_target_data_begin_nowait(int64_t DeviceId, int32_t ArgNum,
276 void __tgt_target_data_begin_mapper(ident_t *Loc, int64_t DeviceId,
282 ident_t *Loc, int64_t DeviceId, int32_t ArgNum, void **ArgsBase,
292 void __tgt_target_data_end_nowait(int64_t DeviceId, int32_t ArgNum,
297 void __tgt_target_data_end_mapper(ident_t *Loc, int64_t DeviceId,
302 ident_t *Loc, int64_t DeviceId, int32_t ArgNum, void **ArgsBase,
311 void __tgt_target_data_update_nowait(int64_t DeviceId, int32_t ArgNum,
317 void __tgt_target_data_update_mapper(ident_t *Loc, int64_t DeviceId,
324 ident_t *Loc, int64_t DeviceId, int32_t ArgNum, void **ArgsBase,
335 int __tgt_target_kernel(ident_t *Loc, int64_t DeviceId, int32_t NumTeams,
[all …]
/llvm-project-15.0.7/openmp/libomptarget/plugins/amdgpu/impl/
H A Dinterop_hsa.h22 int DeviceId, const char *symbol, void **var_addr, unsigned int *var_size);
H A Dimpl_runtime.h16 void *module_bytes, size_t module_size, int DeviceId,
H A Dinterop_hsa.cpp13 int DeviceId, const char *symbol, void **var_addr, unsigned int *var_size) { in interop_hsa_get_symbol_info() argument