Browse Source

Add compile profiling for PyNative mode.

tags/v1.2.0-rc1
Zhang Qinghua 5 years ago
parent
commit
f9b89be832
1 changed files with 8 additions and 1 deletions
  1. +8
    -1
      mindspore/ccsrc/pipeline/jit/action.cc

+ 8
- 1
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) {


Loading…
Cancel
Save