From acd910323e8c6db3b872b3d16b76cfaf9ad7ed88 Mon Sep 17 00:00:00 2001
From: tanghuikang <tanghuikang@huawei.com>
Date: Mon, 7 Dec 2020 20:11:33 +0800
Subject: [PATCH] Erase kAttrPynativeNextOpName and kAttrPynativeNextIndex
 after SelectKernel

---
 mindspore/ccsrc/backend/session/ascend_session.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc
index 4dac2a5d9e..b680b6b782 100644
--- a/mindspore/ccsrc/backend/session/ascend_session.cc
+++ b/mindspore/ccsrc/backend/session/ascend_session.cc
@@ -771,6 +771,8 @@ void AscendSession::SelectKernel(const KernelGraph &kernel_graph) const {
   size_t reduce_precision_count = 0;
   for (const auto &cnode : kernel_graph.execution_order()) {
     auto status = device::ascend::SelectKernelInfo(cnode);
+    AnfAlgo::EraseNodeAttr(kAttrPynativeNextOpName, cnode);
+    AnfAlgo::EraseNodeAttr(kAttrPynativeNextIndex, cnode);
     if (status == device::ascend::kStatusRaisePrecision) {
       raise_precision_count++;
     } else if (status == device::ascend::kStatusReducePrecision) {