Cudf: [BUG] Column selection fails in read_parquet for df with nulls

Created on 20 Aug 2020  路  12Comments  路  Source: rapidsai/cudf

Describe the bug
I am getting a Invalid null mask for non-zero null count. error when trying to round trip a simple DataFrame to/from parquet. The problem occurs when both of the following are true:

  • The DataFrame has an "object" column with 1+ null values.
  • The read_parquet call is specifying a specific set of columns

Steps/Code to reproduce bug

import cudf

path = "test.parquet"
df = cudf.datasets.timeseries()
df["name"] = df["name"].astype("object")
df["name"].iloc[1] = None
df.to_parquet(path, index=False)

cudf.read_parquet(path)  # Works fine
cudf.read_parquet(path, columns=["id", "name"])  # Works fine
cudf.read_parquet(path, columns=["name", "id"])

Trace

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-31-6f0182eeef31> in <module>
----> 1 cudf.read_parquet(path, columns=["name", "id"])

~/workspace/rapids_dev/cudf/python/cudf/cudf/io/parquet.py in read_parquet(filepath_or_buffer, engine, columns, row_groups, skip_rows, num_rows, strings_to_categorical, use_pandas_metadata, *args, **kwargs)
    212             num_rows=num_rows,
    213             strings_to_categorical=strings_to_categorical,
--> 214             use_pandas_metadata=use_pandas_metadata,
    215         )
    216     else:

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/parquet.pyx in cudf._lib.parquet.read_parquet()

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/parquet.pyx in cudf._lib.parquet.read_parquet()

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/table.pyx in cudf._lib.table.Table.from_unique_ptr()

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/column.pyx in cudf._lib.column.Column.from_unique_ptr()

RuntimeError: cuDF failure at: /home/nfs/rzamora/workspace/rapids_dev/cudf/cpp/src/column/column_view.cpp:56: Invalid null mask for non-zero null count.

