p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!35456 fix vmap multi-nested tuple input bug 

Merge pull request !35456 from Erpim/vmap_v15
This commit is contained in:
i-robot 2022-06-07 13:00:01 +00:00 committed by Gitee
parent 63fbca95a4 9cddd36024
commit 28a42e7ab9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/frontend/optimizer/irpass/vmap_eliminate.cc
View File

@ -50,6 +50,7 @@ CNodePtr BuildBindInAxisTupleInput(const AnfNodePtr &input, const ValuePtr &in_a
auto in_axis_value = in_axis_value_sequence == nullptr ? in_axis : (*in_axis_value_sequence)[i];
CNodePtr cur_make_tuple = nullptr;
if (input_abs_element->isa<abstract::AbstractTuple>()) {
tuple_getitem_cnode->set_abstract(input_abs_element);
cur_make_tuple = BuildBindInAxisTupleInput(tuple_getitem_cnode, in_axis_value, fg);
} else {
std::vector<AnfNodePtr> cur_make_tuple_inputs;