Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[PyTorch] Guarantee data input is the first argument #7592

Merged
merged 3 commits into from
Mar 5, 2021

Conversation

comaniac
Copy link
Contributor

@comaniac comaniac commented Mar 4, 2021

Currently we use analyzer to get the free_vars as the converted Relay argument list, but this may cause the data input to be in a random place. For some use cases, we wish to make sure the data inputs can be in the beginning of the function arguments, so this PR fixes this.

cc @masahi

@comaniac comaniac merged commit d5cb3cb into apache:main Mar 5, 2021
@comaniac
Copy link
Contributor Author

comaniac commented Mar 5, 2021

Thanks @masahi

@comaniac comaniac deleted the pytorch_fix_order branch March 5, 2021 17:49
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants