Skip to content

Conversation

@yongwww
Copy link
Member

@yongwww yongwww commented Dec 16, 2023

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

@jinhongyii jinhongyii merged commit 0cf5f47 into apache:unity Jan 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants