Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!31051 fix func graph is null 

Merge pull request !31051 from zhoufeng/fix-func-graph-is-null
This commit is contained in:
i-robot 2022-03-10 01:18:15 +00:00 committed by Gitee
parent f57f862d7b 468fcaef0d
commit 3d39787cd9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/frontend/optimizer/comm_op_attrs.cc
View File

@ -35,6 +35,13 @@ void CommOpAttrs(const FuncGraphPtr &graph) {
std::vector<AnfNodePtr> all_nodes = TopoSort(return_node);
for (auto &node : all_nodes) {
MS_EXCEPTION_IF_NULL(node);
if (!node->isa<CNode>()) {
continue;
}
auto primitive = GetCNodePrimitive(node);
if (primitive == nullptr) {
continue;
}
if (!common::AnfAlgo::IsCommunicationOp(node)) {
continue;
}