m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

move somas model info into somas_meta directory

Browse Source
pull/12047/head
laiyongqiang 4 years ago
parent a7ff89dc50
commit b625073d4f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File

6
mindspore/ccsrc/backend/optimizer/somas/somas.cc
Unescape View File

@ -96,7 +96,7 @@ bool Somas::Allocate(const session::KernelGraph *graph) {
ret = CalcSomasModelHash(graph); ret = CalcSomasModelHash(graph);
if (ret) { if (ret) {
std::string filename = std::string filename =
save_graphs_path_ + "/" + "somas_graph" + std::to_string(graph->graph_id()) + "_" + hash_id_ + ".json"; save_graphs_path_ + "/somas_meta/" + "somas_graph" + std::to_string(graph->graph_id()) + "_" + hash_id_ + ".json";
ret = LoadSomasResult(graph, filename); ret = LoadSomasResult(graph, filename);
if (ret) { if (ret) {
MS_LOG(INFO) << "Load Somas Cache file " << filename << " Successfully."; MS_LOG(INFO) << "Load Somas Cache file " << filename << " Successfully.";
@ -130,7 +130,7 @@ bool Somas::CalcSomasModelHash(const session::KernelGraph *graph) {
hash_id_ = std::to_string(std::hash<std::string>()(model_str)); hash_id_ = std::to_string(std::hash<std::string>()(model_str));
MS_LOG(INFO) << "Graph " << graph->graph_id() << "'s SOMAS Model hash id is " << hash_id_; MS_LOG(INFO) << "Graph " << graph->graph_id() << "'s SOMAS Model hash id is " << hash_id_;
std::string filename = std::string filename =
save_graphs_path_ + "/" + "somas_graph" + std::to_string(graph->graph_id()) + "_" + hash_id_ + ".info"; save_graphs_path_ + "/somas_meta/" + "somas_graph" + std::to_string(graph->graph_id()) + "_" + hash_id_ + ".info";
if (filename.size() > PATH_MAX) { if (filename.size() > PATH_MAX) {
MS_LOG(WARNING) << "File path " << filename << " is too long."; MS_LOG(WARNING) << "File path " << filename << " is too long.";
return false; return false;
@ -185,7 +185,7 @@ bool Somas::SaveSomasResult(const session::KernelGraph *graph) {
somas_json[kTensors] = tensors_json; somas_json[kTensors] = tensors_json;
std::string filename = std::string filename =
save_graphs_path_ + "/" + "somas_graph" + std::to_string(graph->graph_id()) + "_" + hash_id_ + ".json"; save_graphs_path_ + "/somas_meta/" + "somas_graph" + std::to_string(graph->graph_id()) + "_" + hash_id_ + ".json";
if (filename.size() > PATH_MAX) { if (filename.size() > PATH_MAX) {
MS_LOG(WARNING) << "File path " << filename << " is too long."; MS_LOG(WARNING) << "File path " << filename << " is too long.";
return false; return false;

Write Preview
Loading…
Cancel
Save