weiji14
cog3pio
BlogDocsChangelog

:boom: Read CudaSlice into DLPack tensor

#57
Comparing
cu-dlpack
(
c2f551f
) with
main
(
9db1590
)
CodSpeed Performance Gauge
0%
Untouched
5

Benchmarks

Passed

test_read_geotiff_local
python/tests/test_io_geotiff.py
CodSpeed Performance Gauge
+1%
236.7 µs234.8 µs
test_read_geotiff_remote
python/tests/test_io_geotiff.py
CodSpeed Performance Gauge
0%
8.5 ms8.5 ms
test_xarray_backend_open_dataarray[cog3pio]
python/tests/test_xarray_backend.py
CodSpeed Performance Gauge
0%
45.7 ms45.7 ms
test_xarray_backend_open_dataarray[rasterio]
python/tests/test_xarray_backend.py
CodSpeed Performance Gauge
0%
14.9 ms14.9 ms
test_read_geotiff_multi_band
python/tests/test_io_geotiff.py
CodSpeed Performance Gauge
0%
24.1 ms24.2 ms

Commits

Click on a commit to change the comparison range
Base
main
9db1590
+0.35%
:triangular_flag_on_post: Remove dlpark/cuda from default features Can't enable dlpark's 'cuda' feature by default, need to hide it behind cog3pio's 'cuda' feature flag.
28efe58
6 days ago
by weiji14
-0.07%
:sparkles: Transmute CudaSlice to proper dtype before entering DLPack Handle conversion from u8 to other dtypes before putting data into DLPack. Need to determine dtype from sample_format and bits_per_pixel. Transmute turns CudaSlice into CudaView, so need https://github.com/SunDoge/dlpark/pull/29 for dlpark to support conversion to DLPack from CudaView. Also had to bump up opt-level from 1 (basic optimizations) to 2 (some optimizations), because `nvtiffDecodeCheckSupported` is failing with `NVTIFF_STATUS_TIFF_NOT_SUPPORTED (4)` on test/dev builds again. Note to self that transmute is likely happening on native endian? Need to check that little/big endian TIFFs are properly decoded later.
5a89f77
3 hours ago
by weiji14
-0.14%
:white_check_mark: Add tests for reading u8/u16/u32/i16/i32 tiffs Bunch of parametrized tests to ensure that reading various uint/int dtype TIFFs work. The u64/i64 TIFFs from GDAL's test suite are not supported by nvTIFF yet for some reason though.
c2f551f
3 hours ago
by weiji14
© 2025 CodSpeed Technology
Home Terms Privacy Docs