Skip to content

[Relax][PyTorch] Add Stack Op Support for Exported Program #17819

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.

Sign up for GitHub

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

Jump to bottom
Merged
Hzfengsy merged 13 commits into from
Apr 18, 2025
Merged

[Relax][PyTorch] Add Stack Op Support for Exported Program #17819

Hzfengsy merged 13 commits into from
Apr 18, 2025

Conversation

@Pratheesh-04-MCW
Copy link
Contributor

@Pratheesh-04-MCW Pratheesh-04-MCW commented Apr 9, 2025 •
edited
Loading

This PR refactors the Stack Operation to be implemented at the core level using decomposed operations. It allows Stack to be accessed directly via R.Stack, making it more efficient and easier to use in TVM models. By adding this support we can run the following models

1. microsoft/beit-base-patch16-224-pt22k-ft22k (Image Classification)
2. anton-l/wav2vec2-random-tiny-classifier (Audio Classification)
3. microsoft/table-transformer-detection (Object Detection)
4. facebook/detr-resnet-50 (Object Detection)
5. facebook/mask2former-swin-tiny-coco-instance (Image Segmentation)
6. Intel/dpt-beit-large-512 (Depth Estimation)

@Pratheesh-04-MCW Pratheesh-04-MCW marked this pull request as ready for review April 17, 2025 08:21
@Pratheesh-04-MCW Pratheesh-04-MCW marked this pull request as draft April 17, 2025 08:21
@Pratheesh-04-MCW Pratheesh-04-MCW marked this pull request as ready for review April 17, 2025 08:22
@Pratheesh-04-MCW
Copy link
Contributor Author

Pratheesh-04-MCW commented Apr 17, 2025

@Hzfengsy Could you please review this PR?

yongwww
yongwww reviewed Apr 17, 2025
View reviewed changes
@Hzfengsy Hzfengsy merged commit c00f52a into apache:main Apr 18, 2025
12 checks passed
@ysh329 ysh329 mentioned this pull request Jul 16, 2025
Closed
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
@Pratheesh-04-MCW @ShiboXing
[Relax][PyTorch] Add Stack Op Support for Exported Program (apache#17819
a99905f 
)

* add op support for stack

* trailing whitespace issue fixed

* fixed lint issues

* fixed whitespace issue

* fixed lint error

* fixing lint issues

* fixed whitespace issue

* add test script for fx_graph

* fix lint issues

* fixed unity check issues

* unity check

* fixed unity check issues

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

Reviewers

@yongwww yongwww yongwww left review comments

@Hzfengsy Hzfengsy Hzfengsy approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@Pratheesh-04-MCW @yongwww @Hzfengsy