Skip to content

Conversation

@Pratheesh-04-MCW
Copy link
Contributor

Added roll operator support in the TVM Relax exported program and fx graph and test script code to validate the relax module. By adding this support, the following models are now able to run successfully:

1. facebook/mask2former-swin-tiny-coco-instance
2. briaai/RMBG-2.0

@Pratheesh-04-MCW Pratheesh-04-MCW marked this pull request as ready for review April 16, 2025 14:28
@Pratheesh-04-MCW
Copy link
Contributor Author

@Hzfengsy Could you please review this PR?

@Hzfengsy
Copy link
Member

cc @tlopex

@tlopex
Copy link
Member

tlopex commented Apr 18, 2025

LGTM! Thanks! @Pratheesh-04-MCW
Please address the conflict so that we can merge it

@Pratheesh-04-MCW
Copy link
Contributor Author

@Hzfengsy @tlopex Hope the conflicts are resolved. Thanks :)

@Hzfengsy Hzfengsy merged commit 36f2502 into apache:main Apr 18, 2025
10 checks passed
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
* add op support for roll op

* lint fix

* fixed unity check

* add unit test in fx_graph

* lint issues

* lint check

* confilct resolved
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.

3 participants