PQ preprocessor does not natively support padded datasets
Open
Nobody has claimed this yet.
feature request
- Dominant language
- Cuda
- Stars
- 854
- Forks
- 236
- Avg merge
- 3d 3h
- Merged PRs (30d)
- 62
Description
Currently, the make_device_pq_dataset can accept a padded_dataset type but internally copies it to a contiguous dense dataset since its helpers like train_from_rows do not support encoding padded datasets.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start in cpp/src/preprocessing/quantize/pq.cu around lines 99-111 and trace make_device_pq_dataset and its train_from_rows helpers. Determine how padded_dataset encoding can avoid the contiguous dense copy, then verify the preprocessor handles padded input correctly with relevant tests.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- data
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Active
- Clarity
- Mostly clear
- Newbie friendliness
- 58/100