Code

KNN select-top-k variants#551

Code

Merged

rapids-bot[bot] merged 43 commits intorapidsai:branch-22.04rapidsai/raft:branch-22.04from achirkin:enh-knn-topk-variantsachirkin/raft:enh-knn-topk-variantsCopy head branch name to clipboardMar 30, 2022

+2,373-337

Commits

Commits on Mar 11, 2022

• Integrate new select-top-k implementations

  achirkin

  committed
  179e1df

• warpsort_topk: refactoring and fixing some bugs

  achirkin

  committed
  4749295

• Allow passing indices along with keys (values).

  achirkin

  committed
  8504d32

• Adapt to the new bench

  achirkin

  committed
  cef3253

• Use the pooled allocator helper

  achirkin

  committed
  535fa0d

• Remove the step of calculating required buf size.

  achirkin

  committed
  7d10507

• Remove unused code

  achirkin

  committed
  ba66efa

• Allow different types in select-k functions (float/double, int/size_t)

  achirkin

  committed
  3eab24b

Commits on Mar 17, 2022

• More refactoring and comments

  achirkin

  committed
  659bc18

• Update knn.cuh docs

  achirkin

  committed
  8b6351b

Commits on Mar 18, 2022

• Add more comments

  achirkin

  committed
  a43e462

• Use radix top-k as reference, because it supports larger k

  achirkin

  committed
  45f6a35

• Add more comments and refactor vectorized_process

  achirkin

  committed
  0fe93d2

Commits on Mar 21, 2022

• Make bitonic sort use less template parameters for faster compile times

  achirkin

  committed
  50800a4

• Use gridDim.y for the batch dimension to simplify math and use less registers

  achirkin

  committed
  78805f0

• Update tests

  achirkin

  committed
  9cf1f33

• Allow larger batch sizes for radix_topk

  achirkin

  committed
  90293dc

• Merge branch 'branch-22.04' into enh-knn-topk-variants

  achirkin

  committed
  b38c80e

Commits on Mar 22, 2022

• Update docs

  achirkin

  committed
  48ac5c7

Commits on Mar 23, 2022

• More cosmetic refactoring

  achirkin

  committed
  fa76a4d

• Even more cosmetic refactoring

  achirkin

  committed
  3285de5

• Flip the ascending/descending flag for radix_topk

  achirkin

  committed
  faecc32

• Even more cosmetic refactoring

  achirkin

  committed
  db24b10

• Fix a typo

  achirkin

  committed
  a30a2fc

• Rename one of the 'add' overloads to reflect it should be used only once

  achirkin

  committed
  c722d9f

Commits on Mar 24, 2022

• Refactor names and document radix_topk

  achirkin

  committed
  fe95ded

• Choose the batch size dynamically

  achirkin

  committed
  00a62a4

• Rename the detail::topk folder

  achirkin

  committed
  52f863e

• Add the high-level algorithm description

  achirkin

  committed
  2a78c1f

• Rename the warpsort classes

  achirkin

  committed
  d811f75

• Fix a typo

  achirkin

  committed
  fcab684

Commits on Mar 25, 2022

• Merge remote-tracking branch 'rapidsai/branch-22.04' into enh-knn-topk-variants

  achirkin

  committed
  dcb17fe

• Clarify some parts of documentsion for bitonic sort

  achirkin

  committed
  84de3f2

Commits on Mar 28, 2022

• Update cpp/include/raft/spatial/knn/detail/topk/warpsort_topk.cuh

  Show description for 25ff099
  

  achirkin

  and

  tfeher

  authored
  25ff099

• Update cpp/include/raft/spatial/knn/detail/topk/warpsort_topk.cuh

  Show description for 99f6feb
  

  achirkin

  and

  tfeher

  authored
  99f6feb

• Update cpp/include/raft/spatial/knn/detail/topk/warpsort_topk.cuh

  Show description for deb7e44
  

  achirkin

  and

  tfeher

  authored
  deb7e44

• Update cpp/include/raft/spatial/knn/detail/topk/warpsort_topk.cuh

  Show description for 073d0f5
  

  achirkin

  and

  tfeher

  authored
  073d0f5

• Address review comments

  achirkin

  committed
  a811740

• Slightly reduce the number of tests for faster CI

  achirkin

  committed
  ff2d6e6

• Couple more comments

  achirkin

  committed
  e2f7d86

• Address more comments

  achirkin

  committed
  1936abd

Commits on Mar 29, 2022

• Remove commented-out bench cases

  achirkin

  committed
  6b3804c

• Change some bench cases

  achirkin

  committed
  bea83b3