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) {