search

mdbartos / pysheds

:earth_americas: Simple and fast watershed delineation in python.
GNU General Public License v3.0
702 stars 188 forks source link

Pysheds Cupy / Cuspatial support #244

Open dan771 opened 5 months ago

dan771 commented 5 months ago

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.

mdbartos commented 5 months ago

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.