From 468fcaef0d8c9256219aa70f6847bed5349b3383 Mon Sep 17 00:00:00 2001
From: zhoufeng <zhoufeng54@huawei.com>
Date: Wed, 9 Mar 2022 17:55:16 +0800
Subject: [PATCH] fix func graph is null

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
---
 mindspore/ccsrc/frontend/optimizer/comm_op_attrs.cc | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/mindspore/ccsrc/frontend/optimizer/comm_op_attrs.cc b/mindspore/ccsrc/frontend/optimizer/comm_op_attrs.cc
index befad1a545c..8717e2bffa7 100644
--- a/mindspore/ccsrc/frontend/optimizer/comm_op_attrs.cc
+++ b/mindspore/ccsrc/frontend/optimizer/comm_op_attrs.cc
@@ -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;
     }