From 96daa72f202f14733b89b53e84ac358bb784c8b2 Mon Sep 17 00:00:00 2001
From: Margaret_wangrui <wangrui178@huawei.com>
Date: Fri, 11 Sep 2020 07:33:11 +0800
Subject: [PATCH] fix DumpIR arg

---
 mindspore/ccsrc/debug/anf_ir_dump.cc             | 16 ++++++++++++++--
 .../ccsrc/pipeline/pynative/pynative_execute.h   |  2 --
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/mindspore/ccsrc/debug/anf_ir_dump.cc b/mindspore/ccsrc/debug/anf_ir_dump.cc
index f45ff5d26e..f316a1c61f 100644
--- a/mindspore/ccsrc/debug/anf_ir_dump.cc
+++ b/mindspore/ccsrc/debug/anf_ir_dump.cc
@@ -241,7 +241,7 @@ void DumpOperands(const AnfNodePtr &nd, OrderedMap<AnfNodePtr, int32_t> *para_ma
       }
       if (in->isa<Parameter>()) {
         if (!(*para_map)[in]) {
-          gsub->buffer << "%arg";
+          gsub->buffer << "%para_" << in->ToString();
         } else {
           gsub->buffer << "%para" << (*para_map)[in];
         }
@@ -437,7 +437,19 @@ void DumpSubgraph(const OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>
     fout << "subgraph @" << sg.first->ToString() << ".";
     fout << sg.first->debug_info()->get_id() << "(";
     if (sg.first != graph) {
-      fout << "%arg";
+      std::vector<AnfNodePtr> parameters = sg.first->parameters();
+      if (parameters.size() == 1) {
+        MS_EXCEPTION_IF_NULL(parameters[0]);
+        fout << "%para_" << parameters[0]->ToString();
+      } else if (parameters.size() > 1) {
+        for (size_t idx = 0; idx < parameters.size() - 1; idx++) {
+          MS_EXCEPTION_IF_NULL(parameters[idx]);
+          fout << "%para_" << parameters[idx]->ToString();
+          fout << ", ";
+        }
+        MS_EXCEPTION_IF_NULL(parameters[parameters.size() - 1]);
+        fout << "%para_" << parameters[parameters.size() - 1]->ToString();
+      }
     }
     fout << ") {" << std::endl;
     MS_EXCEPTION_IF_NULL(sg.second);
diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.h b/mindspore/ccsrc/pipeline/pynative/pynative_execute.h
index 0e4276559a..d9b5daa6ce 100644
--- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.h
+++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.h
@@ -38,7 +38,6 @@
 
 namespace mindspore {
 namespace pynative {
-
 namespace py = pybind11;
 using ResourcePtr = std::shared_ptr<pipeline::Resource>;
 using GradOperationPtr = std::shared_ptr<prim::GradOperation>;
@@ -168,7 +167,6 @@ class PynativeExecutor : public std::enable_shared_from_this<PynativeExecutor> {
 };
 
 using PynativeExecutorPtr = std::shared_ptr<PynativeExecutor>;
-
 }  // namespace pynative
 }  // namespace mindspore