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

fix core_dump bug in master and update akg submodule

This commit is contained in:
zengzitao 2022-02-09 16:48:18 +08:00
parent a377854d53
commit 2cd4365b70
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
akg

@ -1 +1 @@
Subproject commit 288658ca72e8de65b17c3109170d1ef0d511eb28
Subproject commit 540a6c35cad485e0b6f6ff46e1ba8166752a8ec7

8
mindspore/ccsrc/backend/optimizer/graph_kernel/adapter/graph_kernel_optimization.cc
View File

@ -52,6 +52,7 @@
#include "backend/optimizer/graph_kernel/graph_kernel_recompute.h"
#include "backend/optimizer/graph_kernel/reduce_fake_out_mem.h"
#include "backend/optimizer/graph_kernel/depend_elimination.h"
#include "backend/optimizer/graph_kernel/core/graph_kernel_utils.h"
namespace mindspore::graphkernel {
using opt::CommonSubexpressionElimination;
@ -235,11 +236,8 @@ void GraphKernelOptimizer::Run(const KernelGraphPtr &kernel_graph) {
optimizer->AddPassManager(Combine());
optimizer->AddPassManager(PostProcess());
auto mng = kernel_graph->manager();
if (mng == nullptr) {
mng = Manage(kernel_graph, true);
kernel_graph->set_manager(mng);
}
auto mng = GkUtils::GetFuncGraphManager(kernel_graph);
GkUtils::UpdateFuncGraphManager(mng, kernel_graph);
(void)optimizer->Optimize(kernel_graph);
}