yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!9605 Erase kAttrPynativeNextOpName and kAttrPynativeNextIndex after SelectKernel 

From: @HulkTang
Reviewed-by: @lianliguang,@jjfeing,@chujinjin
Signed-off-by: @jjfeing
This commit is contained in:
mindspore-ci-bot 2020-12-08 09:11:22 +08:00 committed by Gitee
parent 9a6d79f52f acd910323e
commit 00a64bfa7c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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