Skip to content

Conversation

@Pratheesh-04-MCW
Copy link
Contributor

@Pratheesh-04-MCW Pratheesh-04-MCW commented Apr 25, 2025

This PR adds support for the following operations in the exported program frontend for TVM Relax: zeros_like and fill_. By adding these ops the following models run successfully.

Operation Supported Models
zeros_like - mdeberta-v3-base
- ms-marco-MiniLM-L6-v2
- grammar_error_correcter_v1
- google/byt5-base
- google/flan-t5-small
- DeBERTa-v3-base-mnli-fever-anli
- Political_DEBATE_large_v1.0
- microsoft/table-transformer-detection
- facebook/detr-resnet-50
- facebook/mask2former-swin-tiny-coco-instance
fill_ - microsoft/beit-base-patch16-224-pt22k-ft22k
- google-mt5/small

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

@Hzfengsy @tlopex could you please review this PR?

@Hzfengsy Hzfengsy merged commit ed477b0 into apache:main Apr 28, 2025
12 of 13 checks passed
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
* add op support for zeros_like and fill_

* fixing whitespace issues

* unity issue

* solved datatype issue

* unity issue

* lint error
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