| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 340 auto kernel_mod = AnfAlgo::GetKernelMod(item[0]); in AllocInplaceNodeMemory() local
619 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelOutputAddress() local
638 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelWorkspaceAddress() local
681 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelOutputAddress() local
703 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelWorkspaceAddress() local
753 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelDynamic() local
850 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithoutMock() local
877 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpLaunchKernelDynamic() local
913 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithTimeProfiling() local
1080 bool GPUKernelRuntime::AllocKernelDynamicRes(const mindspore::kernel::KernelMod &kernel_mod, in AllocKernelDynamicRes()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | hccl_dynamic_kernel.cc | 42 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs() local
81 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in StaticShapeExecute() local
|
| D | ai_core_dynamic_kernel.cc | 107 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.cc | 111 void KernelInfo::set_kernel_mod(const kernel::KernelModPtr &kernel_mod) { kernel_mod_ = kernel_mod;… in set_kernel_mod()
115 const kernel::KernelMod *KernelInfo::kernel_mod() const { return kernel_mod_.get(); } in kernel_mod() function in mindspore::device::KernelInfo
|
| D | kernel_runtime.cc | 278 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ResetNodeAddress() local
354 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in RunOpClearMemory() local
440 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignOutputMemory() local
488 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignWorkSpaceMemory() local
751 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignCommunicationNodeOutputMem() local
823 auto kernel_mod = AnfAlgo::GetKernelMod(anf_node); in PreAssignCNodeMemory() local
903 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignNodeOutputMem() local
1106 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignWorkSpaceMem() local
1120 void KernelRuntime::GenLaunchArgs(const mindspore::kernel::KernelMod &kernel_mod, const mindspore::… in GenLaunchArgs()
1245 bool KernelRuntime::LaunchKernelWithPynativeProfiling(kernel::KernelMod *kernel_mod, const std::str… in LaunchKernelWithPynativeProfiling()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_simple_mem_plan.cc | 51 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemPlan() local
97 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemAssign() local
|
| D | cpu_memory_manager.cc | 182 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in IncreaseAddressRefCount() local
207 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in DecreaseAddressRefCount() local
|
| D | cpu_kernel_runtime.cc | 159 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in AssignKernelOutputAddress() local
449 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel.cc | 29 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.cc | 375 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in UpdateDynamicShape() local
400 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernel() local
456 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithProfiling() local
475 bool GPUDeviceContext::DoLaunchKernel(KernelMod *kernel_mod, const std::vector<AddressPtr> &inputs, in DoLaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.cc | 239 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernel() local
276 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithProfiling() local
287 bool CPUDeviceContext::DoLaunchKernel(KernelMod *const kernel_mod, const std::vector<AddressPtr> &i… in DoLaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/ |
| D | task_desc_reporter.cc | 39 auto kernel_mod = AnfAlgo::GetKernelMod(node); in ReportData() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse.cc | 41 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicOutputKernelRef() local
106 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicWorkspaceKernelRef() local
217 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetOutputMap() local
232 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetWkMap() local
|
| D | mem_reuse_checker.cc | 114 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriDy() local
129 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriWk() local
221 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_input); in CheckGraphOutputAssigned() local
|
| D | mem_swap_manager.cc | 36 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 97 auto kernel_mod = build_manger->GenKernelMod(input_size_list, output_size_list, kernel_pack); in KernelFusion() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 133 auto kernel_mod = AnfAlgo::GetKernelMod(pre_node); in CalCleanZerosSize() local
218 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_node); in IsAtomicNode() local
|
| D | ascend_kernel_runtime.cc | 436 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in GenDynamicKernel() local
740 auto kernel_mod = AnfAlgo::GetKernelMod(node); in SetKernelModStream() local
777 auto kernel_mod = AnfAlgo::GetKernelMod(kernels[i - 1]); in SetKernelModStream() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | kernel_actor.cc | 346 const auto &kernel_mod = kernel_info_->kernel_mod(); in FetchOutputDeviceTensor() local
|
| D | data_prepare_actor.cc | 66 const auto &kernel_mod = AnfAlgo::GetKernelMod(node); in FetchContinuousMemoryInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 179 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CreateKernelOutputDeviceAddress() local
204 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CreateKernelWorkspaceDeviceAddress() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 151 auto kernel_mod = AnfAlgo::GetKernelMod(anf_node_ptr); in LaunchKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 189 …auto kernel_mod = GenKernelMod(task_iter->second.input_size_list, task_iter->second.output_size_li… in TaskFinishProcess() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 1395 void AnfRuntimeAlgorithm::SetKernelMod(const KernelModPtr &kernel_mod, AnfNode *node) { in SetKernelMod()
2366 auto kernel_mod = GetKernelMod(kernel); in CacheAddrForGraph() local
2376 void AnfRuntimeAlgorithm::CacheAddrForKernel(const AnfNodePtr &node, kernel::KernelMod *kernel_mod)… in CacheAddrForKernel()
2434 …fRuntimeAlgorithm::CacheAddrForAtomicClean(const AnfNodePtr &node, kernel::KernelMod *kernel_mod) { in CacheAddrForAtomicClean()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 437 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitSomasOutputAndWorkspaceTensors() local
796 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RefNodeProcess() local
|