Ubospica commented on PR #16052: URL: https://github.com/apache/tvm/pull/16052#issuecomment-1793951629
> I agree, in this case, moving to use batch_matmul might be easier What I need is a matmul where the shape of A is [b, m, k] and B is [n, k], and A or/and B could be transposed. But batch_matmul only supports 3-dim A and B. And I guess having a matmul operator supporting all input dimensions may be helpful. As described in [data apis](https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__matmul__.html), the additional dimensions will be broadcasted. -- 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]
