Lunderberg opened a new pull request, #16642:
URL: https://github.com/apache/tvm/pull/16642
Prior to this commit, the condition used for `relax::If` node and the
`"relax.assert_op"` operator was required to be a scalar tensor. This made it
difficult to alter behavior based on a runtime shape parameter. For example,
delegating to a vectorized implementation based on a whether a tensor shape is
divisible by the vector size.
This commit adds support for expressions of type `R.Prim('bool')` as the
conditional for `relax::If` and `"relax.assert_op"`, to allow these use cases.
--
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]
