p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!13503 Only continuous loads are eliminated 

From: @Margaret_wangrui
Reviewed-by: 
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2021-03-19 08:56:15 +08:00 committed by Gitee
parent 0fb8cd888d aeb43e5167
commit b22d4f99a5
1 changed files with 3 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mindspore/ccsrc/frontend/optimizer/irpass/load_eliminate.cc
View File

@ -24,27 +24,16 @@
#include "frontend/operator/ops.h"
namespace mindspore::opt::irpass {
namespace {
// Return true if the node has Ref abstract.
bool HasAbstractRef(const AnfNodePtr &node) {
if (node == nullptr) {
return false;
}
auto &abs = node->abstract();
return (abs != nullptr) && abs->isa<abstract::AbstractRef>();
}
} // namespace
AnfNodePtr LoadEliminater::operator()(const OptimizerPtr &, const AnfNodePtr &node) {
auto load_node = dyn_cast<CNode>(node);
if (load_node == nullptr || load_node->inputs().empty()) {
MS_LOG(WARNING) << "LoadEliminater encounter invalid node: " << node->DebugString();
return nullptr;
}
auto load_cnode = load_node->cast<CNodePtr>();
constexpr size_t kFirstInputIndex = 1;
auto &param = load_node->inputs().at(kFirstInputIndex);
if (!HasAbstractRef(param)) {
return param;
if (IsPrimitiveCNode(load_cnode->input(kFirstInputIndex), prim::kPrimLoad)) {
return load_cnode->input(kFirstInputIndex);
}
return nullptr;
}