Home
last modified time | relevance | path

Searched refs:AKG_KERNEL (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_query.cc108 kernel_type = KernelType::AKG_KERNEL; in KernelQuery()
113 kernel_type = KernelType::AKG_KERNEL; in KernelQuery()
117 case KernelType::AKG_KERNEL: in KernelQuery()
Dkernel.h37 AKG_KERNEL, enumerator
/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/ascend/
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()
Dkernel_build_ascend.cc91 case KernelType::AKG_KERNEL: { in KernelBuildParallelCompile()
368 AnfAlgo::GetKernelType(anf_node) == KernelType::AKG_KERNEL) { in TbeClearZeroNodeFusion()
439 AnfAlgo::GetKernelType(anf_node) == KernelType::AKG_KERNEL) { in KernelBuildPreprocess()
Dkernel_select_ascend.cc587 new_kernel_type = AKG_KERNEL; in SetKernelInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc312 SetKernelInfo(cnode, KernelType::AKG_KERNEL); in SetGraphKernelInfo()
336 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in SetGraphKernelInfo()
439 kernel_type = AKG_KERNEL; in SetKernelInfo()
441 } else if (kernel_type == AKG_KERNEL) { in SetKernelInfo()
Dgpu_kernel_build.cc43 if (session::AnfRuntimeAlgorithm::GetKernelType(kernel) == KernelType::AKG_KERNEL) { in CreateGPUKernel()
Dgpu_kernel_runtime.cc762 if (session::AnfRuntimeAlgorithm::GetKernelType(kernel) != KernelType::AKG_KERNEL) { in LaunchKernelDynamic()
882 if (session::AnfRuntimeAlgorithm::GetKernelType(kernel) != KernelType::AKG_KERNEL) { in RunOpLaunchKernelDynamic()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dtask_desc_reporter.cc34 if (AnfAlgo::GetKernelType(node) != TBE_KERNEL && AnfAlgo::GetKernelType(node) != AKG_KERNEL) { in ReportData()
Dgraph_desc_reporter.cc26 if (AnfAlgo::GetKernelType(node) != TBE_KERNEL && AnfAlgo::GetKernelType(node) != AKG_KERNEL) { in ReportData()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dtsa_atomic_add_to_first_tensor.cc122 para_info_builder.SetKernelType(KernelType::AKG_KERNEL); in ProcessTsaFirstNode()
180 new_info_builder.SetKernelType(KernelType::AKG_KERNEL); in CorrectKernelBuildInfo()
Ddecrease_compute_precision.cc49 builder.SetKernelType(KernelType::AKG_KERNEL); in AddCastCNode()
187 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in Process()
Ddecrease_transfer_precision.cc172 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in Process_Father()
285 node_info_builder.SetKernelType(KernelType::AKG_KERNEL); in Process_Son()
Draise_reduction_precision.cc55 info_builder.SetKernelType(KernelType::AKG_KERNEL); in CreateReduceSum()
Dgraph_kernel_helper.cc297 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in BuildSelectKernelBuildInfo()
313 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in BuildSelectKernelBuildInfo()
343 para_info_builder.SetKernelType(KernelType::AKG_KERNEL); in SetNewKernelInfo()
367 graph_info_builder.SetKernelType(KernelType::AKG_KERNEL); in SetNewKernelInfo()
743 info_builder.SetKernelType(KernelType::AKG_KERNEL); in CreateCNode()
Dadd_atomic_clean.cc265 new_info_builder.SetKernelType(KernelType::AKG_KERNEL); in CorrectKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/
Dprofiling_utils.cc209 … if (AnfAlgo::GetKernelType(*iter) == TBE_KERNEL || AnfAlgo::GetKernelType(*iter) == AKG_KERNEL || in GetGraphLastKernelName()
386 if (AnfAlgo::GetKernelType(node) == TBE_KERNEL || AnfAlgo::GetKernelType(node) == AKG_KERNEL) { in ValidComputeGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc99 if (AnfAlgo::GetKernelType(anf_node_ptr) == KernelType::AKG_KERNEL) { in LaunchAddrCleanKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/
Ddata_dumper.cc207 AnfAlgo::GetKernelType(kernel) != AKG_KERNEL) { in KernelNeedDump()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc378 if (session::AnfRuntimeAlgorithm::GetKernelType(kernel) == KernelType::AKG_KERNEL) { in UpdateDynamicShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_decoder.cc265 builder->SetKernelType(KernelType::AKG_KERNEL); in CreateKernelInfo()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc647 builder.SetKernelType(AKG_KERNEL); in TEST_F()
649 EXPECT_EQ(AnfAlgo::GetKernelType(add), AKG_KERNEL); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc348 builder.SetKernelType(KernelType::AKG_KERNEL); in AddCastOpNodeToGraph()

12