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

!1520 perf_opt(manager): avoid generate anonymous function in every acquire node loop. 

Merge pull request !1520 from lyvette/master
This commit is contained in:
mindspore-ci-bot 2020-05-28 09:30:57 +08:00 committed by Gitee
parent a5161a969f 1daad56fd1
commit 343f3c572a
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/ir/manager.cc
View File

@ -87,6 +87,8 @@ void FuncGraphManager::Reset() {
func_graphs_used_total_ = std::make_shared<FuncGraphsUsedTotalComputer>(this);
recursive_ = std::make_shared<RecursiveComputer>(this);
j_total_ = std::make_shared<FuncGraphJTotalComputer>(this);
limit_ = std::bind(&FuncGraphManager::Limit, this, std::placeholders::_1);
}
void FuncGraphManager::Init() {
@ -359,9 +361,7 @@ IncludeType FuncGraphManager::Limit(const AnfNodePtr &node) {
void FuncGraphManager::AcquireNodes(const std::vector<AnfNodePtr> &nodes) {
AnfNodeSet acq;
for (auto &node : nodes) {
std::function<IncludeType(AnfNodePtr)> limit = std::bind(&FuncGraphManager::Limit, this, std::placeholders::_1);
AnfNodeSet new_nodes = AnfNodeSet(DeepScopedGraphSearch(node, limit));
AnfNodeSet new_nodes = AnfNodeSet(DeepScopedGraphSearch(node, limit_));
all_nodes_.update(new_nodes);
acq.update(new_nodes);

1
mindspore/ccsrc/ir/manager.h
View File

@ -515,6 +515,7 @@ class FuncGraphManager : public std::enable_shared_from_this<FuncGraphManager> {
std::shared_ptr<FuncGraphJTotalComputer> j_total_;
bool is_manage_;
std::function<IncludeType(AnfNodePtr)> limit_;
};
class FuncGraphTransaction {