!6215 fix the problem which arg print in DumpIR

Merge pull request !6215 from Margaret_wangrui/fix_DumpIR_arg
pull/6215/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit ffaba08379

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

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