Note that I get the same error for cudf.read_parquet(path, columns=["x", "y"]) (so the object column doesn't need to be in the columns selection)

Expected behavior
There should be no RuntimeError error.

Environment overview (please complete the following information)

  • Environment location: Bare-metal
  • Method of cuDF install: conda

Environment details

Click here to see environment details

 **git***
 commit e51115fba4ffef293467ccb7acc94395e808ac62 (HEAD, upstream/branch-0.15)
 Merge: 7e73aa467 b8d78a8bf
 Author: Keith Kraus <[email protected]>
 Date:   Wed Aug 19 04:22:07 2020 -0400

 Merge pull request #5735 from sriramch/add_months

 [REVIEW] allow timestamps to be constructed only from duration
 **git submodules***

 ***OS Information***
 DGX_NAME="DGX Server"
 DGX_PRETTY_NAME="NVIDIA DGX Server"
 DGX_SWBUILD_DATE="2020-03-04"
 DGX_SWBUILD_VERSION="4.4.0"
 DGX_COMMIT_ID="ee09ebc"
 DGX_PLATFORM="DGX Server for DGX-1"
 DGX_SERIAL_NUMBER="QTFCOU8220024"
 DISTRIB_ID=Ubuntu
 DISTRIB_RELEASE=18.04
 DISTRIB_CODENAME=bionic
 DISTRIB_DESCRIPTION="Ubuntu 18.04.4 LTS"
 NAME="Ubuntu"
 VERSION="18.04.4 LTS (Bionic Beaver)"
 ID=ubuntu
 ID_LIKE=debian
 PRETTY_NAME="Ubuntu 18.04.4 LTS"
 VERSION_ID="18.04"
 HOME_URL="https://www.ubuntu.com/"
 SUPPORT_URL="https://help.ubuntu.com/"
 BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
 PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy"
 VERSION_CODENAME=bionic
 UBUNTU_CODENAME=bionic
 Linux dgx14 4.15.0-76-generic #86-Ubuntu SMP Fri Jan 17 17:24:28 UTC 2020 x86_64 x86_64 x86_64 GNU/Linux

 ***GPU Information***
 Wed Aug 19 17:04:45 2020
 +-----------------------------------------------------------------------------+
 | NVIDIA-SMI 440.64.00    Driver Version: 440.64.00    CUDA Version: 10.2     |
 |-------------------------------+----------------------+----------------------+
 | GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
 | Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
 |===============================+======================+======================|
 |   0  Tesla V100-SXM2...  On   | 00000000:06:00.0 Off |                    0 |
 | N/A   31C    P0    56W / 300W |   1451MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   1  Tesla V100-SXM2...  On   | 00000000:07:00.0 Off |                    0 |
 | N/A   32C    P0    41W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   2  Tesla V100-SXM2...  On   | 00000000:0A:00.0 Off |                    0 |
 | N/A   30C    P0    42W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   3  Tesla V100-SXM2...  On   | 00000000:0B:00.0 Off |                    0 |
 | N/A   29C    P0    41W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   4  Tesla V100-SXM2...  On   | 00000000:85:00.0 Off |                    0 |
 | N/A   30C    P0    43W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   5  Tesla V100-SXM2...  On   | 00000000:86:00.0 Off |                    0 |
 | N/A   30C    P0    41W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   6  Tesla V100-SXM2...  On   | 00000000:89:00.0 Off |                    0 |
 | N/A   32C    P0    43W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+
 |   7  Tesla V100-SXM2...  On   | 00000000:8A:00.0 Off |                    0 |
 | N/A   28C    P0    42W / 300W |     12MiB / 32510MiB |      0%      Default |
 +-------------------------------+----------------------+----------------------+

 +-----------------------------------------------------------------------------+
 | Processes:                                                       GPU Memory |
 |  GPU       PID   Type   Process name                             Usage      |
 |=============================================================================|
 |    0     79453      C   ...mora/miniconda3/envs/nvt_dev/bin/python  1439MiB |
 +-----------------------------------------------------------------------------+

 ***CPU***
 Architecture:        x86_64
 CPU op-mode(s):      32-bit, 64-bit
 Byte Order:          Little Endian
 CPU(s):              80
 On-line CPU(s) list: 0-79
 Thread(s) per core:  2
 Core(s) per socket:  20
 Socket(s):           2
 NUMA node(s):        2
 Vendor ID:           GenuineIntel
 CPU family:          6
 Model:               79
 Model name:          Intel(R) Xeon(R) CPU E5-2698 v4 @ 2.20GHz
 Stepping:            1
 CPU MHz:             3340.227
 CPU max MHz:         3600.0000
 CPU min MHz:         1200.0000
 BogoMIPS:            4390.04
 Virtualization:      VT-x
 L1d cache:           32K
 L1i cache:           32K
 L2 cache:            256K
 L3 cache:            51200K
 NUMA node0 CPU(s):   0-19,40-59
 NUMA node1 CPU(s):   20-39,60-79
 Flags:               fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush dts acpi mmx fxsr sse sse2 ss ht tm pbe syscall nx pdpe1gb rdtscp lm constant_tsc arch_perfmon pebs bts rep_good nopl xtopology nonstop_tsc cpuid aperfmperf pni pclmulqdq dtes64 monitor ds_cpl vmx smx est tm2 ssse3 sdbg fma cx16 xtpr pdcm pcid dca sse4_1 sse4_2 x2apic movbe popcnt tsc_deadline_timer aes xsave avx f16c rdrand lahf_lm abm 3dnowprefetch cpuid_fault epb cat_l3 cdp_l3 invpcid_single pti intel_ppin ssbd ibrs ibpb stibp tpr_shadow vnmi flexpriority ept vpid fsgsbase tsc_adjust bmi1 hle avx2 smep bmi2 erms invpcid rtm cqm rdt_a rdseed adx smap intel_pt xsaveopt cqm_llc cqm_occup_llc cqm_mbm_total cqm_mbm_local dtherm ida arat pln pts md_clear flush_l1d

 ***CMake***
 /datasets/rzamora/miniconda3/envs/nvt_dev/bin/cmake
 cmake version 3.18.0

 CMake suite maintained and supported by Kitware (kitware.com/cmake).

 ***g++***
 /usr/bin/g++
 g++ (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
 Copyright (C) 2017 Free Software Foundation, Inc.
 This is free software; see the source for copying conditions.  There is NO
 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.


 ***nvcc***
 /usr/local/cuda/bin/nvcc
 nvcc: NVIDIA (R) Cuda compiler driver
 Copyright (c) 2005-2019 NVIDIA Corporation
 Built on Wed_Oct_23_19:24:38_PDT_2019
 Cuda compilation tools, release 10.2, V10.2.89

 ***Python***
 /datasets/rzamora/miniconda3/envs/nvt_dev/bin/python
 Python 3.7.8

 ***Environment Variables***
 PATH                            : /datasets/rzamora/miniconda3/envs/nvt_dev/bin:/home/nfs/rzamora/.vscode-server/bin/cd9ea6488829f560dc949a8b2fb789f3cdc05f5d/bin:/home/nfs/rzamora/bin:/home/nfs/rzamora/.local/bin:/datasets/rzamora/miniconda3/bin:/datasets/rzamora/miniconda3/condabin:/usr/local/cuda/bin:/opt/bin:/home/nfs/rzamora/.vscode-server/bin/cd9ea6488829f560dc949a8b2fb789f3cdc05f5d/bin:/home/nfs/rzamora/bin:/home/nfs/rzamora/.local/bin:/usr/local/cuda/bin:/opt/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin
 LD_LIBRARY_PATH                 :
 NUMBAPRO_NVVM                   :
 NUMBAPRO_LIBDEVICE              :
 CONDA_PREFIX                    : /datasets/rzamora/miniconda3/envs/nvt_dev
 PYTHON_PATH                     :

 ***conda packages***
 /datasets/rzamora/miniconda3/bin/conda
 # packages in environment at /datasets/rzamora/miniconda3/envs/nvt_dev:
 #
 # Name                    Version                   Build  Channel
 _libgcc_mutex             0.1                 conda_forge    conda-forge
 _openmp_mutex             4.5                       1_gnu    conda-forge
 abseil-cpp                20200225.2           he1b5a44_2    conda-forge
 aiohttp                   3.6.2            py37h516909a_0    conda-forge
 alabaster                 0.7.12                     py_0    conda-forge
 appdirs                   1.4.3                      py_1    conda-forge
 argon2-cffi               20.1.0           py37h8f50634_1    conda-forge
 arrow-cpp                 0.17.1          py37h1234567_11_cuda    conda-forge
 arrow-cpp-proc            1.0.0                      cuda    conda-forge
 async-timeout             3.0.1                   py_1000    conda-forge
 attrs                     19.3.0                     py_0    conda-forge
 aws-sdk-cpp               1.7.164              hba45d7a_2    conda-forge
 babel                     2.8.0                      py_0    conda-forge
 backcall                  0.2.0              pyh9f0ad1d_0    conda-forge
 backports                 1.0                        py_2    conda-forge
 backports.functools_lru_cache 1.6.1                      py_0    conda-forge
 black                     19.10b0                    py_4    conda-forge
 bleach                    3.1.5              pyh9f0ad1d_0    conda-forge
 bokeh                     2.1.1            py37hc8dfbb8_0    conda-forge
 boost-cpp                 1.72.0               h7b93d67_2    conda-forge
 brotli                    1.0.7             he1b5a44_1004    conda-forge
 brotlipy                  0.7.0           py37h8f50634_1000    conda-forge
 bzip2                     1.0.8                h516909a_2    conda-forge
 c-ares                    1.16.1               h516909a_0    conda-forge
 ca-certificates           2020.6.20            hecda079_0    conda-forge
 certifi                   2020.6.20        py37hc8dfbb8_0    conda-forge
 cffi                      1.14.1           py37h2b28604_0    conda-forge
 cfgv                      3.2.0                      py_0    conda-forge
 chardet                   3.0.4           py37hc8dfbb8_1006    conda-forge
 clang                     8.0.1                hc9558a2_2    conda-forge
 clang-tools               8.0.1                hc9558a2_2    conda-forge
 clangxx                   8.0.1                         2    conda-forge
 click                     7.1.2              pyh9f0ad1d_0    conda-forge
 cloudpickle               1.5.0                      py_0    conda-forge
 cmake                     3.18.0               h5c55442_0    conda-forge
 cmake_setuptools          0.1.3                      py_0    rapidsai
 commonmark                0.9.1                      py_0    conda-forge
 cryptography              3.0              py37hb09aad4_0    conda-forge
 cudatoolkit               10.2.89              h6bb024c_0    nvidia
 cudf                      0.15.0a0+4902.ge51115fba           dev_0    <develop>
 cudnn                     7.6.5                cuda10.2_0
 cupy                      7.8.0            py37h940342b_0    conda-forge
 curl                      7.71.1               he644dc0_5    conda-forge
 cython                    0.29.21          py37h3340039_0    conda-forge
 cytoolz                   0.10.1           py37h516909a_0    conda-forge
 dask                      2.23.0+8.gcd1883e4           dev_0    <develop>
 dask-core                 2.23.0                     py_0    conda-forge
 dask-cuda                 0.14.0b0+95.gb24567d           dev_0    <develop>
 dask-cudf                 0.15.0a0+4902.ge51115fba           dev_0    <develop>
 dask-labextension         3.0.0                      py_0    conda-forge
 decorator                 4.4.2                      py_0    conda-forge
 defusedxml                0.6.0                      py_0    conda-forge
 distlib                   0.3.1              pyh9f0ad1d_0    conda-forge
 distributed               2.23.0+3.g67a9a596           dev_0    <develop>
 dlpack                    0.3                  he1b5a44_1    conda-forge
 docutils                  0.16             py37hc8dfbb8_1    conda-forge
 double-conversion         3.1.5                he1b5a44_2    conda-forge
 editdistance              0.5.3            py37h3340039_1    conda-forge
 entrypoints               0.3             py37hc8dfbb8_1001    conda-forge
 expat                     2.2.9                he1b5a44_2    conda-forge
 fastavro                  0.24.2           py37h8f50634_0    conda-forge
 fastrlock                 0.5              py37h3340039_0    conda-forge
 filelock                  3.0.12             pyh9f0ad1d_0    conda-forge
 flake8                    3.8.3                      py_1    conda-forge
 flatbuffers               1.12.0               he1b5a44_0    conda-forge
 freetype                  2.10.2               he06d7ca_0    conda-forge
 fsspec                    0.8.0                      py_0    conda-forge
 future                    0.18.2           py37hc8dfbb8_1    conda-forge
 gflags                    2.2.2             he1b5a44_1004    conda-forge
 glog                      0.4.0                h49b9bf7_3    conda-forge
 gmp                       6.2.0                he1b5a44_2    conda-forge
 grpc-cpp                  1.30.2               heedbac9_0    conda-forge
 heapdict                  1.0.1                      py_0    conda-forge
 hypothesis                5.26.1                     py_0    conda-forge
 icu                       67.1                 he1b5a44_0    conda-forge
 identify                  1.4.28             pyh9f0ad1d_0    conda-forge
 idna                      2.10               pyh9f0ad1d_0    conda-forge
 imagesize                 1.2.0                      py_0    conda-forge
 importlib-metadata        1.7.0            py37hc8dfbb8_0    conda-forge
 importlib_metadata        1.7.0                         0    conda-forge
 iniconfig                 1.0.1              pyh9f0ad1d_0    conda-forge
 ipykernel                 5.3.4            py37h43977f1_0    conda-forge
 ipython                   7.17.0           py37hc6149b9_0    conda-forge
 ipython_genutils          0.2.0                      py_1    conda-forge
 isort                     5.0.7            py37hc8dfbb8_0    conda-forge
 jedi                      0.17.2           py37hc8dfbb8_0    conda-forge
 jinja2                    2.11.2             pyh9f0ad1d_0    conda-forge
 jpeg                      9d                   h516909a_0    conda-forge
 json5                     0.9.5                      py_0
 jsonschema                3.2.0            py37hc8dfbb8_1    conda-forge
 jupyter-server-proxy      1.5.0                      py_0    conda-forge
 jupyter_client            6.1.6                      py_0    conda-forge
 jupyter_core              4.6.3            py37hc8dfbb8_1    conda-forge
 jupyterlab                2.1.5                      py_0
 jupyterlab-nvdashboard    0.4.0                    pypi_0    pypi
 jupyterlab_server         1.2.0                      py_0
 krb5                      1.17.1               hfafb76e_2    conda-forge
 lcms2                     2.11                 hbd6801e_0    conda-forge
 ld_impl_linux-64          2.34                 hc38a660_9    conda-forge
 libblas                   3.8.0               17_openblas    conda-forge
 libcblas                  3.8.0               17_openblas    conda-forge
 libcurl                   7.71.1               hcdd3856_5    conda-forge
 libedit                   3.1.20191231         he28a2e2_2    conda-forge
 libev                     4.33                 h516909a_0    conda-forge
 libevent                  2.1.10               hcdb4288_1    conda-forge
 libffi                    3.2.1             he1b5a44_1007    conda-forge
 libgcc-ng                 9.3.0               h24d8f2e_15    conda-forge
 libgfortran-ng            7.5.0               hdf63c60_15    conda-forge
 libgomp                   9.3.0               h24d8f2e_15    conda-forge
 liblapack                 3.8.0               17_openblas    conda-forge
 libllvm10                 10.0.1               he513fc3_1    conda-forge
 libllvm8                  8.0.1                hc9558a2_0    conda-forge
 libllvm9                  9.0.1                he513fc3_1    conda-forge
 libnghttp2                1.41.0               hab1572f_1    conda-forge
 libopenblas               0.3.10          pthreads_hb3c22a3_4    conda-forge
 libpng                    1.6.37               hed695b0_2    conda-forge
 libprotobuf               3.12.4               h8b12597_0    conda-forge
 librmm                    0.15.0a200819   cuda10.2_gdba07ba_719    rapidsai-nightly
 libsodium                 1.0.18               h516909a_0    conda-forge
 libssh2                   1.9.0                hab1572f_5    conda-forge
 libstdcxx-ng              9.3.0               hdf63c60_15    conda-forge
 libtiff                   4.1.0                hc7e4089_6    conda-forge
 libuv                     1.38.0               h516909a_0    conda-forge
 libwebp-base              1.1.0                h516909a_3    conda-forge
 llvmlite                  0.33.0           py37hc6ec683_1
 locket                    0.2.0                      py_2    conda-forge
 lz4-c                     1.9.2                he1b5a44_2    conda-forge
 markdown                  3.2.2                      py_0    conda-forge
 markupsafe                1.1.1            py37h8f50634_1    conda-forge
 mccabe                    0.6.1                      py_1    conda-forge
 mistune                   0.8.4           py37h8f50634_1001    conda-forge
 more-itertools            8.4.0                      py_0    conda-forge
 msgpack-python            1.0.0            py37h99015e2_1    conda-forge
 multidict                 4.7.5            py37h8f50634_1    conda-forge
 nbconvert                 5.6.1            py37hc8dfbb8_1    conda-forge
 nbformat                  5.0.7                      py_0    conda-forge
 nbsphinx                  0.7.1              pyh9f0ad1d_0    conda-forge
 nccl                      2.7.8.1              hc6a2c23_0    conda-forge
 ncurses                   6.2                  he1b5a44_1    conda-forge
 nodeenv                   1.4.0              pyh9f0ad1d_0    conda-forge
 nodejs                    14.8.0               h568c755_0    conda-forge
 notebook                  6.1.3            py37hc8dfbb8_0    conda-forge
 numba                     0.50.1           py37h0573a6f_1
 numpy                     1.19.1           py37h8960a57_0    conda-forge
 numpydoc                  1.1.0              pyh9f0ad1d_0    conda-forge
 nvtabular                 0.1.1                     dev_0    <develop>
 olefile                   0.46                       py_0    conda-forge
 openssl                   1.1.1g               h516909a_1    conda-forge
 packaging                 20.4               pyh9f0ad1d_0    conda-forge
 pandas                    1.0.5            py37h0da4684_0    conda-forge
 pandoc                    1.19.2                        0    conda-forge
 pandocfilters             1.4.2                      py_1    conda-forge
 parquet-cpp               1.5.1                         2    conda-forge
 parso                     0.7.1              pyh9f0ad1d_0    conda-forge
 partd                     1.1.0                      py_0    conda-forge
 pathspec                  0.8.0              pyh9f0ad1d_0    conda-forge
 pexpect                   4.8.0            py37hc8dfbb8_1    conda-forge
 pickleshare               0.7.5           py37hc8dfbb8_1001    conda-forge
 pillow                    7.2.0            py37h718be6c_1    conda-forge
 pip                       20.2.2                     py_0    conda-forge
 pluggy                    0.13.1           py37hc8dfbb8_2    conda-forge
 pre-commit                2.6.0            py37hc8dfbb8_0    conda-forge
 pre_commit                2.6.0                         0    conda-forge
 prometheus_client         0.8.0              pyh9f0ad1d_0    conda-forge
 prompt-toolkit            3.0.6                      py_0    conda-forge
 psutil                    5.7.2            py37h8f50634_0    conda-forge
 ptyprocess                0.6.0                   py_1001    conda-forge
 py                        1.9.0              pyh9f0ad1d_0    conda-forge
 pyarrow                   0.17.1          py37h1234567_11_cuda    conda-forge
 pycodestyle               2.6.0              pyh9f0ad1d_0    conda-forge
 pycparser                 2.20               pyh9f0ad1d_2    conda-forge
 pyflakes                  2.2.0              pyh9f0ad1d_0    conda-forge
 pygments                  2.6.1                      py_0    conda-forge
 pynvml                    8.0.4                    pypi_0    pypi
 pyopenssl                 19.1.0                     py_1    conda-forge
 pyparsing                 2.4.7              pyh9f0ad1d_0    conda-forge
 pyrsistent                0.16.0           py37h8f50634_0    conda-forge
 pysocks                   1.7.1            py37hc8dfbb8_1    conda-forge
 pytest                    6.0.1            py37hc8dfbb8_0    conda-forge
 python                    3.7.8           h6f2ec95_1_cpython    conda-forge
 python-dateutil           2.8.1                      py_0    conda-forge
 python_abi                3.7                     1_cp37m    conda-forge
 pytz                      2020.1             pyh9f0ad1d_0    conda-forge
 pyyaml                    5.3.1            py37h8f50634_0    conda-forge
 pyzmq                     19.0.2           py37hac76be4_0    conda-forge
 rapidjson                 1.1.0             he1b5a44_1002    conda-forge
 re2                       2020.07.06           he1b5a44_1    conda-forge
 readline                  8.0                  he28a2e2_2    conda-forge
 recommonmark              0.6.0                      py_0    conda-forge
 regex                     2020.7.14        py37h8f50634_0    conda-forge
 requests                  2.24.0             pyh9f0ad1d_0    conda-forge
 rhash                     1.3.6             h14c3975_1001    conda-forge
 rmm                       0.15.0a200819   cuda_10.2_py37_gdba07ba_719    rapidsai-nightly
 send2trash                1.5.0                      py_0    conda-forge
 setuptools                49.6.0           py37hc8dfbb8_0    conda-forge
 simpervisor               0.3                        py_1    conda-forge
 six                       1.15.0             pyh9f0ad1d_0    conda-forge
 snappy                    1.1.8                he1b5a44_3    conda-forge
 snowballstemmer           2.0.0                      py_0    conda-forge
 sortedcontainers          2.2.2              pyh9f0ad1d_0    conda-forge
 spdlog                    1.7.0                hc9558a2_2    conda-forge
 sphinx                    3.2.1                      py_0    conda-forge
 sphinx-copybutton         0.3.0              pyh9f0ad1d_0    conda-forge
 sphinx-markdown-tables    0.0.14             pyh9f0ad1d_1    conda-forge
 sphinx-multiversion       0.2.4                    pypi_0    pypi
 sphinx_rtd_theme          0.5.0              pyh9f0ad1d_0    conda-forge
 sphinxcontrib-applehelp   1.0.2                      py_0    conda-forge
 sphinxcontrib-devhelp     1.0.2                      py_0    conda-forge
 sphinxcontrib-htmlhelp    1.0.3                      py_0    conda-forge
 sphinxcontrib-jsmath      1.0.1                      py_0    conda-forge
 sphinxcontrib-qthelp      1.0.3                      py_0    conda-forge
 sphinxcontrib-serializinghtml 1.1.4                      py_0    conda-forge
 sphinxcontrib-websupport  1.2.4              pyh9f0ad1d_0    conda-forge
 sqlite                    3.33.0               h4cf870e_0    conda-forge
 streamz                   0.5.5                    pypi_0    pypi
 tbb                       2020.1               hc9558a2_0    conda-forge
 tblib                     1.6.0                      py_0    conda-forge
 terminado                 0.8.3            py37hc8dfbb8_1    conda-forge
 testpath                  0.4.4                      py_0    conda-forge
 thrift-cpp                0.13.0               h62aa4f2_3    conda-forge
 tk                        8.6.10               hed695b0_0    conda-forge
 toml                      0.10.1             pyh9f0ad1d_0    conda-forge
 toolz                     0.10.0                     py_0    conda-forge
 tornado                   6.0.4            py37h8f50634_1    conda-forge
 traitlets                 4.3.3            py37hc8dfbb8_1    conda-forge
 typed-ast                 1.4.1            py37h516909a_0    conda-forge
 typing_extensions         3.7.4.2                    py_0    conda-forge
 urllib3                   1.25.10                    py_0    conda-forge
 virtualenv                20.0.20          py37hc8dfbb8_1    conda-forge
 wcwidth                   0.2.5              pyh9f0ad1d_1    conda-forge
 webencodings              0.5.1                      py_1    conda-forge
 wheel                     0.35.1             pyh9f0ad1d_0    conda-forge
 xz                        5.2.5                h516909a_1    conda-forge
 yaml                      0.2.5                h516909a_0    conda-forge
 yarl                      1.4.2            py37h7b6447c_0
 zeromq                    4.3.2                he1b5a44_3    conda-forge
 zict                      2.0.0                      py_0    conda-forge
 zipp                      3.1.0                      py_0    conda-forge
 zlib                      1.2.11            h516909a_1007    conda-forge
 zstd                      1.4.5                h6597ccf_2    conda-forge

Additional context
Causing NVTabular tests to fail with latest cudf

! - Release bug cuIO

Most helpful comment

@nvdbaranec and I found the root cause and have a tentative fix.
Dave will take over and open a PR once the fix is thoroughly tested.

All 12 comments

Somewhere a column_view is being constructed with a non-zero null_count but a nullptr null_mask.

Updated (simpler) Repro:

import cudf

path = "test.parquet"
df = cudf.DataFrame({"a": ["a", None, "c"], "b": [1, 2, 3]})
df.to_parquet(path, index=False)

cudf.read_parquet(path, columns=["a", "b"]) # GOOD
cudf.read_parquet(path, columns=["b", "a"])
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-38-96c014a638fa> in <module>
----> 1 cudf.read_parquet(path, columns=["b", "a"])

~/workspace/rapids_dev/cudf/python/cudf/cudf/io/parquet.py in read_parquet(filepath_or_buffer, engine, columns, row_groups, skip_rows, num_rows, strings_to_categorical, use_pandas_metadata, *args, **kwargs)
    212             num_rows=num_rows,
    213             strings_to_categorical=strings_to_categorical,
--> 214             use_pandas_metadata=use_pandas_metadata,
    215         )
    216     else:

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/parquet.pyx in cudf._lib.parquet.read_parquet()

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/parquet.pyx in cudf._lib.parquet.read_parquet()

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/table.pyx in cudf._lib.table.Table.from_unique_ptr()

~/workspace/rapids_dev/cudf/python/cudf/cudf/_lib/column.pyx in cudf._lib.column.Column.from_unique_ptr()

RuntimeError: cuDF failure at: /home/nfs/rzamora/workspace/rapids_dev/cudf/cpp/src/column/column_view.cpp:56: Invalid null mask for non-zero null count.

DataFrame (for clarity):

      a  b
0     a  1
1  <NA>  2
2     c  3

Got local repro with the updated code. Looking for root cause.

@nvdbaranec in case this looks related to his recent PR.

Progress so far:
It looks like the error actually comes from the integer column. After page data decode, null count for col "b" is three (should be zero), while it's zero for col "a" (should be one).
The integer col "b" is created as not nullable (correctly), so this aligns with the error message.
I still haven't found how this is caused by column selection. Current assumption is that there is a mix-up between selected column index and output column index at some point, in code related to data validity.

Just saw this. Taking a look.

Some more isolation info:
issue also reproes with
df = cudf.DataFrame({"a": [1, None, 3], "b": [1, 2, 3]}) or
df = cudf.DataFrame({"a": [1, 2, 3], "b": [1, None, 3]})
but not with
df = cudf.DataFrame({"a": [1, None, 3], "b": [1, None, 3]})
In the repro case the valid count is still incorrect in both columns.

Confirmed I can repro this purely on the cpp side as well. Definitely related to reading the columns out of order. Probably an easy fix.

@nvdbaranec and I found the root cause and have a tentative fix.
Dave will take over and open a PR once the fix is thoroughly tested.

Vukasin gets the credit here - he zenned the issue.

Thanks for digging in here @vuule !

Vukasin gets the credit here - he zenned the issue.

It was a clear team effort. I insist on a 50/50 split :)

Was this page helpful?
0 / 5 - 0 ratings