Browse Source

Do not generate file 'ms_output_*.pb'

tags/v0.6.0-beta
fary86 5 years ago
parent
commit
a55a7f3bcf
2 changed files with 0 additions and 31 deletions
  1. +0
    -30
      mindspore/ccsrc/pipeline/pipeline.cc
  2. +0
    -1
      mindspore/ccsrc/pipeline/pipeline.h

+ 0
- 30
mindspore/ccsrc/pipeline/pipeline.cc View File

@@ -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);



+ 0
- 1
mindspore/ccsrc/pipeline/pipeline.h View File

@@ -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);



Loading…
Cancel
Save