search

rapidsai / dask-cuda

Utilities for Dask and CUDA interactions
https://docs.rapids.ai/api/dask-cuda/stable/
Apache License 2.0
292 stars 93 forks source link

Fix get_device_memory_ids #1305

Closed wence- closed 10 months ago

wence- commented 10 months ago

A recent change to the way StringColumns are implemented in cudf threw up that we were never correctly determining the number of device buffers belonging to cudf columns if they had children (e.g. list and struct columns) or masks (any nullable column). Handle those cases and update the test.

pentschev commented 10 months ago

Thanks @wence- !

pentschev commented 10 months ago

/merge