Add GPU COG overview support (#1150)#1174
Merged
brendancol merged 5 commits intomasterfrom Apr 8, 2026
Merged
Conversation
write_geotiff_gpu now accepts cog, overview_levels, and overview_resampling. Overviews are block-reduced and compressed on GPU, then assembled into COG layout. to_geotiff() passes the new params through to the GPU path.
Tests cover resampling methods (mean, nearest, mode, min, max, median), multi-level overviews, auto-generation, round-trip value preservation, public API dispatch, and GPU/CPU parity for block-reduce.
New docs/source/reference/geotiff.rst covers the public I/O functions. README feature matrix now lists write_geotiff_gpu and shows COG overview usage examples.
Covers automatic and explicit overview levels, resampling methods, and round-trip verification. Self-contained with synthetic data.
GDAL and rasterio use TIFF tag 254 (NewSubfileType = 1) to identify overview sub-IFDs. Without it, overviews are written correctly but invisible to GDAL-based tools. Verified: rasterio and gdalinfo now see both overview levels, and pixel values match GDAL's own COG output to within float32 rounding (max relative error < 0.0001%).
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
write_geotiff_gpu()now acceptscog,overview_levels, andoverview_resampling. Overviews are block-reduced and compressed on GPU, then written in COG layout (IFDs first, pixel data after).to_geotiff()passes overview params through to the GPU path, so CuPy arrays get COG output without CPU fallback.Test plan
pytest xrspatial/geotiff/tests/test_cog.py-- 25 tests (9 existing + 16 new), all passCloses #1150