Can we accelerate functions from other packages using juliaGPU package?

In Julia, the GPU or CPU location is encoded in the type of an array. For instance, a CuArray from CUDA.jl lives on an NVIDIA GPU. So if you have a function generic enough to accept arbitrary array types, then applying it to a CuArray will just work.

3 Likes