Home
last modified time | relevance | path

Searched defs:device_address (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_copy_manager.cc32 void GPUMemCopyManager::AddMemSwapOutTask(const DeviceAddressPtr &device_address, const HostAddress… in AddMemSwapOutTask()
50 void GPUMemCopyManager::AddMemSwapInTask(const DeviceAddressPtr &device_address, const HostAddress … in AddMemSwapInTask()
82 void GPUMemCopyManager::AddMemSwapOutTaskMock(const DeviceAddressPtr &device_address) { in AddMemSwapOutTaskMock()
88 void GPUMemCopyManager::AddMemSwapInTaskMock(const DeviceAddressPtr &device_address) { in AddMemSwapInTaskMock()
107 auto device_address = task.first; in UpdateSwapOutQueue() local
123 auto device_address = task.first; in UpdateSwapInQueue() local
138 auto device_address = swap_out_queue_mock_.front(); in UpdateSwapOutQueueMock() local
147 auto device_address = swap_in_queue_mock_.front(); in UpdateSwapInQueueMock() local
Dgpu_kernel_runtime.cc334 auto device_address = GetMutableOutputAddr(item[0], output_index, false); in AllocInplaceNodeMemory() local
628 … auto device_address = CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type); in InitKernelOutputAddress() local
642 auto device_address = CreateDeviceAddress(nullptr, workspace_sizes[i], "", kTypeUnknown); in InitKernelWorkspaceAddress() local
688 auto device_address = AnfAlgo::GetMutableOutputAddr(kernel, i, false); in ClearKernelOutputAddress() local
707 auto device_address = AnfAlgo::GetMutableWorkspaceAddr(kernel, i); in ClearKernelWorkspaceAddress() local
945 auto device_address = GetMutableOutputAddr(need_swap_kernel, mem_swap_info.output_idx_, false); in AddMemorySwapTask() local
996 void GPUKernelRuntime::UpdateHostSwapInQueue(const DeviceAddressPtr device_address, bool mock) { in UpdateHostSwapInQueue()
1053 bool GPUKernelRuntime::AttemptMallocMem(const DeviceAddressPtr &device_address, size_t size, bool m… in AttemptMallocMem()
1103 DeviceAddressPtr device_address; in AllocKernelInputDynamicRes() local
1146 auto device_address = GetMutableOutputAddr(kernel, i, false); in AllocKernelOutputDynamicRes() local
[all …]
Dgpu_bucket.cc59 …DeviceAddressPtr device_address = std::dynamic_pointer_cast<DeviceAddress>(tensor->device_address(… in AllocateAllReduceAddr() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_copy_manager.h106 …virtual void AddMemSwapOutTask(const DeviceAddressPtr &device_address, const HostAddress &host_add… in AddMemSwapOutTask()
108 …virtual void AddMemSwapInTask(const DeviceAddressPtr &device_address, const HostAddress &host_addr… in AddMemSwapInTask()
111 virtual void AddMemSwapOutTaskMock(const DeviceAddressPtr &device_address) {} in AddMemSwapOutTaskMock()
113 virtual void AddMemSwapInTaskMock(const DeviceAddressPtr &device_address) {} in AddMemSwapInTaskMock()
Dmem_reuse_checker.cc388 auto device_address = AnfAlgo::GetPrevNodeOutputAddr(node, real_input_index); in CheckNormalIR() local
Dmem_swap_manager.cc304 void MemSwapManager::AddMemSwapTask(SwapKind swap_kind, const DeviceAddressPtr &device_address, in AddMemSwapTask()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc235 auto device_address = CreateDeviceAddress(nullptr, tensor_size, output_format, output_type); in RunOpMallocPre() local
266 auto device_address = in RunOpMallocPre() local
295 …auto device_address = CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(input_nod… in ResetNodeAddress() local
423 auto device_address = in RunOpAssignInputMemory() local
449 auto device_address = iter.first->device_address(); in RunOpAssignOutputMemory() local
467 auto device_address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, i); in RunOpAssignOutputMemory() local
473 …auto device_address = CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type, {k… in RunOpAssignOutputMemory() local
492 auto device_address = CreateDeviceAddress(nullptr, workspace_lists[i], "", kTypeUnknown); in RunOpAssignWorkSpaceMemory() local
599 DeviceAddressPtr device_address = nullptr; in AssignStaticMemoryInput() local
924 …auto device_address = CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type, {n… in AssignNodeOutputMem() local
[all …]
Dbucket.cc76 auto device_address = std::dynamic_pointer_cast<DeviceAddress>(tensor->device_address()); in UpdateTensorAddr() local
Dkernel_adjust.cc752 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in StepLoadCtrlInputs() local
830 device::DeviceAddressPtr device_address = std::make_shared<device::ascend::AscendDeviceAddress>( in InitCtrlInputs() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc84 auto device_address = AnfAlgo::GetOutputAddr(post_node, index); in LaunchAddrCleanAkgKernel() local
112 auto device_address = AnfAlgo::GetOutputAddr(pre_node, index); in LaunchAddrCleanKernel() local
126 auto device_address = AnfAlgo::GetWorkspaceAddr(pre_node, index); in LaunchAddrCleanKernel() local
182 auto device_address = AnfAlgo::GetPrevNodeOutputAddr(anf_node_ptr, real_input_index); in LaunchKernel() local
221 auto device_address = AnfAlgo::GetWorkspaceAddr(anf_node_ptr, i); in LaunchKernel() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc99 auto device_address = device_context->CreateDeviceAddress(nullptr, tensor_size, in CreateParameterDeviceAddress() local
189 …auto device_address = device_context->CreateDeviceAddress(nullptr, output_sizes[i], output_format,… in CreateKernelOutputDeviceAddress() local
208 …auto device_address = device_context->CreateDeviceAddress(nullptr, workspace_sizes[i], "", kTypeUn… in CreateKernelWorkspaceDeviceAddress() local
241 auto device_address = AnfAlgo::GetMutableOutputAddr(group_nodes[0], output_index, false); in UpdateDeviceAddressForInplaceNode() local
272 auto device_address = AnfAlgo::GetMutableOutputAddr(node, index, false); in SetSummaryNodesRefCount() local
284 …auto device_address = AnfAlgo::GetMutableOutputAddr(item_with_index.first, item_with_index.second,… in UpdateRefCountForGraphOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc44 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData() local
70 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in CompileGraphImpl() local
Dgpu_inference_session.cc44 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData() local
70 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in CompileGraphImpl() local
Dascend_session.cc211 device::DeviceAddressPtr device_address = in GenOpOutputStubTensor() local
243 auto device_address = cur_loop_tensor->device_address(); in LoadCtrlInputTensor() local
262 auto device_address = next_loop_tensor->device_address(); in LoadCtrlInputTensor() local
279 auto device_address = epoch_tensor->device_address(); in LoadCtrlInputTensor() local
326 auto device_address = AnfAlgo::GetMutableOutputAddr(parameter, 0); in TensorNeedSync() local
496 auto device_address = AnfAlgo::GetMutableOutputAddr(input_node, 0); in LoadInputData() local
Danf_runtime_algorithm.cc2361 auto device_address = GetPrevNodeMutableOutputAddr(kernel, real_input); in CacheAddrForGraph() local
2402 auto device_address = GetPrevNodeOutputAddr(node, real_input, visit_nop_node); in CacheAddrForKernel() local
2412 auto device_address = GetOutputAddr(node, i, visit_nop_node); in CacheAddrForKernel() local
2421 auto device_address = GetWorkspaceAddr(node, i); in CacheAddrForKernel() local
2453 auto device_address = GetOutputAddr(pre_node, index); in CacheAddrForAtomicClean() local
2471 auto device_address = GetWorkspaceAddr(pre_node, index); in CacheAddrForAtomicClean() local
Dgpu_session.cc342 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData() local
Dkernel_graph.cc119 auto device_address = std::dynamic_pointer_cast<device::DeviceAddress>(device_sync); in SyncDeviceInfoToValueNode() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc255 auto device_address = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in PrepareDataForHostTensorQueue() local
314 auto device_address = device_context->CreateDeviceAddress( in PrepareDataForStepMode() local
568 auto device_address = AnfAlgo::GetMutableOutputAddr(backend_node, 0, false); in PrepareHostTensorQueueForControlNode() local
Ddata_source_actor.cc225 auto device_address = AnfAlgo::GetMutableOutputAddr(data_node, 0, false); in FillDataBuffer() local
Dkernel_actor.cc74 auto device_address = output_device_tensors_[data_arrow->from_output_index_]; in Init() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc439 auto device_address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, i).get(); in Run() local
445 auto device_address = AnfAlgo::GetMutableOutputAddr(kernel, i).get(); in Run() local
452 auto device_address = AnfAlgo::GetWorkspaceAddr(kernel, i); in Run() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.cc44 …DeviceAddressPtr device_address = std::dynamic_pointer_cast<DeviceAddress>(tensor->device_address(… in AllocateAllReduceAddr() local
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc517 auto device_address = std::make_shared<AscendDeviceAddress>(addr, 1); in TEST_F() local
532 auto device_address = std::make_shared<AscendDeviceAddress>(addr, 1); in TEST_F() local
549 auto device_address = std::make_shared<AscendDeviceAddress>(addr, 1); in TEST_F() local
565 auto device_address = std::make_shared<AscendDeviceAddress>(addr, 1); in TEST_F() local
576 auto device_address = std::make_shared<AscendDeviceAddress>(addr, 1); in TEST_F() local
/third_party/mindspore/mindspore/core/ir/
Dtensor.h288 DeviceSyncPtr device_address() const { return device_sync_; } in device_address() function
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/
Dps_cache_manager.cc223 auto &device_address = item.second.device_address; in AllocMemForHashTable() local

12