Home
last modified time | relevance | path

Searched refs:GetKernelMod (Results 1 – 25 of 31) 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()
373 auto kernel_mode = AnfAlgo::GetKernelMod(cnode); in FetchMemUnitSize()
619 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelOutputAddress()
638 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in InitKernelWorkspaceAddress()
681 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelOutputAddress()
703 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ClearKernelWorkspaceAddress()
753 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelDynamic()
850 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithoutMock()
877 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpLaunchKernelDynamic()
913 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithTimeProfiling()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dhccl_dynamic_kernel.cc42 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs()
81 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in StaticShapeExecute()
Dai_core_dynamic_kernel.cc107 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_simple_mem_plan.cc51 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemPlan()
97 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in MemAssign()
Dcpu_memory_manager.cc182 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in IncreaseAddressRefCount()
207 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in DecreaseAddressRefCount()
Dcpu_kernel_runtime.cc159 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in AssignKernelOutputAddress()
449 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel.cc29 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in UpdateArgs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc41 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicOutputKernelRef()
106 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_cnode); in InitDynamicWorkspaceKernelRef()
217 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetOutputMap()
232 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SetWkMap()
Dmem_reuse_checker.cc114 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriDy()
129 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CalculOriWk()
221 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_input); in CheckGraphOutputAssigned()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dtask_desc_reporter.cc39 auto kernel_mod = AnfAlgo::GetKernelMod(node); in ReportData()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_mul.cc55 return AnfAlgo::GetKernelMod(mul_graph_->execution_order()[0]); in ObtainLaunchMulKernelMod()
Dkernel_runtime.cc278 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in ResetNodeAddress()
354 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in RunOpClearMemory()
440 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignOutputMemory()
488 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in RunOpAssignWorkSpaceMemory()
751 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignCommunicationNodeOutputMem()
823 auto kernel_mod = AnfAlgo::GetKernelMod(anf_node); in PreAssignCNodeMemory()
903 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignNodeOutputMem()
1106 auto kernel_mod = AnfAlgo::GetKernelMod(node); in AssignWorkSpaceMem()
1310 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in AssignKernelAddress()
1347 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in SyncNodeOutputTensors()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc239 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernel()
276 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithProfiling()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc375 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in UpdateDynamicShape()
400 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernel()
456 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in LaunchKernelWithProfiling()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc86 if (AnfAlgo::GetKernelMod(anf_node) == nullptr) { in KernelBuildParallelCompile()
133 auto kernel_mod = AnfAlgo::GetKernelMod(pre_node); in CalCleanZerosSize()
218 auto kernel_mod = AnfAlgo::GetKernelMod(kernel_node); in IsAtomicNode()
Dascend_launch_atomic_clean.cc51 kernel_mod_ = AnfAlgo::GetKernelMod(atomic_clean_graph_->execution_order()[0]); in LaunchOpKernel()
Dascend_launch_transdata.cc53 kernel_mod_ = AnfAlgo::GetKernelMod(transdata_graph_->execution_order()[0]); in LaunchOpKernel()
Dascend_kernel_runtime.cc436 auto kernel_mod = AnfAlgo::GetKernelMod(cnode); in GenDynamicKernel()
740 auto kernel_mod = AnfAlgo::GetKernelMod(node); in SetKernelModStream()
777 auto kernel_mod = AnfAlgo::GetKernelMod(kernels[i - 1]); in SetKernelModStream()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc697 TEST_F(AnfRuntimeAlgorithmTest, GetKernelMod) { in TEST_F() argument
707 EXPECT_EQ(AnfAlgo::GetKernelMod(add), nullptr); in TEST_F()
708 EXPECT_THROW(AnfAlgo::GetKernelMod(nullptr), std::runtime_error); in TEST_F()
718 EXPECT_EQ(AnfAlgo::GetKernelMod(add), nullptr); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc179 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CreateKernelOutputDeviceAddress()
204 auto kernel_mod = AnfAlgo::GetKernelMod(kernel); in CreateKernelWorkspaceDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_parallel_build.cc59 if (AnfAlgo::GetKernelMod(anf_node) != nullptr) { in TbeOpParallelBuild()
Dascend_kernel_compile.cc528 if (AnfAlgo::GetKernelMod(anf_node) == nullptr) { in GetAllAscendNodes()
594 if (AnfAlgo::GetKernelMod(node) != nullptr && !is_tune_flag_) { in AscendSingleOpCompile()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc151 auto kernel_mod = AnfAlgo::GetKernelMod(anf_node_ptr); in LaunchKernel()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h232 static kernel::KernelMod *GetKernelMod(const AnfNodePtr &node);
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc66 const auto &kernel_mod = AnfAlgo::GetKernelMod(node); in FetchContinuousMemoryInfo()

12