psrivas2 opened a new pull request, #14146: URL: https://github.com/apache/tvm/pull/14146
Enhance constant folding to fold relax ops by evaluating them. This uses the registered legalization function attached to the op to lower it to `call_tir` and uses the existing `call_tir` folding mechanism to fold it. This kind of op folding is only allowed within dataflow block as ops could have side-effects. Limitations: * This currently does not support folding ops that could lower to multiple `call_tir` bindings. * Folding by evaluating ops is not always beneficial. We need a heuristic to check if it is useful. This is not implemented yet and folding is always allowed by evaluating expressions. -- 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]
