mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!14729 insert u monad parameter before io monad parameter in auto_monad 

From: @Margaret_wangrui
Reviewed-by: @hwhewei,@zh_qh
Signed-off-by: @zh_qh
This commit is contained in:
mindspore-ci-bot 2021-04-08 09:32:52 +08:00 committed by Gitee
parent 14cb3d692b 3824c91283
commit 3fe401feb5
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mindspore/ccsrc/pipeline/jit/static_analysis/auto_monad.cc
View File

@ -41,8 +41,11 @@ using RefInputs = OrderedMap<AnfNodePtr, std::vector<size_t>>;
// Add or get a monad parameter.
AnfNodePtr AddMonadParameter(const FuncGraphPtr &func_graph, const std::string &name,
const abstract::AbstractBasePtr &abs) {
size_t params_size = func_graph->parameters().size();
size_t io_monad_location = params_size;
// Search for existed parameters, return it if found.
for (auto &node : func_graph->parameters()) {
for (size_t i = 0; i < params_size; i++) {
auto &node = func_graph->parameters()[i];
auto para = dyn_cast<Parameter>(node);
if (para == nullptr) {
continue;
@ -51,13 +54,23 @@ AnfNodePtr AddMonadParameter(const FuncGraphPtr &func_graph, const std::string &
if (para_abs && *para_abs == *abs) {
return para;
}
if (HasAbstractIOMonad(para)) {
io_monad_location = i;
}
}
// Create a new parameter if not existed.
auto para = std::make_shared<Parameter>(func_graph);
para->set_name(name);
para->debug_info()->set_name(name);
para->set_abstract(abs);
func_graph->add_parameter(para);
// If io monad parameter added before u monad parameter, should insert u monad before io monad in parameters
if (io_monad_location != params_size && abs->isa<abstract::AbstractUMonad>()) {
std::vector<AnfNodePtr> params = func_graph->parameters();
params.insert(params.begin() + io_monad_location, para);
func_graph->set_parameters(params);
} else {
func_graph->add_parameter(para);
}
return para;
}