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

!3636 Fix opt unnecessary catch 

Merge pull request !3636 from Kang/opt
This commit is contained in:
mindspore-ci-bot 2020-07-30 18:24:12 +08:00 committed by Gitee
parent 699e616b3a f1e639a86d
commit f87983833d
3 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/backend/optimizer/mem_reuse/mem_swap_manager.cc
View File

@ -24,9 +24,8 @@ namespace device {
namespace memswap {
bool MemSwapManager::Init(const mindspore::session::KernelGraph *kernel_graph, size_t swap_mem_size) {
MS_EXCEPTION_IF_NULL(kernel_graph);
graph_manager_ = kernel_graph->manager();
MS_EXCEPTION_IF_NULL(graph_manager_);
execution_order_ = kernel_graph->execution_order();
kernel_graph_ = kernel_graph;
size_t kernel_index = 0;
for (const auto &kernel : execution_order_) {
@ -177,7 +176,10 @@ bool MemSwapManager::IsCommunicationRelevantOp(const AnfNodePtr &kernel) const {
return true;
}
NodeUsersMap &user_map = graph_manager_->node_users();
MS_EXCEPTION_IF_NULL(kernel_graph_);
const auto &graph_manager = kernel_graph_->manager();
MS_EXCEPTION_IF_NULL(graph_manager);
NodeUsersMap &user_map = graph_manager->node_users();
auto iter = user_map.find(kernel);
bool adjacent_with_communication_op = false;
if (iter != user_map.end()) {
@ -190,7 +192,10 @@ bool MemSwapManager::IsCommunicationRelevantOp(const AnfNodePtr &kernel) const {
}
void MemSwapManager::SaveUserKernelTopoOrder() {
NodeUsersMap &user_map = graph_manager_->node_users();
MS_EXCEPTION_IF_NULL(kernel_graph_);
const auto &graph_manager = kernel_graph_->manager();
MS_EXCEPTION_IF_NULL(graph_manager);
NodeUsersMap &user_map = graph_manager->node_users();
for (const auto &kernel : execution_order_) {
auto iter = user_map.find(kernel);
if (iter == user_map.end()) {

2
mindspore/ccsrc/backend/optimizer/mem_reuse/mem_swap_manager.h
View File

@ -156,7 +156,7 @@ class MemSwapManager {
size_t distance_decay_step_;
MemCopyManagerPtr mem_copy_manager_{nullptr};
FuncGraphManagerPtr graph_manager_{nullptr};
const mindspore::session::KernelGraph *kernel_graph_{nullptr};
bool mem_swap_initialized_{false};
bool swap_info_already_set_{false};
bool trigger_swap_{false};

9
mindspore/ccsrc/frontend/optimizer/irpass/merge_addn.h
View File

@ -37,9 +37,11 @@ class MergeAddN : public AnfVisitor {
public:
AnfNodePtr operator()(const OptimizerPtr &optimizer, const AnfNodePtr &node) override {
Reset();
optimizer_ = optimizer;
mng_ = optimizer->resource()->manager();
is_outer_ = true;
AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(node);
// do not hold this manager
mng_ = nullptr;
if (!is_match_ || node->func_graph() == nullptr) {
return nullptr;
}
@ -104,8 +106,7 @@ class MergeAddN : public AnfVisitor {
}
bool is_unique(const AnfNodePtr &node) {
auto mng = optimizer_->resource()->manager();
auto &node_users = mng->node_users();
auto &node_users = mng_->node_users();
if (node_users.find(node) == node_users.end()) {
return false;
}
@ -124,7 +125,7 @@ class MergeAddN : public AnfVisitor {
}
private:
OptimizerPtr optimizer_{nullptr};
FuncGraphManagerPtr mng_{nullptr};
std::vector<AnfNodePtr> Xs_{}, Ys_{}, args_{};
bool is_inner_{false}, is_outer_{false}, is_match_{false};
};