yongwww opened a new pull request, #16254: URL: https://github.com/apache/tvm/pull/16254
The PR is to dispatch cumsum and sort to their platform dependent implementation in Relax. If no target is found, will default to go with GPU implementation. The dispatch pass was implemented in python, we might update the pattern a little bit when we introduce some specialized implementation, for example @vinx13's top2. cc: @tqchen @vinx13 @jinhongyii -- 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]
