Lines Matching refs:Request
25 Status RemoteOffloadImpl::Shutdown(ServerContext *Context, const Null *Request, in Shutdown() argument
56 const Pointer *Request, I32 *Reply) { in UnregisterLib() argument
57 if (Descriptions.find((void *)Request->number()) == Descriptions.end()) { in UnregisterLib()
62 PM->RTLs.UnregisterLib(Descriptions[(void *)Request->number()].get()); in UnregisterLib()
63 freeTargetBinaryDescription(Descriptions[(void *)Request->number()].get()); in UnregisterLib()
64 Descriptions.erase((void *)Request->number()); in UnregisterLib()
143 const DevicePair *Request, in IsDataExchangeable() argument
146 if (PM->Devices[mapHostRTLDeviceId(Request->src_dev_id())] in IsDataExchangeable()
148 Reply->set_number(PM->Devices[mapHostRTLDeviceId(Request->src_dev_id())] in IsDataExchangeable()
149 ->RTL->is_data_exchangable(Request->src_dev_id(), in IsDataExchangeable()
150 Request->dst_dev_id())); in IsDataExchangeable()
153 Request->src_dev_id(), Request->dst_dev_id()) in IsDataExchangeable()
158 const AllocData *Request, Pointer *Reply) { in DataAlloc() argument
160 (uint64_t)PM->Devices[Request->device_id()]->RTL->data_alloc( in DataAlloc()
161 mapHostRTLDeviceId(Request->device_id()), Request->size(), in DataAlloc()
162 (void *)Request->hst_ptr(), TARGET_ALLOC_DEFAULT); in DataAlloc()
173 SubmitData Request; in DataSubmit() local
175 while (Reader->Read(&Request)) { in DataSubmit()
176 if (Request.start() == 0 && Request.size() == Request.data().size()) { in DataSubmit()
179 Reply->set_number(PM->Devices[Request.device_id()]->RTL->data_submit( in DataSubmit()
180 mapHostRTLDeviceId(Request.device_id()), (void *)Request.tgt_ptr(), in DataSubmit()
181 (void *)Request.data().data(), Request.data().size())); in DataSubmit()
184 Request.data().size(), (void *)Request.tgt_ptr(), in DataSubmit()
185 Request.device_id()) in DataSubmit()
190 HostCopy = new uint8_t[Request.size()]; in DataSubmit()
194 memcpy((void *)((char *)HostCopy + Request.start()), Request.data().data(), in DataSubmit()
195 Request.data().size()); in DataSubmit()
198 Reply->set_number(PM->Devices[Request.device_id()]->RTL->data_submit( in DataSubmit()
199 mapHostRTLDeviceId(Request.device_id()), (void *)Request.tgt_ptr(), in DataSubmit()
200 HostCopy, Request.size())); in DataSubmit()
204 SERVER_DBG("Submitted %lu bytes to (%p) on device %d", Request.data().size(), in DataSubmit()
205 (void *)Request.tgt_ptr(), Request.device_id()) in DataSubmit()
211 const RetrieveData *Request, in DataRetrieve() argument
213 auto HstPtr = std::make_unique<char[]>(Request->size()); in DataRetrieve()
215 auto Ret = PM->Devices[Request->device_id()]->RTL->data_retrieve( in DataRetrieve()
216 mapHostRTLDeviceId(Request->device_id()), HstPtr.get(), in DataRetrieve()
217 (void *)Request->tgt_ptr(), Request->size()); in DataRetrieve()
222 if (Request->size() > BlockSize) { in DataRetrieve()
224 for (auto I = 0; I < ceil((float)Request->size() / BlockSize); I++) { in DataRetrieve()
228 Reply->set_size(Request->size()); in DataRetrieve()
237 End, Request->size(), (void *)Request->tgt_ptr(), in DataRetrieve()
238 mapHostRTLDeviceId(Request->device_id())) in DataRetrieve()
242 if (End >= Request->size()) in DataRetrieve()
243 End = Request->size(); in DataRetrieve()
249 Reply->set_size(Request->size()); in DataRetrieve()
250 Reply->set_data((char *)HstPtr.get(), Request->size()); in DataRetrieve()
253 SERVER_DBG("Retrieved %lu bytes from (%p) on device %d", Request->size(), in DataRetrieve()
254 (void *)Request->tgt_ptr(), in DataRetrieve()
255 mapHostRTLDeviceId(Request->device_id())) in DataRetrieve()
264 const ExchangeData *Request, in DataExchange() argument
266 if (PM->Devices[Request->src_dev_id()]->RTL->data_exchange) { in DataExchange()
267 int32_t Ret = PM->Devices[Request->src_dev_id()]->RTL->data_exchange( in DataExchange()
268 mapHostRTLDeviceId(Request->src_dev_id()), (void *)Request->src_ptr(), in DataExchange()
269 mapHostRTLDeviceId(Request->dst_dev_id()), (void *)Request->dst_ptr(), in DataExchange()
270 Request->size()); in DataExchange()
278 mapHostRTLDeviceId(Request->src_dev_id()), (void *)Request->src_ptr(), in DataExchange()
279 mapHostRTLDeviceId(Request->dst_dev_id()), (void *)Request->dst_ptr(), in DataExchange()
280 Request->size()) in DataExchange()
285 const DeleteData *Request, I32 *Reply) { in DataDelete() argument
286 auto Ret = PM->Devices[Request->device_id()]->RTL->data_delete( in DataDelete()
287 mapHostRTLDeviceId(Request->device_id()), (void *)Request->tgt_ptr()); in DataDelete()
290 SERVER_DBG("Deleted data from (%p) on device %d", (void *)Request->tgt_ptr(), in DataDelete()
291 mapHostRTLDeviceId(Request->device_id())) in DataDelete()
296 const TargetRegion *Request, in RunTargetRegion() argument
298 std::vector<uint8_t> TgtArgs(Request->arg_num()); in RunTargetRegion()
299 for (auto I = 0; I < Request->arg_num(); I++) in RunTargetRegion()
300 TgtArgs[I] = (uint64_t)Request->tgt_args()[I]; in RunTargetRegion()
302 std::vector<ptrdiff_t> TgtOffsets(Request->arg_num()); in RunTargetRegion()
303 const auto *TgtOffsetItr = Request->tgt_offsets().begin(); in RunTargetRegion()
304 for (auto I = 0; I < Request->arg_num(); I++, TgtOffsetItr++) in RunTargetRegion()
307 void *TgtEntryPtr = ((__tgt_offload_entry *)Request->tgt_entry_ptr())->addr; in RunTargetRegion()
309 int32_t Ret = PM->Devices[Request->device_id()]->RTL->run_region( in RunTargetRegion()
310 mapHostRTLDeviceId(Request->device_id()), TgtEntryPtr, in RunTargetRegion()
311 (void **)TgtArgs.data(), TgtOffsets.data(), Request->arg_num()); in RunTargetRegion()
316 mapHostRTLDeviceId(Request->device_id()), Request->arg_num()) in RunTargetRegion()
321 const TargetTeamRegion *Request, in RunTargetTeamRegion() argument
323 std::vector<uint64_t> TgtArgs(Request->arg_num()); in RunTargetTeamRegion()
324 for (auto I = 0; I < Request->arg_num(); I++) in RunTargetTeamRegion()
325 TgtArgs[I] = (uint64_t)Request->tgt_args()[I]; in RunTargetTeamRegion()
327 std::vector<ptrdiff_t> TgtOffsets(Request->arg_num()); in RunTargetTeamRegion()
328 const auto *TgtOffsetItr = Request->tgt_offsets().begin(); in RunTargetTeamRegion()
329 for (auto I = 0; I < Request->arg_num(); I++, TgtOffsetItr++) in RunTargetTeamRegion()
332 void *TgtEntryPtr = ((__tgt_offload_entry *)Request->tgt_entry_ptr())->addr; in RunTargetTeamRegion()
334 int32_t Ret = PM->Devices[Request->device_id()]->RTL->run_team_region( in RunTargetTeamRegion()
335 mapHostRTLDeviceId(Request->device_id()), TgtEntryPtr, in RunTargetTeamRegion()
336 (void **)TgtArgs.data(), TgtOffsets.data(), Request->arg_num(), in RunTargetTeamRegion()
337 Request->team_num(), Request->thread_limit(), Request->loop_tripcount()); in RunTargetTeamRegion()
342 mapHostRTLDeviceId(Request->device_id()), Request->arg_num()) in RunTargetTeamRegion()