From c3c31f90a4960bfd8bc7fe370a61ca30dd180213 Mon Sep 17 00:00:00 2001
From: hang yang <hankyang1992@gmail.com>
Date: Mon, 12 Apr 2021 17:18:20 +0200
Subject: [PATCH] dump tensor relation log

---
 .../optimizer/somas/somas_solver_pre.cc       | 37 +++++++++++++++++++
 .../optimizer/somas/somas_solver_pre.h        |  5 ++-
 2 files changed, 40 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.cc b/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.cc
index 3cdc5787b0..ef9201596f 100644
--- a/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.cc
+++ b/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.cc
@@ -199,6 +199,43 @@ void SomasSolverPre::Log(const session::KernelGraph *graph, const TensorsDescMap
                          const std::vector<DynamicBitSet> *pConstraints, const vector<vector<size_t>> &continuous_v) {
   SolverInputLog(graph, tensors, pConstraints, continuous_v);
   SolverOutputLog(graph, tensors);
+  TensorRelationLog(pConstraints, graph);
+}
+
+void SomasSolverPre::TensorRelationLog(const std::vector<DynamicBitSet> *pConstraints,
+                                       const session::KernelGraph *graph) {
+  MS_LOG(INFO) << "SomasSolver::Log Writing somas_tensor_relation.ir..";
+  auto context_ptr = MsContext::GetInstance();
+  MS_EXCEPTION_IF_NULL(context_ptr);
+  auto save_graphs_path = context_ptr->get_param<std::string>(MS_CTX_SAVE_GRAPHS_PATH);
+  std::string filename = save_graphs_path + "/" + "somas_tensor_relation_" + std::to_string(graph->graph_id()) + ".ir";
+  if (filename.size() > PATH_MAX) {
+    MS_LOG(ERROR) << "File path " << filename << " is too long.";
+    return;
+  }
+  auto real_path = Common::GetRealPath(filename);
+  if (!real_path.has_value()) {
+    MS_LOG(ERROR) << "Get real path failed. path=" << filename;
+    return;
+  }
+
+  ChangeFileMode(real_path.value(), S_IRWXU);
+  std::ofstream ofs(real_path.value());
+
+  if (!ofs.is_open()) {
+    MS_LOG(ERROR) << "Open log file '" << real_path.value() << "' failed!";
+    return;
+  }
+  for (size_t tid1 = 0; tid1 < pConstraints->size(); tid1++) {
+    ofs << 't' << tid1 << ' ';
+    for (size_t tid2 = 0; tid2 < (*pConstraints)[tid1].bit_size_; tid2++) {
+      ofs << 'H' << std::hex << (*pConstraints)[tid1].bit_[tid2];
+    }
+    ofs << std::endl << std::dec;
+  }
+  ofs.close();
+
+  MS_LOG(INFO) << "SomasSolver somas_tensor_relation Log done";
 }
 
 void SomasSolverPre::SolverInputLog(const session::KernelGraph *graph, const TensorsDescMap &tensors,
diff --git a/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.h b/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.h
index a3cfbc013f..f418ead759 100644
--- a/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.h
+++ b/mindspore/ccsrc/backend/optimizer/somas/somas_solver_pre.h
@@ -68,8 +68,6 @@ enum FittingType {
 
 class DynamicBitSet {
   const size_t bit_width_ = 64;
-  size_t bit_size_;
-  std::vector<uint64_t> bit_;
 
   inline size_t GetIndex(size_t index) const { return index / bit_width_; }
 
@@ -85,6 +83,8 @@ class DynamicBitSet {
   }
 
  public:
+  size_t bit_size_;
+  std::vector<uint64_t> bit_;
   explicit DynamicBitSet(size_t count) {
     bit_size_ = (count + bit_width_ - 1) / bit_width_;
     Reset(0x0);
@@ -207,6 +207,7 @@ class SomasSolverPre {
                       const std::vector<DynamicBitSet> *pConstraints_v, const vector<vector<size_t>> &continuous_v);
   void SolverOutputLog(const session::KernelGraph *graph, const TensorsDescMap &tensors) const;
   vector<TensorsDescMap> createTensorsMaps(const TensorsDescMap &tensors, size_t total_sol);
+  void TensorRelationLog(const std::vector<DynamicBitSet> *pConstraints, const session::KernelGraph *graph);
 };
 using SomasSolverPrePtr = std::shared_ptr<SomasSolverPre>;
 }  // namespace somas