| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.h | 85 void LaunchKernelWithTimeProfiling(const AnfNodePtr &kernel, const AddressPtrList &inputs,
86 … const AddressPtrList &workspace, const AddressPtrList &outputs);
89 AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces,
90 AddressPtrList *kernel_outputs, bool mock);
91 …bool AllocKernelInputDynamicRes(const mindspore::AnfNodePtr &kernel, AddressPtrList *kernel_inputs…
93 AddressPtrList *kernel_outputs, bool mock);
95 … const mindspore::AnfNodePtr &kernel, AddressPtrList *kernel_workspaces,
118 const AddressPtrList &inputs, const AddressPtrList &workspaces,
119 const AddressPtrList &outputs, bool profiling);
|
| D | gpu_kernel_runtime.cc | 773 AddressPtrList kernel_inputs; in LaunchKernelDynamic()
774 AddressPtrList kernel_workspaces; in LaunchKernelDynamic()
775 AddressPtrList kernel_outputs; in LaunchKernelDynamic()
830 … const AddressPtrList &inputs, const AddressPtrList &workspaces, in LaunchKernelWithoutMock()
831 const AddressPtrList &outputs, bool profiling) { in LaunchKernelWithoutMock()
893 AddressPtrList kernel_inputs; in RunOpLaunchKernelDynamic()
894 AddressPtrList kernel_workspaces; in RunOpLaunchKernelDynamic()
895 AddressPtrList kernel_outputs; in RunOpLaunchKernelDynamic()
910 void GPUKernelRuntime::LaunchKernelWithTimeProfiling(const AnfNodePtr &kernel, const AddressPtrList… in LaunchKernelWithTimeProfiling()
911 … const AddressPtrList &workspace, const AddressPtrList &outputs) { in LaunchKernelWithTimeProfiling()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | hccl_dynamic_kernel.cc | 45 AddressPtrList kernel_inputs; in UpdateArgs()
46 AddressPtrList kernel_workspaces; in UpdateArgs()
47 AddressPtrList kernel_outputs; in UpdateArgs()
83 AddressPtrList kernel_inputs; in StaticShapeExecute()
84 AddressPtrList kernel_workspaces; in StaticShapeExecute()
85 AddressPtrList kernel_outputs; in StaticShapeExecute()
|
| D | ai_core_dynamic_kernel.cc | 110 AddressPtrList kernel_inputs; in UpdateArgs()
111 AddressPtrList kernel_workspaces; in UpdateArgs()
112 AddressPtrList kernel_outputs; in UpdateArgs()
|
| /third_party/mindspore/mindspore/ccsrc/debug/rdr/ |
| D | mem_address_recorder.h | 33 using AddressPtrList = std::vector<kernel::AddressPtr>; variable
35 AddressPtrList *inputs_;
36 AddressPtrList *workspaces_;
37 AddressPtrList *outputs_;
|
| D | running_data_recorder.h | 36 using AddressPtrList = std::vector<kernel::AddressPtr>; variable
|
| D | mem_address_recorder.cc | 24 std::string MemInfo2String(const std::string &label, const AddressPtrList &info) { in MemInfo2String()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.h | 44 using AddressPtrList = std::vector<mindspore::kernel::AddressPtr>; variable
90 AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces,
91 AddressPtrList *kernel_outputs);
150 AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces,
151 AddressPtrList *kernel_outputs);
161 static void GenAddrCleanLaunchArgs(const CNodePtr &cnode, AddressPtrList *kernel_inputs,
|
| D | kernel_runtime.cc | 1121 … AddressPtrList *kernel_inputs, AddressPtrList *const kernel_workspaces, in GenLaunchArgs()
1122 AddressPtrList *kernel_outputs) { in GenLaunchArgs()
1181 void KernelRuntime::GenAddrCleanLaunchArgs(const CNodePtr &cnode, AddressPtrList *kernel_inputs, in GenAddrCleanLaunchArgs()
1299 … AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces, in AssignKernelAddress()
1300 AddressPtrList *kernel_outputs) { in AssignKernelAddress()
1418 AddressPtrList kernel_inputs; in LaunchKernel()
1419 AddressPtrList kernel_workspaces; in LaunchKernel()
1420 AddressPtrList kernel_outputs; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.h | 37 using AddressPtrList = std::vector<mindspore::kernel::AddressPtr>; variable
67 static void LaunchAddrCleanKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inputs);
68 static void LaunchAddrCleanAkgKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inputs);
|
| D | task_generator.cc | 60 void TaskGenerator::LaunchAddrCleanAkgKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_i… in LaunchAddrCleanAkgKernel()
95 void TaskGenerator::LaunchAddrCleanKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inpu… in LaunchAddrCleanKernel()
148 AddressPtrList kernel_inputs; in LaunchKernel()
149 AddressPtrList kernel_workspaces; in LaunchKernel()
150 AddressPtrList kernel_outputs; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 34 using AddressPtrList = std::vector<AddressPtr>; typedef
236 AddressPtrList kernel_inputs; in HashSwapOut()
237 AddressPtrList kernel_outputs = { in HashSwapOut()
239 AddressPtrList kernel_workspaces; in HashSwapOut()
277 AddressPtrList kernel_inputs; in HashSwapIn()
278 AddressPtrList kernel_outputs; in HashSwapIn()
279 AddressPtrList kernel_workspaces; in HashSwapIn()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 28 using AddressPtrList = std::vector<mindspore::kernel::AddressPtr>; typedef
114 AddressPtrList kernel_inputs; in GenDynamicKernel()
115 AddressPtrList kernel_workspaces; in GenDynamicKernel()
116 AddressPtrList kernel_outputs; in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | memcpy_async.cc | 136 AddressPtrList kernel_inputs; in GenDynamicKernel()
137 AddressPtrList kernel_workspaces; in GenDynamicKernel()
138 AddressPtrList kernel_outputs; in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 207 AddressPtrList kernel_inputs; in GenDynamicKernel()
208 AddressPtrList kernel_workspaces; in GenDynamicKernel()
209 AddressPtrList kernel_outputs; in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.cc | 276 AddressPtrList inputs; in GenDynamicKernel()
277 AddressPtrList workspaces; in GenDynamicKernel()
278 AddressPtrList outputs; in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 31 using AddressPtrList = std::vector<mindspore::kernel::AddressPtr>; typedef
|