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

!10901 Add compile profiling for PyNative mode. 

From: @zh_qh
Reviewed-by: @ginfung,@zhunaipan
Signed-off-by: @zhunaipan
This commit is contained in:
mindspore-ci-bot 2021-01-04 15:35:30 +08:00 committed by Gitee
parent 4ceb9e633c f9b89be832
commit 61869a6602
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/ccsrc/pipeline/jit/action.cc
View File

@ -315,7 +315,14 @@ bool GeOptimizeAction(const ResourcePtr &res) { return OptimizeAction(res, kGePa
bool VmOptimizeAction(const ResourcePtr &res) { return OptimizeAction(res, kVmPasses); }
bool PynativeOptimizeAction(const ResourcePtr &res) { return OptimizeAction(res, kPynativePasses); }
bool PynativeOptimizeAction(const ResourcePtr &resource) {
WITH(MsProfile::GetProfile())[&resource]() { (void)OptimizeAction(resource, kPynativePasses); };
#ifdef ENABLE_PROFILE
MsProfile::Print();
MsProfile::Reset();
#endif
return true;
}
bool PynativeElimOpt(const ResourcePtr &res) {
if (res->manager() == nullptr) {