Home
last modified time | relevance | path

Searched refs:kProcessorAiCore (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_convert_utils.h28 constexpr auto kProcessorAiCore = "aicore"; variable
Dtbe_convert_utils.cc110 device = kProcessorAiCore; in GetProcessor()
Dtbe_kernel_parallel_build.cc118 task_info.processor = tbe::kProcessorAiCore; in SavePreBuildTaskInfo()
134 task_info.processor = tbe::kProcessorAiCore; in SaveTaskInfo()
Dtbe_utils.cc243 if (processor != kProcessorAiCore) { in InsertCache()
Dascend_kernel_compile.cc691 … build_manager_->SaveSameFusionOpInfo(fusion_scope_iter.scope_id, json_name, tbe::kProcessorAiCore, in AscendFusionOpCompile()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_build.cc34 return tbe::TbeUtils::InsertCache(kernel_name, kProcessorAiCore, true); in AkgInsertCache()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc239 if (processor == kProcessorAiCore || processor == kProcessorAiCpu) { in InsertCache()
515 if (processor == kProcessorAiCore) return Processor::AICORE; in GetProcessor()
527 device = kProcessorAiCore; in GetProcessor()
847 processor = kProcessorAiCore; in GetProcessorStr()
885 str_processor = kernel::kProcessorAiCore; in GetStrProcessorFromContext()
Dkernel_fusion.cc105 …r->SaveSameFusionOpInfo(fusion_scope_iter.scope_id, json_name, tbe::kProcessorAiCore, input_size_l… in KernelFusion()
Dcommon_utils.h38 constexpr auto kProcessorAiCore = "aicore"; variable