What is a good way to get the n largest elements of a CuArray?

Specific Domains GPU
1

I noticed that partialsort! is not yet defined forCuArray, I don’t need the n largest in any particular order, either.

Thank you!

2

Sorting is not implemented in CUDA.jl. See https://github.com/JuliaGPU/CUDA.jl/pull/431 for a WIP PR.

1 Like