Home
last modified time | relevance | path

Searched refs:dst_device (Results 1 – 25 of 31) sorted by relevance

12

/external/llvm-project/openmp/libomptarget/src/
Dapi.cpp127 size_t dst_offset, size_t src_offset, int dst_device, int src_device) { in omp_target_memcpy() argument
130 "src offset %zu, length %zu\n", dst_device, src_device, DPxPTR(dst), in omp_target_memcpy()
143 if (dst_device != omp_get_initial_device() && !device_is_ready(dst_device)) { in omp_target_memcpy()
153 dst_device == omp_get_initial_device()) { in omp_target_memcpy()
160 DeviceTy &DstDev = PM->Devices[dst_device]; in omp_target_memcpy()
162 } else if (dst_device == omp_get_initial_device()) { in omp_target_memcpy()
169 DeviceTy &DstDev = PM->Devices[dst_device]; in omp_target_memcpy()
192 const size_t *src_dimensions, int dst_device, int src_device) { in omp_target_memcpy_rect() argument
196 "volume " DPxMOD ", element size %zu, num_dims %d\n", dst_device, in omp_target_memcpy_rect()
217 dst_device, src_device); in omp_target_memcpy_rect()
[all …]
/external/llvm-project/openmp/libomptarget/test/offloading/
Dd2d_memcpy.c25 int dst_device = num_devices - 1; in main() local
29 int *dst_ptr = omp_target_alloc(length, dst_device); in main()
41 omp_target_memcpy(dst_ptr, src_ptr, length, 0, 0, dst_device, src_device); in main()
49 #pragma omp target teams distribute parallel for device(dst_device) \ in main()
63 omp_target_free(dst_ptr, dst_device); in main()
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc70 Device* dst_device; in SameWorkerRecvDone() local
71 s = device_mgr->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
82 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone()
87 uint64 safe_alloc_frontier = dst_device->SafeAllocFrontier(0); in SameWorkerRecvDone()
88 std::function<uint64()> freed_by_func = [dst_device, in SameWorkerRecvDone()
90 safe_alloc_frontier = dst_device->SafeAllocFrontier(safe_alloc_frontier); in SameWorkerRecvDone()
111 src_device, dst_device, send_args.alloc_attrs, recv_args.alloc_attrs, &in, in SameWorkerRecvDone()
Dreplicate_per_replica_nodes.cc104 const string& dst_device = dst->assigned_device_name(); in ReplicateFromCompositeDeviceToRegularDevice() local
107 if (replicated_node->assigned_device_name() == dst_device) { in ReplicateFromCompositeDeviceToRegularDevice()
155 " assigned to ", dst_device); in ReplicateFromCompositeDeviceToRegularDevice()
Dcolocation_graph.cc231 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice() argument
232 return src_device.has_type && dst_device.has_type && in IsEdgeFromCompositeDeviceToPhysicalDevice()
234 !IsCompositeDevice(dst_device.type); in IsEdgeFromCompositeDeviceToPhysicalDevice()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_logger.cc89 const string& dst_device, in RecordRecvTensor() argument
92 dst_device, bytes, "", "RecvTensor"); in RecordRecvTensor()
99 const string& dst_device, in RecordDataTransfer() argument
115 " from ", src_device, " to ", dst_device); in RecordDataTransfer()
130 Save(dst_device, step_id, ns); in RecordDataTransfer()
Dcollective_rma_distributed_test.cc317 Device* dst_device = nullptr; in TEST_F() local
319 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
325 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
347 Device* dst_device = nullptr; in TEST_F() local
349 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
355 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
382 Device* dst_device = nullptr; in TEST_F() local
384 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
390 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
410 Device* dst_device = nullptr; in TEST_F() local
[all …]
Dbase_rendezvous_mgr.cc213 if (!is_src && !IsLocalDevice(sess->worker_name(), parsed.dst_device)) { in ValidateDevices()
255 Device* dst_device; in SameWorkerRecvDone() local
256 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
267 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone()
269 uint64 safe_alloc_frontier = dst_device->SafeAllocFrontier(0); in SameWorkerRecvDone()
271 std::function<uint64()> freed_by_func = [dst_device, &safe_alloc_frontier]() { in SameWorkerRecvDone()
272 safe_alloc_frontier = dst_device->SafeAllocFrontier(safe_alloc_frontier); in SameWorkerRecvDone()
288 src_device, dst_device, send_args.alloc_attrs, recv_args.alloc_attrs, &in, in SameWorkerRecvDone()
Dworker_cache_logger.h61 const string& dst_device, int64 bytes);
67 const string& dst_device, int64 bytes,
/external/tensorflow/tensorflow/core/framework/
Drendezvous.cc45 dst_device = StringPiece(buf_.data() + (b.dst_device.data() - b_base), in operator =()
46 b.dst_device.size()); in operator =()
55 const string& dst_device, const string& name, in CreateKey() argument
67 dst_device, ";", name, ";", frame_iter.frame_id, ":", frame_iter.iter_id); in CreateKey()
110 out->dst_device = StringPiece(parts[2].data(), parts[2].size()); in ParseKey()
Drendezvous.h63 StringPiece dst_device; member
174 const std::string& dst_device,
Dlocal_rendezvous.cc121 ->GetCell(string(key.src_device), string(key.dst_device)) in Send()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr.cc63 AllocatorAttributes alloc_attrs, Device* dst_device, in Init() argument
67 dst_device_ = dst_device; in Init()
130 Device* dst_device() const { return dst_device_; } in dst_device() function in tensorflow::__anon2b42d2c80111::RpcRecvTensorCall
259 Device* dst_device; in RecvFromRemoteAsync() local
261 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
272 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device, in RecvFromRemoteAsync()
/external/llvm-project/mlir/test/mlir-rocm-runner/
Dtwo-modules.mlir12 %dst_device = call @mgpuMemGetDeviceMemRef1dInt32(%dst) : (memref<?xi32>) -> (memref<?xi32>)
16 store %t0, %dst_device[%tx] : memref<?xi32>
22 store %t0, %dst_device[%tx] : memref<?xi32>
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.cc72 const ClientAndPtr<PjRtDevice>& dst_device) const { in CopyToDevice()
73 CHECK(dst_device.get() != nullptr); in CopyToDevice()
78 TF_ASSIGN_OR_RETURN(out, buffer_->CopyToDevice(dst_device.get())); in CopyToDevice()
81 return std::make_unique<PyBuffer>(dst_device.client, std::move(out), in CopyToDevice()
Dpy_buffer.h69 const ClientAndPtr<PjRtDevice>& dst_device) const;
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc146 [](PyTpuBuffer* buffer, std::shared_ptr<PjRtDevice> dst_device) { in PYBIND11_MODULE() argument
147 CHECK(dst_device != nullptr); in PYBIND11_MODULE()
150 return buffer->CopyToDevice(std::move(dst_device)); in PYBIND11_MODULE()
Dtpu_client.cc391 std::shared_ptr<PjRtDevice> dst_device) { in CopyToDevice() argument
398 if (dst_device->id() == device_->id()) { in CopyToDevice()
417 client_, std::move(dst_device))); in CopyToDevice()
/external/llvm-project/openmp/libomptarget/include/
Domptarget.h145 size_t src_offset, int dst_device, int src_device);
149 const size_t *src_dimensions, int dst_device, int src_device);
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc681 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
683 if (src_device == dst_device) continue; in AddControlFlow()
714 const string& cl_key = strings::StrCat(curr_frame_name, "$$", dst_device); in AddControlFlow()
763 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
764 if (src_device != dst_device) { in AddControlFlow()
771 strings::StrCat(dst_frame_name, "$$", dst_device); in AddControlFlow()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1294 PjRtDevice* dst_device, LocalDeviceState* dst_local_device, in CopyToDeviceHelper() argument
1300 dst_device, dst_local_device, transfer_stream, in CopyToDeviceHelper()
1355 PjRtDevice* dst_device) { in CopyToDevice() argument
1358 if (dst_device == device_) { in CopyToDevice()
1364 if (dst_device->client() != client_) { in CopyToDevice()
1368 return dst_device->client()->BufferFromHostBuffer( in CopyToDevice()
1371 [literal{std::move(literal)}]() { /* frees literal */ }, dst_device); in CopyToDevice()
1376 tensorflow::down_cast<PjRtStreamExecutorDevice*>(dst_device) in CopyToDevice()
1405 dst_device, dst_local_device, transfer_local_device, transfer_stream, in CopyToDevice()
Dpjrt_stream_executor_client.h519 PjRtDevice* dst_device) override;
582 CopyToDeviceHelper(PjRtDevice* dst_device, LocalDeviceState* dst_local_device,
Dpjrt_client.h369 PjRtDevice* dst_device) = 0;
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc289 int64 dst_device = sharding.tile_assignment()(target_indices); in GetFinalFftUsingCollectivePermute() local
290 src_dst_pairs.emplace_back(src_device, dst_device); in GetFinalFftUsingCollectivePermute()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_tracer.cc366 size_t num_bytes, uint32 src_device, uint32 dst_device, bool async, in PopulateMemcpyCallbackEvent() argument
379 event.memcpy_info.destination = dst_device; in PopulateMemcpyCallbackEvent()
444 uint32 dst_device = -1, src_device = -1; in AddP2PMemcpyEventUponApiExit() local
448 cupti_interface->GetDeviceId(p2p_params->dstContext, &dst_device); in AddP2PMemcpyEventUponApiExit()
450 << " dst: " << dst_device << " size:" << num_bytes; in AddP2PMemcpyEventUponApiExit()
453 dst_device, async, start_time, end_time); in AddP2PMemcpyEventUponApiExit()

12