-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
[TIR] Add DeclBuffer IR node and functors #12300
Conversation
CC @cyx-6 please review for potential TVMScript syntactic/semantic change :-) |
LGTM. New parser is compatible with this change, and will be updated soon. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Overall LGTM. A quick question: after this change, the parser will treat T.decl_buffer
as an IR construct instead of an intrinsic that declares a buffer - is my understanding correct?
yes. |
* [TIR] Add DeclBuffer node * [TIR] Add IR functors for DeclBuffer * [TVMScript] Add printer and parser for DeclBuffer * Update printer * Update printer * Add test case * lint * fix
`DeclBufferNode` was added recently in apache#12300, but support to traverse it in codegen_llvm was not added. Since this just stmt just declares a new buffer for a body of TIR, we just need to traverse the body to properly generate code for LLVM.
`DeclBufferNode` was added recently in #12300, but support to traverse it in codegen_llvm was not added. Since this just stmt just declares a new buffer for a body of TIR, we just need to traverse the body to properly generate code for LLVM.
`DeclBufferNode` was added recently in apache#12300, but support to traverse it in codegen_llvm was not added. Since this just stmt just declares a new buffer for a body of TIR, we just need to traverse the body to properly generate code for LLVM.
This PR is part of #11627. It added the IR node
DeclBufferNode
, IR functors and TVMScript printer / parser.cc @tqchen @Lunderberg @Hzfengsy @junrushao1994 @wrongtest-intellif