quic-sanirudh opened a new pull request, #14356: URL: https://github.com/apache/tvm/pull/14356
This PR adds an argument to the relay to relax translator to append the relay Op Attrs as function attributes to the generated TIR PrimFuncs This information is really useful in Relax as the absence of this prevents us from being able schedule efficiently for ops that are heavily sensitive to the attributes. For example, the `groups` attribute to `conv2d` op is needed to differentiate between regular conv2d and depthwise conv2d. -- 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]
