diff --git a/mindspore/ccsrc/pipeline/pipeline.cc b/mindspore/ccsrc/pipeline/pipeline.cc index 6abe198f5a..c9d79a3a39 100644 --- a/mindspore/ccsrc/pipeline/pipeline.cc +++ b/mindspore/ccsrc/pipeline/pipeline.cc @@ -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); diff --git a/mindspore/ccsrc/pipeline/pipeline.h b/mindspore/ccsrc/pipeline/pipeline.h index 3f1274c417..58456c4d3b 100644 --- a/mindspore/ccsrc/pipeline/pipeline.h +++ b/mindspore/ccsrc/pipeline/pipeline.h @@ -72,7 +72,6 @@ class ExecutorPy : public std::enable_shared_from_this { ~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);