-
Notifications
You must be signed in to change notification settings - Fork 282
Drop deprecated CUB iterators #3831
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
Drop deprecated CUB iterators #3831
Conversation
| #include <cub/util_allocator.cuh> | ||
|
|
||
| #include <thrust/iterator/counting_iterator.h> | ||
| #include <thrust/iterator/transform_iterator.h> |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
That include was missing before, since the file used thrust::transform_iterator.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🔥
🟩 CI finished in 1h 03m: Pass: 100%/93 | Total: 18h 29m | Avg: 11m 55s | Max: 36m 30s | Hits: 95%/133473
|
| Project | |
|---|---|
| CCCL Infrastructure | |
| libcu++ | |
| +/- | CUB |
| Thrust | |
| CUDA Experimental | |
| +/- | python |
| CCCL C Parallel Library | |
| Catch2Helper |
Modifications in project or dependencies?
| Project | |
|---|---|
| CCCL Infrastructure | |
| libcu++ | |
| +/- | CUB |
| +/- | Thrust |
| CUDA Experimental | |
| +/- | python |
| +/- | CCCL C Parallel Library |
| +/- | Catch2Helper |
🏃 Runner counts (total jobs: 93)
| # | Runner |
|---|---|
| 66 | linux-amd64-cpu16 |
| 9 | windows-amd64-cpu16 |
| 6 | linux-amd64-gpu-rtxa6000-latest-1 |
| 4 | linux-arm64-cpu16 |
| 3 | linux-amd64-gpu-h100-latest-1 |
| 3 | linux-amd64-gpu-rtx4090-latest-1 |
| 2 | linux-amd64-gpu-rtx2080-latest-1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good. We probably want to help - or at least inform - our RAFT friends to switch to thrust iterators.
…sage in ATen (#153373) A major release of CCCL 3.0.0 will introduce some bc-breaking changes. Namely iterators like TransformInputIterator and ConstantInputIterator were moved from CUB to Thrust, some operators like Max and Sum were moved to LibCUDACXX. For the more info on changes please visit: https://nvidia.github.io/cccl/cccl/3.0_migration_guide.html This is a follow up to PR #147493. A description from the original PR: > Several cub iterators have been deprecated and removed in the latest CCCL (cub) development NVIDIA/cccl#3831. This PR replaced the usage of those cub iterators with thrust iterators. > > Some cub thread operators were also deprecated and removed in NVIDIA/cccl#3918. This PR replaced those operators with libcudacxx ops. > > This might also affect ROCM usability a bit. > > This patch is tested to work with CCCL commit at NVIDIA/cccl@82befb0 > > Tracking of CCCL/CUB deprecations in the most recent development NVIDIA/cccl#101 Pull Request resolved: #153373 Approved by: https://github.com/cyyever, https://github.com/atalman
No description provided.