pysheds icon indicating copy to clipboard operation
pysheds copied to clipboard

Pysheds Cupy / Cuspatial support

Open dan771 opened this issue 1 year ago • 1 comments

One way for huge optimizations would be to support natively Cupy or Cuspatial. I tried replacing the numpy import on the sys level with cupy however a few small inconsistencies arise. Although cupy requires Nvidia and cuda I feel the headline 10 times performance boost on most numpy operations feels highly worth it for the amount of work you'd need to facilitate native support.

dan771 avatar Feb 07 '24 22:02 dan771

Thanks @dan771. I am interested in supporting GPU processing, especially for the highly parallelizable functions like flow directions. Numba also supports GPU processing, and so I may leverage numba for this purpose in the future.

mdbartos avatar Feb 10 '24 03:02 mdbartos