vinx13 opened a new pull request, #12913: URL: https://github.com/apache/tvm/pull/12913
The current Einsum doesn't utilize `ReduceOp` for reduction. It instead unrolls the whole reduction when defining the computation. It makes the generated result not suitable for analysis as it doesn't contain reduction information. It's also broken for large reductions. This PR provided a new implementation that generates the reduction axes correctly. Graph level optimizations might be needed for efficient Einsum. This PR only provides compute definition for further analysis and optimizations. cc @spectrometerHBH @MasterJH5574 @junrushao @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]
