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

[TVMScript] IR Fragment Printing #13742

Merged
merged 1 commit into from
Jan 14, 2023

Conversation

junrushao
Copy link
Member

@junrushao junrushao commented Jan 10, 2023

This PR introduces support for TIR fragment printing

@tvm-bot
Copy link
Collaborator

tvm-bot commented Jan 10, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

  • No users to tag found in teams: tvmscript See #10317 for details

Generated by tvm-bot

@junrushao junrushao changed the title [TVMScript] Printing IR Fragments [TVMScript] IR Fragments Printing Jan 10, 2023
@junrushao junrushao force-pushed the feature/2022-01-09/fragment branch 2 times, most recently from 07d7f3e to bcf2252 Compare January 10, 2023 06:53
@junrushao junrushao changed the title [TVMScript] IR Fragments Printing [TVMScript] IR Fragment Printing Jan 10, 2023
@junrushao junrushao marked this pull request as ready for review January 10, 2023 06:53
@junrushao junrushao force-pushed the feature/2022-01-09/fragment branch from bcf2252 to e788615 Compare January 10, 2023 07:14
@Hzfengsy
Copy link
Member

ir_prefix is important since T and I are only alias for tir and ir. Please restore the API Interface to specify the ir_prefix

@junrushao
Copy link
Member Author

junrushao commented Jan 10, 2023

Will expose a configuration API in python in a follow-up PR tomorrow

Update: added in this PR

@junrushao junrushao force-pushed the feature/2022-01-09/fragment branch 14 times, most recently from e4dc02e to 673984f Compare January 13, 2023 23:30
This PR introduces support for TIR fragment printing.
Fragment printing makes it possible to print TIR fragments in the text
format consistency with TVMScript PrimFunc/IRModule printing.

This PR still preserves the legacy ReprPrinter format by introducing an
API `LegacyTIRPrint` for TIR PrimExpr. This method is used in
AutoScheduler and TIR CSE for full backward compatibility.
@junrushao junrushao force-pushed the feature/2022-01-09/fragment branch from 673984f to bd08d72 Compare January 13, 2023 23:46
@tqchen tqchen merged commit c452e69 into apache:main Jan 14, 2023
@junrushao junrushao deleted the feature/2022-01-09/fragment branch January 16, 2023 03:40
fzi-peccia pushed a commit to fzi-peccia/tvm that referenced this pull request Mar 27, 2023
This PR introduces support for TIR fragment printing.
Fragment printing makes it possible to print TIR fragments in the text
format consistency with TVMScript PrimFunc/IRModule printing.

This PR still preserves the legacy ReprPrinter format by introducing an
API `LegacyTIRPrint` for TIR PrimExpr. This method is used in
AutoScheduler and TIR CSE for full backward compatibility.
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.

4 participants