NVIDIA / NVIDIA/cuvs

[FEA] CAGRA-Q support pq_len=16 and pq_len=8

Open
#287 1 comment 0 reactions 1 assignee View on GitHub

@enp1s0 is already working on this.

Since Aug 12, 2024.

feature request
Dominant language
Cuda
Stars
854
Forks
236
Avg merge
3d 3h
Merged PRs (30d)
62

Description

Currently CAGRA supports PQ compression with pq_len=2 ad pq_len=4. A larger compression ratio can be achieved if we allow larger pq_len values, e.g. 8 and 16.

pq_len is a template parameter of the distance computation kernels. In the current setting we would need to instantiate new kernels for the larger pq_len values. This would add ~ 160 MB to the binary size.

A preferred solution would be to make pq_len a runtime parameter.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.