Running on GPU using CuArray

KristofferC / NearestNeighbors.jl

High performance nearest neighbor data structures (KDTree and BallTree) and algorithms for Julia.

Other

425 stars 66 forks source link

Running on GPU using CuArray #104

Closed lukehutch closed 4 years ago

lukehutch commented 4 years ago

I'm new to Julia, and I'm trying it out because it seems to be the simplest way to program GPUs right now using CuArray. I need a fast kd-tree algorithm that runs on a GPU, among other things, and I found NearestNeighbors.jl. How much work would be involved in getting this code working out of CuArray instances rather than its current CPU-bound data structures?

KristofferC commented 4 years ago

My guess is quite a lot of work is needed but I have no experience with KD trees on GPUs so I might be wrong.

lukehutch commented 4 years ago

This post might be helpful:

https://nextjournal.com/sdanisch/julia-gpu-programming

KristofferC commented 4 years ago

Don't see a reason to keep this open since there isn't really any actionable here.