[Numpy-discussion] Direct GPU support on NumPy

Jerome Kieffer jerome.kieffer at esrf.fr
Tue Jan 2 15:22:01 EST 2018


On Tue, 02 Jan 2018 15:37:16 +0000
Yasunori Endo <jo7ueb at gmail.com> wrote:

> If the reason is just about human resources,
> I'd like to try implementing GPU support on my NumPy fork.
> My goal is to create standard NumPy interface which supports
> both CUDA and OpenCL, and more devices if available.

I think this initiative already exists ... something which merges the
approach of cuda and opencl but I have no idea on the momentum behind
it.

> Are there other reason not to support GPU on NumPy?

yes. Matlab has such support and the performances gain are in the order
of 2x vs 10x when addressing the GPU directly. All the time is spent in
sending data back & forth. Numba is indeed a good candidate bu limited
to the PTX assembly (i.e. cuda, hence nvidia hardware) 

Cheers,

Jerome


More information about the NumPy-Discussion mailing list