xarray-spatial
xarray-spatial copied to clipboard
[FEA] Cast xarray.DataArrays to cupy
Reason or Problem
Seems like there is a way to cast an xarray.Dataset to cupy, maybe not for xarray.DataArray
Proposal
I would like a way to cast an xarray.DataArray to cupy , as it does with xarray.Dataset
Usage: GPU direct read of DataArray objects
Stakeholders and Impacts
RAPIDS
Drawbacks
Unknown
Alternatives
data = cupy.asarray(xarray.open_rasterio('data.tif').data) works, but isn't as efficient as making it cupy in the first place
Unresolved Questions
shrugs
Additional Notes or Context
None....for now :)
@thuydotm can you take a look at this?