-
-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
Working towards fully GPU-native decompression of (Geo)TIFFs into device memory (CUDA GPU) on Rust/FFI side, and zero-copy to CuPy arrays on Python side.
Steps:
- TIFF to CUDA device memory via
nvtiff- ✨ Decode GeoTIFF images to CUDA stream #27 - CUDA memory to DLPack in-memory format via
dlpark- 💥 Read into CUDA memory DLPack tensor #57 - DLPack to CuPy via raw pointers - ✨ Python bindings for CudaCogReader #58
Notes:
- Expose this through a
read_geotiff_dlpackAPI function? - Bump min numpy version to 2.2 to get MAINT: from_dlpack thread safety fixes (#28883) numpy/numpy#28889
- Consider pathway through cuTENSOR -> DLPack? CuPy uses cuTENSOR apparently, but unsure if ABI is stable.
- Potentially bring this to
cupy-xarrayin the future
References:
Metadata
Metadata
Assignees
Labels
featureNew feature or requestNew feature or request