iree-org / iree-torch

Torch Frontend for IREE
Apache License 2.0
25 stars 11 forks source link

Change interface of `_unwrap_single_tuple_return` to return bool #17

Closed ramiro050 closed 2 years ago

ramiro050 commented 2 years ago

The helper function _unwrap_single_tuple_return modifies the input graph in-place, so it does not makes sense to have the function return the final graph. This commit also makes the unit-tests a bit more comprehensive.