Home
last modified time | relevance | path

Searched refs:KernelType (Results 1 – 25 of 58) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc71 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
74 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
86 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
89 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
142 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
145 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
165 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
168 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
178 builder2.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
181 builder2.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.h31 KernelType kernel_type = KernelType::UNKNOWN_KERNEL_TYPE);
34 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type);
Dkernel_build_ascend.cc48 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in SerialCompileImpl()
50 case KernelType::AICPU_KERNEL: { in SerialCompileImpl()
54 case KernelType::HOST_KERNEL: { in SerialCompileImpl()
58 case KernelType::RT_KERNEL: { in SerialCompileImpl()
62 case KernelType::HCCL_KERNEL: { in SerialCompileImpl()
83 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in KernelBuildParallelCompile()
85 case KernelType::TBE_KERNEL: { in KernelBuildParallelCompile()
91 case KernelType::AKG_KERNEL: { in KernelBuildParallelCompile()
177 builder->SetKernelType(KernelType::TBE_KERNEL); in AddTbeClearZeroNode()
204 builder->SetKernelType(KernelType::TBE_KERNEL); in AddFusionTbeClearZeroNode()
[all …]
Dkernel_select_graph_kernel.cc88 kernel::KernelQuery(cnode, &kernel_info_list, KernelType::AKG_KERNEL); in UpdateKernelInfo()
315 SelectKernelInfo(cnode, KernelType::AKG_KERNEL); in UpdateEquivFormat()
338 SelectKernelInfo(trans_node, KernelType::AKG_KERNEL); in UpdateEquivFormat()
444 SelectKernelInfo(node_list[i]->cast<CNodePtr>(), KernelType::AKG_KERNEL); in UpdateFormatsAndDtypes()
473 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in SetGraphKernelInfo()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc66 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
69 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
109 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
112 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
167 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
170 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
221 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
224 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
266 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
269 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_query.cc99 KernelType kernel_type) { in KernelQuery()
108 kernel_type = KernelType::AKG_KERNEL; in KernelQuery()
113 kernel_type = KernelType::AKG_KERNEL; in KernelQuery()
117 case KernelType::AKG_KERNEL: in KernelQuery()
Dkernel_query.h29 KernelType kernel_type = KernelType::UNKNOWN_KERNEL_TYPE);
Dkernel_build_info.h51 KernelType kernel_type() const { return kernel_type_; } in kernel_type()
108 KernelType kernel_type_;
149 void SetKernelType(const KernelType &kernel_type);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dfusion_base_pass.cc33 return AnfAlgo::GetKernelType(node) == KernelType::TBE_KERNEL && in CheckEltWiseNode()
48 return AnfAlgo::GetKernelType(node) == KernelType::TBE_KERNEL && in CheckDoubleInEltWiseNode()
63 return AnfAlgo::GetKernelType(node) == KernelType::TBE_KERNEL && in CheckMultiOutputEltWiseNode()
Dmatmul_eltwise_fusion_pass.cc54 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
62 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dconv_single_in_fusion_pass.cc49 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchConvSingleInEltwise()
68 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dconv_double_in_fusion_pass.cc45 if (AnfAlgo::GetKernelType(double_in_eltwise_input) == KernelType::TBE_KERNEL && in MatchConvDoubleInEltwise()
64 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dsegment_eltwise_fusion_pass.cc45 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchSegmentEltwise()
78 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dreduce_eltwise_fusion_pass.cc47 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchReduceEltwise()
81 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dbnupdate_eltwise_fusion_pass.cc58 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Deltwise_fusion_pass.cc67 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dmulti_output_fusion_pass.cc72 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dinsert_cast_test.cc63 builder.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
71 builder1.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
125 builder1.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dallreduce_fusion_test.cc59 builder.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
100 builder.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
141 builder.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
198 builder.SetKernelType(KernelType::AKG_KERNEL); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.h70 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type = KernelType::UNKNOWN_KERNEL…
Dgpu_kernel_build.cc43 if (session::AnfRuntimeAlgorithm::GetKernelType(kernel) == KernelType::AKG_KERNEL) { in CreateGPUKernel()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbn_split_test.cc86 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
135 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
Dtransdata_split_test.cc125 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
173 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.h87 void ResetKernelInfo(const AnfNodePtr &node, KernelType kernel_type = KernelType::UNKNOWN_KERNEL_TY…
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc84 selected_kernel_builder.SetKernelType(KernelType::RT_KERNEL); in CreateSendApplyKernel()
104 selected_kernel_builder.SetKernelType(KernelType::RT_KERNEL); in CreateRecvApplyKernel()
568 selected_kernel_builder.SetKernelType(KernelType::RT_KERNEL); in CreateMngKernelBuilder()
647 selected_kernel_builder.SetKernelType(KernelType::AICPU_KERNEL); in CreateEndOfSequenceOP()
699 selected_kernel_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateStreamAssignAddnOP()
908 selected_kernel_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateNPUGetFloatStatus()
931 selected_kernel_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateNPUClearStatus()
953 selected_kernel_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateNPUAllocStatus()
984 selected_kernel_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateAssignAdd()
1033 selected_kernel_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateAssign()

123