rapidsai/cudf
[BUG] `cudf.read_text` throws an exception when reading a host buffer
Fermée
#13 734 ouverte le 22 juil. 2023
0 - BacklogPythonbugcuIOgood first issue
Métriques du dépôt
- Stars
- (6 000 étoiles)
- Métriques de merge PR
- (Merge moyen 17j 21h) (230 PRs mergées en 30 j)
Description
Describe the bug
cudf.read_text throws an exception when reading a host buffer, unlike the CSV, Parquet, ORC and JSON readers. Hopefully it is straightforward to enable cudf.read_text to read from host buffers.
Steps/Code to reproduce bug Here is a small code snippet that shows the exception:
from io import BytesIO
import cudf
df = cudf.DataFrame({'a':['aaaa','bbbb']})
buf = BytesIO()
df.to_csv(buf, index=False)
df2 = cudf.read_csv(buf) # this is ok
print(df2)
buf = BytesIO()
df.to_csv(buf, index=False)
df2 = cudf.read_text(buf, delimiter='\n') # this crashes
print(df2)
Expected behavior
I expect that read_text would support host buffers correctly in the python layer. "multibyte_split" works fine with HOST_BUFFER data source in the libcudf benchmarks.
Environment overview (please complete the following information) nightly docker image from 23.08, A100 DGX workstation