Skip to content

Preserve output names for returned input aliases - #3018

Open
TANGBUDU wants to merge 1 commit into
microsoft:mainfrom
TANGBUDU:fix/return-alias-output-name-2714
Open

Preserve output names for returned input aliases#3018
TANGBUDU wants to merge 1 commit into
microsoft:mainfrom
TANGBUDU:fix/return-alias-output-name-2714

Conversation

@TANGBUDU

Copy link
Copy Markdown

Fixes #2714.

When a graph input is returned through a local alias, use the alias name for the inserted Identity output instead of the generic return_val name. Directly returning the graph input keeps the existing naming behavior.

Tests:

  • pytest onnxscript/_internal/converter_test.py -q: 54 passed, 1 skipped, 1 xfailed, 3 xpassed; 160 subtests passed
  • lintrunner -a
  • git diff --check

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Development

Successfully merging this pull request may close these issues.

Unintuitive behavior: Returned input aliases generate Identity nodes with generic names instead of preserving Python variable names

1 participant