Skip to content

[Unity] Dispatch cumsum and sort - #16254

Merged
jinhongyii merged 10 commits into
apache:unityfrom
yongwww:dispatch_scan_sort
Jan 3, 2024
Merged

jinhongyii merged 10 commits into
apache:unityfrom
yongwww:dispatch_scan_sort

Conversation

@yongwww

@yongwww yongwww commented Dec 16, 2023

Copy link
Copy Markdown
Member

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

Comment thread python/tvm/relax/backend/dispatch_ops.py Outdated
Comment thread include/tvm/relax/attrs/manipulate.h Outdated
Comment thread tests/python/relax/test_backend_dispatch_sort_scan.py
Comment thread python/tvm/relax/backend/dispatch_sort_scan.py
@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