Home
last modified time | relevance | path

Searched refs:kernel_mode_ret (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dascend_kernel_compile.cc426 void AscendKernelCompileManager::QueryFusionFinishJob(KernelModMap *kernel_mode_ret) { in QueryFusionFinishJob() argument
428 MS_EXCEPTION_IF_NULL(kernel_mode_ret); in QueryFusionFinishJob()
449 (void)kernel_mode_ret->emplace(kernel_mode_item); in QueryFusionFinishJob()
660 KernelModMap kernel_mode_ret; in AscendFusionOpCompile() local
683 kernel_mode_ret[fusion_scope_iter.scope_id] = kernel_mod; in AscendFusionOpCompile()
720 (void)kernel_mode_ret.emplace(kernel_mode_item); in AscendFusionOpCompile()
725 QueryFusionFinishJob(&kernel_mode_ret); in AscendFusionOpCompile()
726 if (!build_manager_->GenSameFusionOpKernelMod(&kernel_mode_ret)) { in AscendFusionOpCompile()
729 return kernel_mode_ret; in AscendFusionOpCompile()
Dtbe_kernel_parallel_build.cc255 …lelBuildManager::GenSameFusionOpKernelMod(std::map<int64_t, KernelModPtr> *kernel_mode_ret) const { in GenSameFusionOpKernelMod()
256 MS_EXCEPTION_IF_NULL(kernel_mode_ret); in GenSameFusionOpKernelMod()
263 (*kernel_mode_ret)[task_info.scope_id] = kernel_mode; in GenSameFusionOpKernelMod()
Dascend_kernel_compile.h73 void QueryFusionFinishJob(KernelModMap *kernel_mode_ret);
Dtbe_kernel_parallel_build.h55 bool GenSameFusionOpKernelMod(std::map<int64_t, KernelModPtr> *kernel_mode_ret) const;