mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

dump step auto parallel begin graphs

This commit is contained in:
Xiaoda Zhang 2022-08-15 15:06:38 +08:00
parent b2ad845781
commit 0492895c87
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/frontend/parallel/ops_info/ops_utils.h
View File

@ -120,7 +120,7 @@ constexpr char OUTPUT_TYPE[] = "output_type";
constexpr char CROSS_BATCH[] = "cross_batch";
constexpr char STEP_PARALLEL_BEGIN[] = "step_parallel_begin";
constexpr char STEP_PARALLEL_END[] = "step_parallel_end";
constexpr char STEP_AUTO_PARALLEL_BEGIN[] = "step_auto_parallel_begin.dot";
constexpr char STEP_AUTO_PARALLEL_BEGIN[] = "step_auto_parallel_begin";
constexpr char REQUIRES_GRAD[] = "requires_grad";
constexpr char PARAM_NAME[] = "name";
constexpr char RESHAPEINFO[] = "ReshapeInfo";

2
mindspore/ccsrc/frontend/parallel/step_auto_parallel.cc
View File

@ -98,7 +98,7 @@ bool StepAutoParallel(const FuncGraphPtr &root, const opt::OptimizerPtr &) {
msTime.Start();
#ifdef ENABLE_DUMP_IR
if (MsContext::GetInstance()->get_param<bool>(MS_CTX_SAVE_GRAPHS_FLAG)) {
draw::Draw(STEP_AUTO_PARALLEL_BEGIN, root);
DumpGraph(root, std::string(STEP_AUTO_PARALLEL_BEGIN));
}
#endif
MS_LOG(INFO) << "Now entering step auto parallel";