Home
last modified time | relevance | path

Searched refs:kernel_mod (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc340 auto kernel_mod = AnfAlgo::GetKernelMod(item[0]); in AllocInplaceNodeMemory() local
341 MS_EXCEPTION_IF_NULL(kernel_mod); in AllocInplaceNodeMemory()
342 auto output_size = kernel_mod->GetOutputSizeList(); in AllocInplaceNodeMemory()
619 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelOutputAddress() local
620 MS_EXCEPTION_IF_NULL(kernel_mod); in InitKernelOutputAddress()
621 auto output_sizes = kernel_mod->GetOutputSizeList(); in InitKernelOutputAddress()
638 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelWorkspaceAddress() local
639 MS_EXCEPTION_IF_NULL(kernel_mod); in InitKernelWorkspaceAddress()
640 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitKernelWorkspaceAddress()
681 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelOutputAddress() local
[all …]
Dgpu_kernel_runtime.h88 …bool AllocKernelDynamicRes(const mindspore::kernel::KernelMod &kernel_mod, const mindspore::AnfNod…
92 …bool AllocKernelOutputDynamicRes(const mindspore::kernel::KernelMod &kernel_mod, const mindspore::…
94 bool AllocKernelWorkspaceDynamicRes(const mindspore::kernel::KernelMod &kernel_mod,
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dhccl_dynamic_kernel.cc42 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs() local
43 MS_EXCEPTION_IF_NULL(kernel_mod); in UpdateArgs()
48 …KernelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outpu… in UpdateArgs()
81 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in StaticShapeExecute() local
82 MS_EXCEPTION_IF_NULL(kernel_mod); in StaticShapeExecute()
86 …KernelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outpu… in StaticShapeExecute()
87 kernel_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in StaticShapeExecute()
Dai_core_dynamic_kernel.cc107 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs() local
108 MS_EXCEPTION_IF_NULL(kernel_mod); in UpdateArgs()
113 …KernelRuntime::GenLaunchArgs(*kernel_mod, cnode, &kernel_inputs, &kernel_workspaces, &kernel_outpu… in UpdateArgs()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_simple_mem_plan.cc51 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemPlan() local
52 MS_EXCEPTION_IF_NULL(kernel_mod); in MemPlan()
53 for (size_t i = 0; i < kernel_mod->GetWorkspaceSizeList().size(); ++i) { in MemPlan()
97 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemAssign() local
98 MS_EXCEPTION_IF_NULL(kernel_mod); in MemAssign()
99 for (size_t i = 0; i < kernel_mod->GetWorkspaceSizeList().size(); ++i) { in MemAssign()
Dcpu_memory_manager.cc182 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in IncreaseAddressRefCount() local
183 MS_EXCEPTION_IF_NULL(kernel_mod); in IncreaseAddressRefCount()
184 for (size_t i = 0; i < kernel_mod->GetWorkspaceSizeList().size(); ++i) { in IncreaseAddressRefCount()
207 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in DecreaseAddressRefCount() local
208 MS_EXCEPTION_IF_NULL(kernel_mod); in DecreaseAddressRefCount()
209 for (size_t i = 0; i < kernel_mod->GetWorkspaceSizeList().size(); ++i) { in DecreaseAddressRefCount()
Dcpu_kernel_runtime.cc159 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in AssignKernelOutputAddress() local
160 MS_EXCEPTION_IF_NULL(kernel_mod); in AssignKernelOutputAddress()
161 auto output_sizes = kernel_mod->GetOutputSizeList(); in AssignKernelOutputAddress()
168 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in AssignKernelOutputAddress()
449 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in Run() local
450 MS_EXCEPTION_IF_NULL(kernel_mod); in Run()
451 for (size_t i = 0; i < kernel_mod->GetWorkspaceSizeList().size(); ++i) { in Run()
471 ret = kernel_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, 0); in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc239 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernel() local
240 MS_EXCEPTION_IF_NULL(kernel_mod); in LaunchKernel()
241 auto cpu_kernel_mod = dynamic_cast<kernel::CPUKernel *>(kernel_mod); in LaunchKernel()
264 return DoLaunchKernel(kernel_mod, inputs, workspace, outputs); in LaunchKernel()
276 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithProfiling() local
277 MS_EXCEPTION_IF_NULL(kernel_mod); in LaunchKernelWithProfiling()
281 bool ret = DoLaunchKernel(kernel_mod, inputs, workspace, outputs); in LaunchKernelWithProfiling()
287 bool CPUDeviceContext::DoLaunchKernel(KernelMod *const kernel_mod, const std::vector<AddressPtr> &i… in DoLaunchKernel() argument
290 MS_EXCEPTION_IF_NULL(kernel_mod); in DoLaunchKernel()
291 return kernel_mod->Launch(inputs, workspace, outputs, nullptr); in DoLaunchKernel()
Dcpu_device_context.h74 bool DoLaunchKernel(KernelMod *const kernel_mod, const std::vector<AddressPtr> &inputs,
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc278 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ResetNodeAddress() local
279 MS_EXCEPTION_IF_NULL(kernel_mod); in ResetNodeAddress()
300 auto output_sizes = kernel_mod->GetOutputSizeList(); in ResetNodeAddress()
307 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in ResetNodeAddress()
354 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in RunOpClearMemory() local
355 MS_EXCEPTION_IF_NULL(kernel_mod); in RunOpClearMemory()
356 auto workspace_lists = kernel_mod->GetWorkspaceSizeList(); in RunOpClearMemory()
440 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignOutputMemory() local
441 MS_EXCEPTION_IF_NULL(kernel_mod); in RunOpAssignOutputMemory()
442 auto output_sizes = kernel_mod->GetOutputSizeList(); in RunOpAssignOutputMemory()
[all …]
Dkernel_info.cc111 void KernelInfo::set_kernel_mod(const kernel::KernelModPtr &kernel_mod) { kernel_mod_ = kernel_mod;… in set_kernel_mod() argument
115 const kernel::KernelMod *KernelInfo::kernel_mod() const { return kernel_mod_.get(); } in kernel_mod() function in mindspore::device::KernelInfo
Dkernel_info.h60 void set_kernel_mod(const kernel::KernelModPtr &kernel_mod);
62 const kernel::KernelMod *kernel_mod() const;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel.cc29 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs() local
30 MS_EXCEPTION_IF_NULL(kernel_mod); in UpdateArgs()
31 auto gpu_kernel_mod = dynamic_cast<GpuKernel *>(kernel_mod); in UpdateArgs()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc375 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in UpdateDynamicShape() local
376 MS_EXCEPTION_IF_NULL(kernel_mod); in UpdateDynamicShape()
382 kernel::GpuKernel *gpu_kernel = dynamic_cast<kernel::GpuKernel *>(kernel_mod); in UpdateDynamicShape()
400 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernel() local
401 MS_EXCEPTION_IF_NULL(kernel_mod); in LaunchKernel()
410 ret = DoLaunchKernel(kernel_mod, inputs, workspace, outputs); in LaunchKernel()
432 kernel::GpuKernel *gpu_kernel = dynamic_cast<kernel::GpuKernel *>(kernel_mod); in LaunchKernel()
456 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithProfiling() local
457 MS_EXCEPTION_IF_NULL(kernel_mod); in LaunchKernelWithProfiling()
460 bool ret = DoLaunchKernel(kernel_mod, inputs, workspace, outputs); in LaunchKernelWithProfiling()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc41 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicOutputKernelRef() local
42 if (kernel_mod == nullptr) { in InitDynamicOutputKernelRef()
49 auto output_sizes = kernel_mod->GetOutputSizeList(); in InitDynamicOutputKernelRef()
106 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicWorkspaceKernelRef() local
107 if (kernel_mod == nullptr) { in InitDynamicWorkspaceKernelRef()
111 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitDynamicWorkspaceKernelRef()
217 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetOutputMap() local
218 MS_EXCEPTION_IF_NULL(kernel_mod); in SetOutputMap()
219 for (size_t k = 0; k < kernel_mod->GetOutputSizeList().size(); ++k) { in SetOutputMap()
232 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetWkMap() local
[all …]
Dmem_reuse_checker.cc114 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriDy() local
115 MS_EXCEPTION_IF_NULL(kernel_mod); in CalculOriDy()
116 for (auto &dy_size : kernel_mod->GetOutputSizeList()) { in CalculOriDy()
129 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriWk() local
130 MS_EXCEPTION_IF_NULL(kernel_mod); in CalculOriWk()
131 for (auto &wk_size : kernel_mod->GetWorkspaceSizeList()) { in CalculOriWk()
221 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_input); in CheckGraphOutputAssigned() local
222 if (kernel_mod == nullptr) { in CheckGraphOutputAssigned()
225 auto output_sizes = kernel_mod->GetOutputSizeList(); in CheckGraphOutputAssigned()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc151 auto kernel_mod = AnfAlgo::GetKernelMod(anf_node_ptr); in LaunchKernel() local
152 MS_EXCEPTION_IF_NULL(kernel_mod); in LaunchKernel()
153 kernel_mod->set_unique_name(anf_node_ptr->UniqueName()); in LaunchKernel()
154 kernel_mod->set_fullname(anf_node_ptr->fullname_with_scope()); in LaunchKernel()
155 …kernel_mod->set_is_monad(AnfAlgo::IsNodeInputContainMonad(anf_node_ptr) && HasAbstractMonad(anf_no… in LaunchKernel()
220 for (size_t i = 0; i < kernel_mod->GetWorkspaceSizeList().size(); ++i) { in LaunchKernel()
232 auto ascend_kernel_mod = dynamic_cast<kernel::AscendKernelMod *>(kernel_mod); in LaunchKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc133 auto kernel_mod = AnfAlgo::GetKernelMod(pre_node); in CalCleanZerosSize() local
134 MS_EXCEPTION_IF_NULL(kernel_mod); in CalCleanZerosSize()
140 auto output_men_size = kernel_mod->GetOutputSizeList(); in CalCleanZerosSize()
149 auto workspace_men_sizes = kernel_mod->GetWorkspaceSizeList(); in CalCleanZerosSize()
218 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_node); in IsAtomicNode() local
219 MS_EXCEPTION_IF_NULL(kernel_mod); in IsAtomicNode()
220 auto parameters_indexs = kernel_mod->GenParameters(); in IsAtomicNode()
233 size_t workspace_num = kernel_mod->GetWorkspaceSizeList().size(); in IsAtomicNode()
Dascend_kernel_runtime.cc436 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in GenDynamicKernel() local
437 MS_EXCEPTION_IF_NULL(kernel_mod); in GenDynamicKernel()
438 auto dynamic_kernel = kernel_mod->GenDynamicKernel(cnode, stream_); in GenDynamicKernel()
740 auto kernel_mod = AnfAlgo::GetKernelMod(node); in SetKernelModStream() local
741 auto ascend_kernel_mod = dynamic_cast<kernel::AscendKernelMod *>(kernel_mod); in SetKernelModStream()
777 auto kernel_mod = AnfAlgo::GetKernelMod(kernels[i - 1]); in SetKernelModStream() local
778 auto ascend_kernel_mod = dynamic_cast<kernel::AscendKernelMod *>(kernel_mod); in SetKernelModStream()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dtask_desc_reporter.cc39 auto kernel_mod = AnfAlgo::GetKernelMod(node); in ReportData() local
40 auto ascend_kernel_mod = dynamic_cast<kernel::AscendKernelMod *>(kernel_mod); in ReportData()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_fusion.cc97 auto kernel_mod = build_manger->GenKernelMod(input_size_list, output_size_list, kernel_pack); in KernelFusion() local
98 if (kernel_mod != nullptr) { in KernelFusion()
99 kernel_mod_ret[fusion_scope_iter.scope_id] = kernel_mod; in KernelFusion()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc179 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CreateKernelOutputDeviceAddress() local
180 MS_EXCEPTION_IF_NULL(kernel_mod); in CreateKernelOutputDeviceAddress()
181 auto output_sizes = kernel_mod->GetOutputSizeList(); in CreateKernelOutputDeviceAddress()
204 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CreateKernelWorkspaceDeviceAddress() local
205 MS_EXCEPTION_IF_NULL(kernel_mod); in CreateKernelWorkspaceDeviceAddress()
206 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in CreateKernelWorkspaceDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_parallel_build.cc189 …auto kernel_mod = GenKernelMod(task_iter->second.input_size_list, task_iter->second.output_size_li… in TaskFinishProcess() local
190 MS_EXCEPTION_IF_NULL(kernel_mod); in TaskFinishProcess()
199 AnfAlgo::SetKernelMod(kernel_mod, cur_node.get()); in TaskFinishProcess()
212 auto ret = std::make_pair(task_iter->second.scope_id, kernel_mod); in TaskFinishProcess()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dkernel_actor.cc346 const auto &kernel_mod = kernel_info_->kernel_mod(); in FetchOutputDeviceTensor() local
347 MS_EXCEPTION_IF_NULL(kernel_mod); in FetchOutputDeviceTensor()
348 const auto &output_size_list = kernel_mod->GetOutputSizeList(); in FetchOutputDeviceTensor()
Ddata_prepare_actor.cc66 const auto &kernel_mod = AnfAlgo::GetKernelMod(node); in FetchContinuousMemoryInfo() local
67 MS_EXCEPTION_IF_NULL(kernel_mod); in FetchContinuousMemoryInfo()
73 const auto &intput_sizes = kernel_mod->GetInputSizeList(); in FetchContinuousMemoryInfo()
82 const auto &output_sizes = kernel_mod->GetOutputSizeList(); in FetchContinuousMemoryInfo()

12