Browse Source

!6215 fix the problem which arg print in DumpIR

Merge pull request !6215 from Margaret_wangrui/fix_DumpIR_arg
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ffaba08379
2 changed files with 14 additions and 4 deletions
  1. +14
    -2
      mindspore/ccsrc/debug/anf_ir_dump.cc
  2. +0
    -2
      mindspore/ccsrc/pipeline/pynative/pynative_execute.h

+ 14
- 2
mindspore/ccsrc/debug/anf_ir_dump.cc View File

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


+ 0
- 2
mindspore/ccsrc/pipeline/pynative/pynative_execute.h View File

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



Loading…
Cancel
Save