-
Notifications
You must be signed in to change notification settings - Fork 146
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable index_sort
#306
Enable index_sort
#306
Conversation
e3e5b99
to
1270d99
Compare
@rusty1s @mszarma @JakubPietrakIntel Please take a look. |
Enable `index_sort` introduced in [pyg-lib](https://pyg-lib.readthedocs.io/en/latest/modules/ops.html#pyg_lib.ops.index_sort). Till now, sorting indices (1d tensor) have been working sequentially. This PR enables the `index_sort` operation, which makes such a sort work in parallel (parallelized radix sort). Related PR: [pytorch_sparse:306](rusty1s/pytorch_sparse#306) In training benchmarks, data set load time was improved on average by 3.82 times, and training time by 1.06 times. --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: rusty1s <matthias.fey@tu-dortmund.de>
@rusty1s One PR left to go 😉 |
Indeed, will merge tomorrow :D |
Codecov Report
📣 This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more @@ Coverage Diff @@
## master #306 +/- ##
==========================================
- Coverage 72.54% 72.50% -0.05%
==========================================
Files 28 29 +1
Lines 1122 1131 +9
==========================================
+ Hits 814 820 +6
- Misses 308 311 +3
📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
Enable
index_sort
introduced in pyg-lib.Related PR: pytorch_geometric:6554