yelite opened a new pull request, #14166: URL: https://github.com/apache/tvm/pull/14166
This PR adds the support of batch matmul to Relax CUTLASS BYOC. This includes batch matmul with shape 2D x ND, ND x 2D and ND x ND. Additionally, this supports bias, transposed y and all supported epilogues from regular matmul. This PR also includes: 1. Add check function to all APIs related to PatternRegistryEntry (a follow up of #14109) 2. Add check function to CUTLASS pattern table, blocking unsupported matmul from being partitioned. 3. Reorganize the tests related to CUTLASS matmul. cc @vinx13 @masahi -- 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]
