Home
last modified time | relevance | path

Searched refs:GraphKernelFlags (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/context/
Dgraph_kernel_flags.h39 class GraphKernelFlags {
41 static const GraphKernelFlags &GetInstance() { in GetInstance()
42 static std::unique_ptr<GraphKernelFlags> flags(nullptr); in GetInstance()
45 flags.reset(new GraphKernelFlags(contexts.first, contexts.second)); in GetInstance()
57 GraphKernelFlags(const GraphKernelFlags &flags) = delete;
58 ~GraphKernelFlags() = default;
195 GraphKernelFlags(const std::string &graph_kernel_flags, bool enable_graph_kernel) in GraphKernelFlags() function
Dgraph_kernel_flags.cc153 void GraphKernelFlags::Refresh() { in Refresh()
171 void GraphKernelFlags::RegisterFlags(std::map<std::string, std::string> *flag_map) { in RegisterFlags()
213 std::string GraphKernelFlags::DumpAllFlags() const { in DumpAllFlags()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_pass_manager.h32 : PassManager(name, true), stage_(stage), flags_(context::GraphKernelFlags::GetInstance()) {} in GraphKernelPassManager()
46 const context::GraphKernelFlags &flags_;
Dgraph_kernel_optimization.cc127 …auto level = GetPassLevelByFlag(context::GraphKernelFlags::GetInstance().enable_trans_op_optimize); in HighLevelOpt1()
161 auto level = GetPassLevelByFlag(context::GraphKernelFlags::GetInstance().enable_stitch_fusion); in HighLevelOpt2()
165 …auto level_low_precision = GetPassLevelByFlag(context::GraphKernelFlags::GetInstance().enable_low_… in HighLevelOpt2()
179 auto level = GetPassLevelByFlag(context::GraphKernelFlags::GetInstance().enable_parallel_fusion); in Combine()
Dgraph_kernel_cluster.cc98 const auto &flags = context::GraphKernelFlags::GetInstance(); in GetClusterableOpList()
427 if (context::GraphKernelFlags::GetInstance().dump_as_text) { in CreateFuncGraph()
506 if (context::GraphKernelFlags::GetInstance().dump_as_text) { in Run()
Dgraph_kernel_expander.cc99 const auto &flags = context::GraphKernelFlags::GetInstance(); in GetExpandOps()
Darithmetic_simplify.cc534 const auto &flags = context::GraphKernelFlags::GetInstance(); in GetExpressions()
Dgraph_kernel_splitter.cc879 const auto &flags = context::GraphKernelFlags::GetInstance(); in CollectSplitFlags()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dmatmul_eltwise_fusion_pass.cc53 if (context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in MatchSingleFusionPattern()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc166 if (!context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in Optimize()
230 if (!context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in GraphKernelOptimize()
605 if (context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in UpdateOutputTensors()
Dascend_session.cc1258 if (!context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in GraphKernelOptimize()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc219 if (context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in OptimizeGraph()
280 if (!context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in FuseOperators()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_backend_optimization.cc498 if (!context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in AscendBackendUBFusionOptimization()
503 if (!context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { in AscendBackendUBFusionOptimization()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_build.cc579 auto &flags = context::GraphKernelFlags::GetInstance(); in CollectBuildAttrs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc471 enable_visit_kernel_cache_ = context::GraphKernelFlags::GetInstance().IsEnableGraphKernel(); in SetAllInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc77 enable_relation_cache_ = context::GraphKernelFlags::GetInstance().IsEnableGraphKernel(); in Init()