#177 raised this a while back but got closed without a real plan. Worth revisiting more narrowly: could UXarray support cupy arrays for on-node GPU parallelism, similar to how dask is already supported for CPU task-parallelism? They're complementary, not either/or.
Cupy is NVIDIA-only, which might be a problem given DOE's multi-vendor GPU needs (see @rljacob's comment on #177) — worth weighing against something like numba.cuda or array-API-standard compatibility instead.
Scoping this as a prototype + discussion, not a full implementation.
#177 raised this a while back but got closed without a real plan. Worth revisiting more narrowly: could UXarray support
cupyarrays for on-node GPU parallelism, similar to how dask is already supported for CPU task-parallelism? They're complementary, not either/or.Cupy is NVIDIA-only, which might be a problem given DOE's multi-vendor GPU needs (see @rljacob's comment on #177) — worth weighing against something like
numba.cudaor array-API-standard compatibility instead.Scoping this as a prototype + discussion, not a full implementation.