Are there implementations of Thrust::sort, Thrust::reduce, Thrust::exclusive_scan?

[Wilson1211]

Are the source codes of these functions available?
After tracing through the code in the repository, I found that they eventually call functions like thrust::reduce. The same applies to other functions like sort and exclusive_scan.

However, I’m interested in understanding how these functions are actually implemented.
Even if the implementations are outdated, I’d still appreciate having access to them.

[miscco]

Thrust has different backends that have different implementations to accelerate the various algorithms.
For reduce there is

• thrust/thrust/system/cpp/detail/reduce.h
• thrust/thrust/system/cuda/detail/reduce.h
• thrust/thrust/system/omp/detail/reduce.h
• thrust/thrust/system/tbb/detail/reduce.h
• thrust/thrust/system/detail/generic/reduce.h
• thrust/thrust/system/detail/sequential/reduce.h

All our code is open source so you can read them up. The openmp and tbb backends could see some love but the cuda backend of cub is actively developed

[jrhemstad]

@Wilson1211 assuming you are interested in the CUDA implementations, what you'll find from looking at the Thrust headers is that the Thrust algorithms ultimately just invoke a corresponding CUB algorithm. So if you're interested in looking at the CUDA implementations, you'll ultimately just want to look at the CUB implementations https://nvidia.github.io/cccl/cub/device_wide.html#device-module

[Wilson1211]

@jrhemstad Yes, thank you! I'm also interested in the CUDA implementations! Really appreciate the extra details!