quic-sanirudh opened a new pull request, #17075:
URL: https://github.com/apache/tvm/pull/17075
Arguments of a fused TIR PrimFunc generated from a fused relax function do
not retain all the buffer attributes from their original PrimFuncs as the
buffers are created from the StructInfo of the Relax vars. This patch collects
a mapping of relax vars to its corresponding TIR buffers in a fused relax
function and uses that info to propagate its buffer attributes such as
`axis_separators` and `storage_scope`
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
