| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | memcpy_async.cc | 138 AddressPtrList kernel_outputs; in GenDynamicKernel() local
139 …ernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in GenDynamicKernel()
145 if (kernel_outputs.size() != 1) { in GenDynamicKernel()
146 MS_LOG(EXCEPTION) << "MemCpyAsync op output is not one, got " << kernel_outputs.size(); in GenDynamicKernel()
149 MS_EXCEPTION_IF_NULL(kernel_outputs[0]); in GenDynamicKernel()
151 if (kernel_outputs[0]->size < kernel_inputs[0]->size) { in GenDynamicKernel()
152 …MS_LOG(EXCEPTION) << "rtMemcpyAsync destMax " << kernel_outputs[0]->size << " is less than src siz… in GenDynamicKernel()
156 if (kernel_outputs[0]->size > kernel_inputs[0]->size) { in GenDynamicKernel()
160 return std::make_shared<MemcpyRtsDynamicKernel>(stream_ptr, cnode_ptr, kernel_outputs[0]->addr, in GenDynamicKernel()
161 kernel_outputs[0]->size, kernel_inputs[0]->addr, in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | hccl_dynamic_kernel.cc | 47 AddressPtrList kernel_outputs; in UpdateArgs() local
48 …nelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in UpdateArgs()
49 if (kernel_inputs.empty() || kernel_outputs.empty()) { in UpdateArgs()
53 auto output0 = kernel_outputs.at(0); in UpdateArgs()
85 AddressPtrList kernel_outputs; in StaticShapeExecute() local
86 …nelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in StaticShapeExecute()
87 kernel_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in StaticShapeExecute()
|
| D | ai_core_dynamic_kernel.cc | 112 AddressPtrList kernel_outputs; in UpdateArgs() local
113 …nelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in UpdateArgs()
119 …(void)std::transform(std::begin(kernel_outputs), std::end(kernel_outputs), std::back_inserter(runt… in UpdateArgs()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_kernel.cc | 62 auto kernel_outputs = ObtainKernelAddress(outputs_list, &outputs_addr_); in ObtainKernelOutputs() local
63 return kernel_outputs; in ObtainKernelOutputs()
82 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchSingleKernel() local
86 auto ret_status = kernel_mod_->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in LaunchSingleKernel()
|
| D | kernel_runtime.cc | 1122 AddressPtrList *kernel_outputs) { in GenLaunchArgs() argument
1126 MS_EXCEPTION_IF_NULL(kernel_outputs); in GenLaunchArgs()
1167 kernel_outputs->emplace_back(output); in GenLaunchArgs()
1300 AddressPtrList *kernel_outputs) { in AssignKernelAddress() argument
1304 MS_EXCEPTION_IF_NULL(kernel_outputs); in AssignKernelAddress()
1331 kernel_outputs->emplace_back(output); in AssignKernelAddress()
1420 AddressPtrList kernel_outputs; in LaunchKernel() local
1435 AssignKernelAddress(mem_scheduler, kernel, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in LaunchKernel()
1438 kernel_outputs = kernel_mod->GetOutputsAddr(); in LaunchKernel()
1441 GenLaunchArgs(*kernel_mod, kernel, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in LaunchKernel()
[all …]
|
| D | kernel_runtime.h | 91 AddressPtrList *kernel_outputs);
151 AddressPtrList *kernel_outputs);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 116 AddressPtrList kernel_outputs; in GenDynamicKernel() local
117 …ernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in GenDynamicKernel()
128 …(void)std::transform(std::begin(kernel_outputs), std::end(kernel_outputs), std::back_inserter(runt… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 237 AddressPtrList kernel_outputs = { in HashSwapOut() local
244 auto ret = hash_swap_out_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in HashSwapOut()
278 AddressPtrList kernel_outputs; in HashSwapIn() local
287 (void)kernel_outputs.emplace_back(std::make_shared<Address>(offset_addr_, sizeof(int))); in HashSwapIn()
288 auto ret = hash_swap_in_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in HashSwapIn()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 137 …const std::vector<mindspore::kernel::AddressPtr> &kernel_outputs, int exec_order, void *stream_ptr, in LoadKernelData() argument
193 auto addr = kernel_outputs[j]; in LoadKernelData()
775 AddressPtrList kernel_outputs; in LaunchKernelDynamic() local
776 …ocKernelDynamicRes(*kernel_mod, kernel, &kernel_inputs, &kernel_workspaces, &kernel_outputs, mock); in LaunchKernelDynamic()
788 kernel::KernelLaunchInfo mem_info = {kernel_inputs, kernel_workspaces, kernel_outputs}; in LaunchKernelDynamic()
793 …LaunchKernelWithoutMock(graph, kernel, kernel_inputs, kernel_workspaces, kernel_outputs, profiling… in LaunchKernelDynamic()
801 …LoadKernelData(debugger_.get(), kernel, kernel_inputs, kernel_workspaces, kernel_outputs, exec_ord… in LaunchKernelDynamic()
895 AddressPtrList kernel_outputs; in RunOpLaunchKernelDynamic() local
896 GenLaunchArgs(*kernel_mod, kernel, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in RunOpLaunchKernelDynamic()
898 auto ret = kernel_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in RunOpLaunchKernelDynamic()
[all …]
|
| D | gpu_kernel_runtime.h | 90 AddressPtrList *kernel_outputs, bool mock);
93 AddressPtrList *kernel_outputs, bool mock);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_atomic_clean.cc | 62 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel() local
66 auto ret_status = kernel_mod_->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in LaunchOpKernel()
|
| D | ascend_launch_transdata.cc | 64 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel() local
68 auto ret_status = kernel_mod_->Launch(kernel_inputs, kernel_workspace, kernel_outputs, stream_); in LaunchOpKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 209 AddressPtrList kernel_outputs; in GenDynamicKernel() local
210 …ernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in GenDynamicKernel()
212 CreateCpuKernelInfo(kernel_inputs, kernel_outputs); in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 84 auto kernel_outputs = launch_info_->outputs_; in LoadOutputs() local
91 auto addr = kernel_outputs[j]; in LoadOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 150 AddressPtrList kernel_outputs; in LaunchKernel() local
216 kernel_outputs.push_back(output); in LaunchKernel()
235 ascend_kernel_mod->GenTask(kernel_inputs, kernel_workspaces, kernel_outputs, stream_id); in LaunchKernel()
248 debug_info->output_addrs_ = kernel_outputs; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 436 std::vector<kernel::AddressPtr> kernel_outputs; in Run() local
447 AddRuntimeAddress(device_address, &kernel_outputs); in Run()
466 kernel::KernelLaunchInfo mem_info = {kernel_inputs, kernel_workspaces, kernel_outputs}; in Run()
471 ret = kernel_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, 0); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2381 std::vector<AddressPtr> kernel_outputs; in CacheAddrForKernel() local
2418 kernel_outputs.emplace_back(output); in CacheAddrForKernel()
2431 kernel_mod->set_outputs_addr(kernel_outputs); in CacheAddrForKernel()
|