Browse Source

!12047 move somas model info into somas_meta directory

From: @laiyongqiang
Reviewed-by: @stsuteng,@zhoufeng54
Signed-off-by: @stsuteng
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
8e4ec7c2ca
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      mindspore/ccsrc/backend/optimizer/somas/somas.cc

+ 3
- 3
mindspore/ccsrc/backend/optimizer/somas/somas.cc View File

@@ -96,7 +96,7 @@ bool Somas::Allocate(const session::KernelGraph *graph) {
ret = CalcSomasModelHash(graph);
if (ret) {
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);
if (ret) {
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));
MS_LOG(INFO) << "Graph " << graph->graph_id() << "'s SOMAS Model hash id is " << hash_id_;
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) {
MS_LOG(WARNING) << "File path " << filename << " is too long.";
return false;
@@ -185,7 +185,7 @@ bool Somas::SaveSomasResult(const session::KernelGraph *graph) {
somas_json[kTensors] = tensors_json;

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) {
MS_LOG(WARNING) << "File path " << filename << " is too long.";
return false;


Loading…
Cancel
Save