Erase kAttrPynativeNextOpName and kAttrPynativeNextIndex after SelectKernel

4 years ago · acd910323e
parent c8dec08070
commit acd910323e
1 changed files with 2 additions and 0 deletions
--- 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) {