forked from mindspore-Ecosystem/mindspore
!27117 [ME][Fallback] Handle interpret nodes in tuple.
Merge pull request !27117 from Margaret_wangrui/tuple_interpret
This commit is contained in:
commit
eb25c473d0
|
@ -1016,6 +1016,7 @@ AnfNodePtr Parser::ParseTuple(const FunctionBlockPtr &block, const py::object &n
|
|||
tuple_vec.emplace_back(make_tuple_op);
|
||||
for (size_t i = 0; i < elts.size(); i++) {
|
||||
AnfNodePtr node_ptr = ParseExprNode(block, elts[i]);
|
||||
node_ptr = HandleInterpret(block, node_ptr, elts[i]);
|
||||
tuple_vec.emplace_back(node_ptr);
|
||||
}
|
||||
CNodePtr tuple_app = block->func_graph()->NewCNodeInOrder(std::move(tuple_vec));
|
||||
|
|
|
@ -75,7 +75,6 @@ def use_tuple_of_tensor():
|
|||
return me_x
|
||||
|
||||
|
||||
@pytest.mark.skip(reason='Not support graph fallback feature yet')
|
||||
def test_tuple_of_tensor():
|
||||
"""
|
||||
Feature: JIT Fallback
|
||||
|
|
Loading…
Reference in New Issue