Skip to content

Conversation

@hugolatendresse
Copy link
Contributor

@hugolatendresse hugolatendresse commented Mar 8, 2025

Torch's Upsample module can only accomodate the Size or Scale Factor argument but not both. Regarding that limitation, there was previously a bug regarding the parsing of the arguments (the wrong order of arguments was assume). This PR fixes that and adds a unit test

cc: @MasterJH5574

@hugolatendresse hugolatendresse changed the title Allow ingesting Upsample module from torch.export either using Size or Scale Factor argument [Relax] Allow ingesting Upsample module from torch.export either using Size or Scale Factor argument Mar 8, 2025
Copy link
Contributor

@MasterJH5574 MasterJH5574 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thank you @hugolatendresse!

@MasterJH5574 MasterJH5574 merged commit cf6e61b into apache:main Mar 10, 2025
10 checks passed
@hugolatendresse hugolatendresse deleted the ingest_exported_upsample branch March 10, 2025 18:57
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
…g Size or Scale Factor argument (apache#17721)

Torch's Upsample module can only accomodate the Size or
Scale Factor argument but not both. Regarding that limitation,
there was previously a bug regarding the parsing of the arguments
(the wrong order of arguments was assume).

This PR fixes that and adds a unit test.
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