Searched refs:io_addrs (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | aicpu_task.cc | 52 std::vector<void *> io_addrs; in Distribute() local
53 …io_addrs.insert(io_addrs.end(), task_info_->input_data_addrs().begin(), task_info_->input_data_add… in Distribute()
54 …io_addrs.insert(io_addrs.end(), task_info_->output_data_addrs().begin(), task_info_->output_data_a… in Distribute()
55 auto io_addrs_num = static_cast<uint32_t>(io_addrs.size()); in Distribute()
70 SetInputOutputAddrs(io_addrs, io_addr_offset); in Distribute()
137 void AicpuTask::SetInputOutputAddrs(const std::vector<void *> &io_addrs, uint32_t io_addr_offset) { in SetInputOutputAddrs() argument
139 if (!io_addrs.empty()) { in SetInputOutputAddrs()
141 static_cast<uint32_t>(io_addrs.size()) * sizeof(void *), io_addrs.data(), in SetInputOutputAddrs()
142 … static_cast<uint32_t>(io_addrs.size()) * sizeof(void *), RT_MEMCPY_HOST_TO_DEVICE); in SetInputOutputAddrs()
|
| D | aicpu_task.h | 41 void SetInputOutputAddrs(const std::vector<void *> &io_addrs, uint32_t io_addr_offset);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_cpu_dynamic_kernel.cc | 111 std::vector<uint64_t> io_addrs; in UpdateInputOutputAddr() local
112 io_addrs.reserve(input_num_ + output_num_); in UpdateInputOutputAddr()
118 io_addrs.emplace_back(reinterpret_cast<uintptr_t>(input_addr->GetMutablePtr())); in UpdateInputOutputAddr()
124 io_addrs.emplace_back(reinterpret_cast<uintptr_t>(output_addr->GetMutablePtr())); in UpdateInputOutputAddr()
133 if (io_addrs.empty()) { in UpdateInputOutputAddr()
138 …io_ptr, args_.size() - sizeof(kernel::AicpuParamHead), &io_addrs[0], sizeof(uint64_t) * io_addrs.s… in UpdateInputOutputAddr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 87 vector<void *> io_addrs; in CreateCpuKernelInfo() local
88 (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(io_addrs), in CreateCpuKernelInfo()
90 (void)std::transform(std::begin(outputs), std::end(outputs), std::back_inserter(io_addrs), in CreateCpuKernelInfo()
93 auto io_addrs_num = io_addrs.size(); in CreateCpuKernelInfo()
122 (void)args_.append(reinterpret_cast<const char *>(io_addrs.data()), io_addrs_size); in CreateCpuKernelInfo()
|