Browse Source

!9605 Erase kAttrPynativeNextOpName and kAttrPynativeNextIndex after SelectKernel

From: @HulkTang
Reviewed-by: @lianliguang,@jjfeing,@chujinjin
Signed-off-by: @jjfeing
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
00a64bfa7c
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ccsrc/backend/session/ascend_session.cc

+ 2
- 0
mindspore/ccsrc/backend/session/ascend_session.cc View File

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


Loading…
Cancel
Save