NVIDIA / numba-cuda

BSD 2-Clause "Simplified" License
36 stars 8 forks source link

[FEA] Allow libraries that implement __array_ufunc__ to override CUDAUFuncDispatcher #36

Closed gmarkall closed 3 months ago

gmarkall commented 3 months ago

Details / notes:

From Allow libraries that implement __array_ufunc__ to override CUDAUFuncDispatcher on the Numba Discourse.

There was a PR implementing this for the CPU target: https://github.com/numba/numba/pull/8995

A related issue on the Awkward issue tracker: https://github.com/scikit-hep/awkward/issues/3179

This is to support using Coffea on CUDA.

cc @ianna @lgray

Initial thoughts:

I am not able to dig deep into this right now, but there are a couple of additional potential hurdles for the CUDA target:

gmarkall commented 3 months ago

See #37, which is a better-written request.