Skip to content

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

@relativityhd

Description

@relativityhd

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?

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions