Home
last modified time | relevance | path

Searched refs:KernelGraphPtr (Results 1 – 25 of 77) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.h112 void AssignStream(const NotNull<KernelGraphPtr> &graph_ptr);
122 …CNodePtr CreateSendApplyKernel(const NotNull<KernelGraphPtr> &graph_ptr, uint32_t event_id, uint32…
123 …CNodePtr CreateRecvApplyKernel(const NotNull<KernelGraphPtr> &graph_ptr, uint32_t event_id, uint32…
124 void CheckResourceAssign(const NotNull<KernelGraphPtr> &graph_ptr);
125 void CheckStreamAssign(const NotNull<KernelGraphPtr> &graph_ptr);
126 void CheckEventAssign(const NotNull<KernelGraphPtr> &graph_ptr);
127 void AssignAllNodesStream(const NotNull<KernelGraphPtr> &graph_ptr);
129 void AssignHcom(const NotNull<KernelGraphPtr> &graph_ptr);
131 void AssignIndependent(const NotNull<KernelGraphPtr> &graph_ptr);
133 void UpdateAtomicAddrCleanStreamId(const NotNull<KernelGraphPtr> &graph_ptr);
[all …]
Dascend_stream_assign.cc154 CNodePtr GetHcomAndOverflowMarker(const NotNull<KernelGraphPtr> &graph_ptr, vector<CNodePtr> *hcom_… in GetHcomAndOverflowMarker()
207 void SetNodeStreamIDAttr(const NotNull<KernelGraphPtr> &graph_ptr) { in SetNodeStreamIDAttr()
215 void AscendStreamAssign::AssignStream(const NotNull<KernelGraphPtr> &graph_ptr) { in AssignStream()
266 void AscendStreamAssign::ReorderIndependentOrders(const NotNull<KernelGraphPtr> &graph_ptr) { in ReorderIndependentOrders()
324 void AscendStreamAssign::CheckScenario(const NotNull<KernelGraphPtr> &graph_ptr, in CheckScenario()
380 const NotNull<KernelGraphPtr> &graph_ptr) { in GetCNodesNeededMoved()
435 … const CNodePtr first_node, const NotNull<KernelGraphPtr> &graph_ptr) { in GetTargetOutputNode()
527 void AscendStreamAssign::TrailingTimeOptimizationByReorder(const NotNull<KernelGraphPtr> &graph_ptr… in TrailingTimeOptimizationByReorder()
588 void AscendStreamAssign::AssignAllNodesStream(const NotNull<KernelGraphPtr> &graph_ptr) { in AssignAllNodesStream()
666 void AscendStreamAssign::AssignHcom(const NotNull<KernelGraphPtr> &graph_ptr) { in AssignHcom()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.h51 void UnifyMindIR(const KernelGraphPtr &graph) override;
64 KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
91 void CompileChildGraph(const KernelGraphPtr &child_graph);
102 void AssignStream(NotNull<KernelGraphPtr> kernel_graph) const;
119 void DumpAllGraphs(const std::vector<KernelGraphPtr> &all_graphs);
124 …void RootGraphExecutorValidate(NotNull<KernelGraphPtr> graph, const std::vector<KernelGraphPtr> &a…
137 …void CreateMultiBranchOutput(NotNull<KernelGraphPtr> graph, NotNull<std::set<KernelGraphPtr> *> me…
138 void SelectKernel(NotNull<KernelGraphPtr> root_graph);
139 …void RecurseSelectKernelInfo(NotNull<KernelGraphPtr> graph, NotNull<std::set<KernelGraphPtr> *> co…
141 void IrFusionPass(const NotNull<KernelGraphPtr> graph, NotNull<std::set<KernelGraphPtr> *> memo);
[all …]
Dsession_basic.h124 std::vector<KernelGraphPtr> *all_out_graph);
133 void AssignParamKey(const KernelGraphPtr &kernel_graph);
134 …void InitPSParamAndOptim(const KernelGraphPtr &kernel_graph, const std::vector<tensor::TensorPtr> …
147 KernelGraphPtr GetGraph(GraphId graph_id) const;
155 void RunOpRemoveNopNode(const KernelGraphPtr &kernel_graph) const;
156 static void RunOpHideNopNode(const KernelGraphPtr &kernel_graph);
188 …void CreateOutputPlaceholder(const KernelGraphPtr &kernel_graph, const std::vector<tensor::TensorP…
219 virtual void UnifyMindIR(const KernelGraphPtr &graph);
220 virtual void FinalOptimize(const KernelGraphPtr &graph) const;
230 virtual KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl()
[all …]
Dascend_auto_monad.cc40 using GraphArgPair = std::pair<KernelGraphPtr, std::vector<AnfNodePtr>>;
68 void DumpAllGraphs(NotNull<KernelGraphPtr> kg, std::set<KernelGraphPtr> *memo) { in DumpAllGraphs()
83 void DumpGraphForDebug(const NotNull<KernelGraphPtr> kg) { in DumpGraphForDebug()
85 std::set<KernelGraphPtr> memo; in DumpGraphForDebug()
92 void DumpExecuteOrder(const NotNull<KernelGraphPtr> kg) { in DumpExecuteOrder()
145 uint32_t GetGraphLabel(const KernelGraphPtr &kg) { in GetGraphLabel()
202 KernelGraphPtr graph;
266 explicit ParameterPool(const KernelGraphPtr &top_graph) : top_graph_(top_graph) {} in ParameterPool()
293 const KernelGraphPtr &top_graph_;
304 void MarkVisited(const KernelGraphPtr &kg) { visited_graphs_.insert(kg); } in MarkVisited()
[all …]
Dascend_session.cc154 void HandleControlFlow(NotNull<KernelGraphPtr> graph) { in HandleControlFlow()
159 void SetStreamDistinctionLabel(const KernelGraphPtr &graph, uint32_t label, bool is_override) { in SetStreamDistinctionLabel()
178 void GenOpOutputStubTensor(const KernelGraphPtr &single_op_graph, const CNodePtr &kernel, in GenOpOutputStubTensor()
365 void AddGraphToManager(const NotNull<KernelGraphPtr> graph, NotNull<FuncGraphManagerPtr> manager, in AddGraphToManager()
366 NotNull<std::set<KernelGraphPtr> *> memo) { in AddGraphToManager()
381 void AscendSession::UnifyMindIR(const KernelGraphPtr &graph) { in UnifyMindIR()
538 std::vector<KernelGraphPtr> all_graphs; in CompileGraphImpl()
559 std::set<KernelGraphPtr> memo; in CompileGraphImpl()
713 void AscendSession::CompileChildGraph(const KernelGraphPtr &child_graph) { in CompileChildGraph()
792 KernelGraphPtr AscendSession::BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info, in BuildOpImpl()
[all …]
Dascend_auto_monad.h28 explicit AscendAutoMonad(NotNull<KernelGraphPtr> kg) : kernel_graph_(kg) {} in AscendAutoMonad()
38 NotNull<KernelGraphPtr> kernel_graph_;
Dgpu_session.h40 void UnifyMindIR(const KernelGraphPtr &graph) override { SessionBasic::UnifyMindIR(graph); } in UnifyMindIR()
48 KernelGraphPtr BuildOpImpl(const OpRunInfo &op_run_info, const GraphInfo &graph_info,
101 GraphId CompileGraphImpl(const KernelGraphPtr &kernel_graph);
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger.h77 void PreExecuteGraphDebugger(const std::vector<KernelGraphPtr> &graphs);
81 void PreExecute(const KernelGraphPtr &graph_ptr);
91 void Dump(const KernelGraphPtr &kernel_graph) const;
95 void DumpSetup(const KernelGraphPtr &kernel_graph) const;
97 void DumpInGraphCompiler(const KernelGraphPtr &kernel_graph);
132 void LoadParametersAndConst(const KernelGraphPtr &graph);
144 void LoadGraphs(const KernelGraphPtr &graph_ptr);
148 void SetGraphPtr(const KernelGraphPtr &graph_ptr) { graph_ptr_ = graph_ptr; } in SetGraphPtr()
150 const KernelGraphPtr GetGraphPtr() const { return graph_ptr_; } in GetGraphPtr()
152 const std::list<KernelGraphPtr> GetGraphPtrList() const { return graph_ptr_list_; } in GetGraphPtrList()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.h59 …GraphCompilerInfo(const std::vector<KernelGraphPtr> &graphs, const std::vector<DeviceContext *> &d… in GraphCompilerInfo()
79 std::vector<KernelGraphPtr> graphs_;
108 KernelGraphPtr Fetch(GraphId graph_id) const;
111 KernelGraphPtr Fetch(const GraphInfo &graph_info) const;
118 void GetParamAndOutputIndex(const KernelGraphPtr &graph, const std::vector<TensorPtr> &inputs,
138 …void CalculateRefCount(const KernelGraphPtr &graph, std::map<KernelWithIndex, size_t> *ref_count) …
165 void Summary(const std::vector<KernelGraphPtr> &graphs) const;
175 GraphId CompileGraphImpl(const KernelGraphPtr &graph, const DeviceContext *device_context) const;
178 void CreateDeviceAddress(const KernelGraphPtr &graph, const DeviceContext *device_context) const;
181 std::unordered_map<GraphInfo, KernelGraphPtr> run_op_graphs_;
Dcontrol_node_parser.h75 KernelWithIndex GetFrontNodeByKernelGraph(const AnfNodePtr &backend_node, const KernelGraphPtr &gra…
84 std::vector<KernelWithIndex> FetchParameterbyKernelGraph(const KernelGraphPtr &graph);
90 …void Parse(const std::vector<AnfNodePtr> &control_nodes, const std::vector<KernelGraphPtr> &graphs,
120 bool IsCallInputKernelGraph(const KernelGraphPtr &graph);
144 …tValueNode(const std::vector<AnfNodePtr> &control_nodes, const std::vector<KernelGraphPtr> &graphs,
154 void FetchFrontToBackendParameter(const std::vector<KernelGraphPtr> &graphs,
159 void FetchFrontToBackendKernel(const std::vector<KernelGraphPtr> &graphs,
180 void FetchCallInputKernelGraph(const std::vector<KernelGraphPtr> &graphs,
183 void FetchBackendInputNode(const std::vector<KernelGraphPtr> &graphs,
188 void FetchBackendParameterNode(const std::vector<KernelGraphPtr> &graphs,
[all …]
Dgraph_scheduler.h94 void Clear(const ActorInfo &actor_info, const std::vector<KernelGraphPtr> &graphs) noexcept;
145 … const KernelGraphPtr &graph, const KernelWithIndex &from_kernel_with_output_idx,
153 … const KernelWithIndex &to_kernel_with_input_idx, const KernelGraphPtr &graph);
156 … const KernelWithIndex &to_kernel_with_input_idx, const KernelGraphPtr &graph);
159 … const KernelWithIndex &to_kernel_with_input_idx, const KernelGraphPtr &graph);
162 … const KernelWithIndex &to_kernel_with_input_idx, const KernelGraphPtr &graph);
165 … const KernelWithIndex &to_kernel_with_input_idx, const KernelGraphPtr &graph);
172 …lArrowByAutoMonad(KernelActor *to_actor, const AnfNodePtr &from_node, const KernelGraphPtr &graph);
176 void LinkControlArrowBySendRecvNodes(const KernelGraphPtr &graph);
213 … const std::vector<KernelGraphPtr> &graphs, const ControlNodeParserPtr &parser);
[all …]
Dgraph_compiler.cc57 void CreateParameterDeviceAddress(const DeviceContext *device_context, const KernelGraphPtr &graph)… in CreateParameterDeviceAddress()
146 void CreateValueNodeDeviceAddress(const DeviceContext *device_context, const KernelGraphPtr &graph)… in CreateValueNodeDeviceAddress()
170 void CreateKernelOutputDeviceAddress(const DeviceContext *device_context, const KernelGraphPtr &gra… in CreateKernelOutputDeviceAddress()
195 void CreateKernelWorkspaceDeviceAddress(const DeviceContext *device_context, const KernelGraphPtr &… in CreateKernelWorkspaceDeviceAddress()
214 void UpdateDeviceAddressForInplaceNode(const KernelGraphPtr &graph) { in UpdateDeviceAddressForInplaceNode()
298 KernelGraphPtr graph = session_->ConstructKernelGraph(nodes, outputs); in CompileGraph()
312 GraphId GraphCompiler::CompileGraphImpl(const KernelGraphPtr &graph, const DeviceContext *device_co… in CompileGraphImpl()
399 …KernelGraphPtr graph = session_->ConstructSingleOpGraph(op_run_info, *input_tensors, *tensors_mask… in CompileGraph()
428 KernelGraphPtr GraphCompiler::Fetch(GraphId graph_id) const { in Fetch()
433 KernelGraphPtr GraphCompiler::Fetch(const GraphInfo &graph_info) const { in Fetch()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.h47 void OptimizeGraph(const KernelGraphPtr &graph) const override;
48 void OptimizeSingleOpGraph(const KernelGraphPtr &graph) const override;
53 void PreprocessBeforeRunGraph(const KernelGraphPtr &graph) const override;
54 void PreprocessBeforeRunSingleOpGraph(const KernelGraphPtr &graph) const override;
64 void UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &graph) const;
66 void OptimizeGraphImpl(const KernelGraphPtr &graph) const;
Dcpu_device_context.cc107 void CPUDeviceContext::OptimizeGraph(const KernelGraphPtr &graph) const { in OptimizeGraph()
118 void CPUDeviceContext::OptimizeSingleOpGraph(const KernelGraphPtr &graph) const { in OptimizeSingleOpGraph()
124 void CPUDeviceContext::OptimizeGraphImpl(const KernelGraphPtr &graph) const { in OptimizeGraphImpl()
134 void CPUDeviceContext::UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &graph) const { in UpdateGraphDynamicShapeAttr()
209 void ProcessCast(const KernelGraphPtr &graph) { in ProcessCast()
222 void CPUDeviceContext::PreprocessBeforeRunGraph(const KernelGraphPtr &graph) const { in PreprocessBeforeRunGraph()
232 void CPUDeviceContext::PreprocessBeforeRunSingleOpGraph(const KernelGraphPtr &graph) const { Proces… in PreprocessBeforeRunSingleOpGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.h52 void OptimizeGraph(const KernelGraphPtr &graph) const override;
54 void OptimizeSingleOpGraph(const KernelGraphPtr &graph) const override;
79 void OptimizeGraphWithoutDeviceInfo(const KernelGraphPtr &graph) const;
81 void OptimizeGraphWithDeviceInfo(const KernelGraphPtr &graph) const;
84 void FuseOperators(const KernelGraphPtr &graph) const;
87 void UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &graph) const;
Dgpu_device_context.cc204 void GPUDeviceContext::OptimizeGraph(const KernelGraphPtr &graph) const { in OptimizeGraph()
228 void GPUDeviceContext::OptimizeGraphWithoutDeviceInfo(const KernelGraphPtr &graph) const { in OptimizeGraphWithoutDeviceInfo()
237 void GPUDeviceContext::OptimizeGraphWithDeviceInfo(const KernelGraphPtr &graph) const { in OptimizeGraphWithDeviceInfo()
270 void GPUDeviceContext::FuseOperators(const KernelGraphPtr &graph) const { in FuseOperators()
294 void GPUDeviceContext::UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &graph) const { in UpdateGraphDynamicShapeAttr()
306 void RunOpOptimize(const KernelGraphPtr &kernel_graph) { in RunOpOptimize()
317 void RunOpHardwareOptimize(const KernelGraphPtr &kernel_graph) { in RunOpHardwareOptimize()
327 void RunOpHideNopNode(const KernelGraphPtr &kernel_graph) { in RunOpHideNopNode()
335 void RunOpRemoveNopNode(const KernelGraphPtr &kernel_graph) { in RunOpRemoveNopNode()
344 void GPUDeviceContext::OptimizeSingleOpGraph(const KernelGraphPtr &graph) const { in OptimizeSingleOpGraph()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_atomic_clean.h81 …virtual CNodePtr CreateAtomicCleanCompositeNode(const KernelGraphPtr &main_graph, TypeId dst_type);
84 …void InsertAtomicClean(const KernelGraphPtr &main_graph, const AnfNodePtr &anf_node, const FuncGra…
85 …CNodePtr InsertUpdateState(const KernelGraphPtr &main_graph, const CNodePtr &composite_node) const;
88 void ProcessOriginCNodeUser(const KernelGraphPtr &main_graph, const AnfNodePtr &composite_node,
97 std::vector<std::pair<AnfNodePtr, int>> FindOriginCNodeUsers(const KernelGraphPtr &main_graph,
101 bool IsExistStructuralObstacle(const KernelGraphPtr &main_graph, const AnfNodePtr &node,
Dtsa_atomic_add_to_first_tensor.h55 …void ProcessTsa(const KernelGraphPtr &main_graph, const AnfNodePtr &anf_node, const FuncGraphManag…
56 AnfNodePtr ProcessTsaFirstNode(const KernelGraphPtr &main_graph, const AnfNodePtr &node);
57 AnfNodePtr FindTsaFirstRealInputInGraph(const KernelGraphPtr &main_graph, const AnfNodePtr &node);
Dgraph_kernel_optimization.h29 void Run(const KernelGraphPtr &kernel_graph);
51 void GraphKernelOptimize(const KernelGraphPtr &kernel_graph);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_scalar_to_tensor.cc27 AnfNodePtr CreateTensorInput(const KernelGraphPtr &kernel_graph, const AnfNodePtr &input_node) { in CreateTensorInput()
65 return CreateTensorInput(func_graph->cast<KernelGraphPtr>(), node); in Process()
74 auto new_input = CreateTensorInput(func_graph->cast<KernelGraphPtr>(), cnode->inputs()[i]); in Process()
80 auto kernel_graph = func_graph->cast<KernelGraphPtr>(); in Process()
Dconvert_tuple_output_to_maketuple.cc36 auto kernel_graph = graph->cast<KernelGraphPtr>(); in ConvertTupleInputToMakeTuple()
39 kernel_graph = anf_graph->cast<KernelGraphPtr>(); in ConvertTupleInputToMakeTuple()
87 auto kernel_graph = graph->cast<KernelGraphPtr>(); in Process()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/
Ddevice_context.h78 virtual void OptimizeGraph(const KernelGraphPtr &graph) const {} in OptimizeGraph()
81 virtual void OptimizeSingleOpGraph(const KernelGraphPtr &graph) const {} in OptimizeSingleOpGraph()
93 virtual void PreprocessBeforeRunGraph(const KernelGraphPtr &graph) const {} in PreprocessBeforeRunGraph()
95 virtual void PreprocessBeforeRunSingleOpGraph(const KernelGraphPtr &graph) const {} in PreprocessBeforeRunSingleOpGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dactor_common.h104 bool IsHostQueueDSActor(const AnfNodePtr &node, const KernelGraphPtr &graph = nullptr,
117 bool IsInternalParameter(const AnfNodePtr &node, const KernelGraphPtr &graph);
134 AnfNodePtr FetchFrontNodeByBackendNode(const AnfNodePtr &backend_node, const KernelGraphPtr &graph);
136 const KernelGraphPtr &graph);
Dactor_common.cc57 bool IsHostQueueDSActor(const AnfNodePtr &node, const KernelGraphPtr &graph, in IsHostQueueDSActor()
91 bool IsInternalParameter(const AnfNodePtr &node, const KernelGraphPtr &graph) { in IsInternalParameter()
186 AnfNodePtr FetchFrontNodeByBackendNode(const AnfNodePtr &backend_node, const KernelGraphPtr &graph)… in FetchFrontNodeByBackendNode()
205 const KernelGraphPtr &graph) { in FetchFrontNodeWithIndexByGraphOutput()

1234