Code

IVF-FLAT support k > 256#2169

Code

Merged

rapids-bot[bot] merged 20 commits intorapidsai:branch-24.04rapidsai/raft:branch-24.04from mfoerste4:ivfflat-max_k-gt-256mfoerste4/raft:ivfflat-max_k-gt-256Copy head branch name to clipboardMar 4, 2024

+694-462

Commits

Commits on Feb 9, 2024

• initial commit k>256

  mfoerste4

  committed
  ca5d090

• add missing include

  mfoerste4

  committed
  8da5427

• Merge branch 'branch-24.04' into ivfflat-max_k-gt-256

  mfoerste4

  authored
  e5d9e81

• remove some debug logging

  mfoerste4

  committed
  aa4f109

• resolved merge issue

  mfoerste4

  committed
  3c7524e

Commits on Feb 12, 2024

• align max_samples, remove sorting in selectk

  mfoerste4

  committed
  c0b5890

Commits on Feb 15, 2024

• compute max_samples & adjust query batch_size

  mfoerste4

  committed
  ae40620

• some fixes

  mfoerste4

  committed
  ae834f5

Commits on Feb 16, 2024

• merge with 24.04

  mfoerste4

  committed
  6ce9d85

• some more small adjustments

  mfoerste4

  committed
  2e95c57

• de-duplicate common code

  mfoerste4

  committed
  5282501

Commits on Feb 19, 2024

• add c++/python test, bugfixes

  mfoerste4

  committed
  5b136f8

• Merge branch 'branch-24.04' into ivfflat-max_k-gt-256

  mfoerste4

  authored
  a1f1f53

Commits on Feb 22, 2024

• review suggestions

  mfoerste4

  committed
  c42167c

• extract recompute_internal_state

  mfoerste4

  committed
  5c49e9d

• Merge branch 'branch-24.04' into ivfflat-max_k-gt-256

  mfoerste4

  authored
  9f4e7ca

Commits on Feb 23, 2024

• fix test compile

  mfoerste4

  committed
  760112f

• Merge branch 'ivfflat-max_k-gt-256' of github.com:mfoerste4/raft into ivfflat-max_k-gt-256

  mfoerste4

  committed
  a7fa668

• update copyright date

  mfoerste4

  committed
  439267c

• reverted eager optimization not copying anything when in != out...

  mfoerste4

  committed
  2384b98