xarray-spatial icon indicating copy to clipboard operation
xarray-spatial copied to clipboard

Add dask+cupy (cuda) implementation for Aspect (and other convolutions)

Open relativityhd opened this issue 1 month ago • 1 comments

I tried implementing it by myself for the aspect, it seems to work just fine:

def _run_dask_cupy(data: da.Array) -> da.Array:
    data = data.astype(cupy.float32)
    _func = partial(_run_cupy)
    out = data.map_overlap(_func,
                           depth=(1, 1),
                           boundary=cupy.nan,
                           meta=cupy.array(()))
    return out

So is there any reason I am not aware of why this is disabled / not implemented?

relativityhd avatar Nov 21 '25 13:11 relativityhd

@relativityhd Great Question and point! Any chance you could submit a PR for that and I'll prioritize review and merging

brendancol avatar Dec 03 '25 13:12 brendancol