Code

Use fusedL2NN in ANN kmeans#821

Code

Merged

rapids-bot[bot] merged 20 commits intorapidsai:branch-22.10rapidsai/raft:branch-22.10from Nyrio:enh-ann-kmeansNyrio/raft:enh-ann-kmeansCopy head branch name to clipboardOct 4, 2022

+548-371

Commits

Commits on Sep 6, 2022

• Initial integration of fusedL2NN in ANN kmeans

  Nyrio

  committed
  90730e3

Commits on Sep 7, 2022

• Precompute X norm for L2 metrics

  Nyrio

  committed
  e69d532

Commits on Sep 8, 2022

• Fix integer overflow and pass index type to predict_float_core

  Nyrio

  committed
  c42c0b7

Commits on Sep 12, 2022

• Merge remote-tracking branch 'origin/branch-22.10' into enh-ann-kmeans

  Nyrio

  committed
  eb2ee66

• Fix for integer types

  Nyrio

  committed
  76d4908

• Cleanup debug function

  Nyrio

  committed
  4b497ff

Commits on Sep 13, 2022

• Use int64_t indices in IVF-Flat test to enable large test cases

  Nyrio

  committed
  fd290fa

• Replace if by switch-case, C-style casts by C++ style, use int8 shared mem instead of template param

  Nyrio

  committed
  8cfee2c

Commits on Sep 22, 2022

• Do not batch when using fusedL2NN

  Nyrio

  committed
  4736f93

• Support for unsigned label type

  Nyrio

  committed
  2f8500d

Commits on Sep 23, 2022

• Make pre-computed norm an optional argument

  Nyrio

  committed
  0a1ce14

Commits on Sep 26, 2022

• Rewrite batch heuristic + change hard-coded integer types to templated

  Nyrio

  committed
  ab2890a

• Merge remote-tracking branch 'origin/branch-22.10' into enh-ann-kmeans

  Show description for ef9b281
  Nyrio

  committed
  ef9b281

Commits on Sep 29, 2022

• Index type capacity assertion + use make_device_mdarray with memory resource

  Nyrio

  committed
  1d0ab4c

Commits on Sep 30, 2022

• Use thrust::gather instead of copy_selected for norm

  Nyrio

  committed
  3dcac56

• Solve warnings due to printf qualifier / type mismatch

  Nyrio

  committed
  0bb179d

• Merge remote-tracking branch 'origin/branch-22.10' into enh-ann-kmeans

  Show description for 0e4ad82
  Nyrio

  committed
  0e4ad82

• Use fusedL2NN wrapper compiled in distance library

  Nyrio

  committed
  5cfb67e

Commits on Oct 4, 2022

• Merge remote-tracking branch 'origin/branch-22.10' into enh-ann-kmeans

  Nyrio

  committed
  4e35df5

• Fix ivf-pq / ann utils merge

  Nyrio

  committed
  fa946c5