提交 146b941e 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2883 Do not generate file ms_output_*.pb when compiling graph

Merge pull request !2883 from fary86/remove_ms_output_xxx_pb
......@@ -374,34 +374,6 @@ void ExecutorPy::SaveCompiledGraph(const std::string &phase_s) {
MS_LOG(INFO) << "End save compiled func graph!";
}
void ExecutorPy::SaveCompiledGraphToPb(const std::string &phase_s) {
#ifdef ENABLE_DUMP_IR
// save the graph to file in protobuf format
FuncGraphPtr func_graph = info_[phase_s]->resource->func_graph();
MS_EXCEPTION_IF_NULL(func_graph);
if (phase_s.empty()) {
MS_LOG(ERROR) << "`phase` is empty '" << phase_s << "'!";
return;
}
std::string name_prefix = phase_s.substr(0, phase_s.find("."));
std::string pb_filename = std::string("ms_output_") + name_prefix + ".pb";
std::string filename = GetFilePathName(pb_filename);
MS_LOG(INFO) << "Begin saving graph to file <<'" << filename << "' in protobuf formart.";
ChangeFileMode(filename, S_IRWXU);
std::ofstream ofs(filename);
if (!ofs.is_open()) {
MS_LOG(ERROR) << "Open file '" << filename << "' failed!";
return;
}
ofs << GetFuncGraphProtoString(func_graph);
ofs.close();
// set file mode to read only by user
ChangeFileMode(filename, S_IRUSR);
MS_LOG(INFO) << "End saving graph to file in protobuf format";
#endif
}
bool ExecutorPy::ChangeExportGeirUseVmFlag(bool use_vm, const std::string &phase_s) const {
std::string phase_prefix = GetPhasePrefix(phase_s);
......@@ -476,8 +448,6 @@ bool ExecutorPy::CompileInner(const py::object &obj, const py::tuple &args, cons
info_[phase_s] = executor_info;
pip->Run();
// save compile graph to file in protobuf format
SaveCompiledGraphToPb(phase_s);
// save the run graph func to MsPipeLine
SaveCompiledGraph(phase_s);
......
......@@ -72,7 +72,6 @@ class ExecutorPy : public std::enable_shared_from_this<ExecutorPy> {
~ExecutorPy();
void SaveCompiledGraph(const std::string &phase_s);
void SaveCompiledGraphToPb(const std::string &phase_s);
bool CompileInner(const py::object &obj, const py::tuple &args, const py::object &phase, bool use_vm);
bool Compile(const py::object &obj, const py::tuple &args, const py::object &phase, bool use_vm);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册