Ubospica opened a new pull request, #14932: URL: https://github.com/apache/tvm/pull/14932
This PR updates the logic of the `_fit_shape` helper function to first use BlockBuilder to acquire the shape, instead of providing the shape. It will make it easier for expressions that are hard to provide shapes, such as matmul. The correctness of these op gradients are checked in `tests/python/relax/test_op_gradient_numeric.py`, and does not need additional checks. -- 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]
