Code

CAGRA#1375

Code

Merged

rapids-bot[bot] merged 47 commits intorapidsai:branch-23.04rapidsai/raft:branch-23.04from tfeher:cagra_experimentaltfeher/raft:cagra_experimentalCopy head branch name to clipboardApr 6, 2023

+7,157-2

Commits

Commits on Mar 27, 2023

• Add CAGRA, initial experimental version

  tfeher

  committed
  8793adb

• Restructuring search params in progress

  tfeher

  committed
  2748758

• replacing printf statements with RAFT_LOG_DEBUG

  tfeher

  committed
  c51cc7a

• remove topk.cu

  tfeher

  committed
  25d35ad

Commits on Mar 28, 2023

• Fix logging, revert some of the search_params refactoring

  tfeher

  committed
  9adb9b0

Commits on Mar 29, 2023

• adding specializations

  tfeher

  committed
  9dd0d46

• corrections

  tfeher

  committed
  d844e78

• Enabled test for distance values, test team size

  tfeher

  committed
  7c7819c

• added int8 and uint8 test and specializations

  tfeher

  committed
  7991a56

• correct copyright year for test files

  tfeher

  committed
  eb46fcf

• temporarily disabling int8 & uint8 tests

  tfeher

  committed
  60dfb3d

• Adding new search_plan

  tfeher

  committed
  1be9514

• single_cta params factored out

  tfeher

  committed
  7e8ba3f

Commits on Mar 30, 2023

• Single cta plan creation works

  tfeher

  committed
  e7cd010

• all search configs added to plan

  tfeher

  committed
  72d2dff

Commits on Mar 31, 2023

• refactored compiles

  tfeher

  committed
  0e30822

• Search dispatch refatored

  tfeher

  committed
  48a5161

• Remove old search dispatch

  tfeher

  committed
  af96473

• Remove old search specialization

  tfeher

  committed
  b9639a3

• Restoring CMakeLists

  tfeher

  committed
  07c4607

• Add serialization

  tfeher

  committed
  bc4ca55

• fix style error'

  tfeher

  committed
  078ce17

Commits on Apr 2, 2023

• Fixing topk size for small dataset

  tfeher

  committed
  6beb3f6

• Fix top-k size in search

  tfeher

  committed
  a27e9a7

• Restore team size calculation

  tfeher

  committed
  a0ee761

• Error message if dim is incompatible

  tfeher

  committed
  6ae6d32

• extend tests

  tfeher

  committed
  41b45e7

• Disabling multi_cta tests

  tfeher

  committed
  82f718e

• Merge branch 'branch-23.04' into cagra_experimental

  cjnolet

  authored
  ce99ead

Commits on Apr 3, 2023

• Replace CAGRA_HOST_DEVICE macros with RAFT_HOST_DEVICE

  tfeher

  committed
  38155ff

• Fix docstring

  tfeher

  committed
  d5f0a00

• Remove subsampling code from cagra_build

  tfeher

  committed
  2b0a14e

• Fix metric type test

  tfeher

  committed
  f0d9210

Commits on Apr 4, 2023

• Decrease the number of template specializations

  tfeher

  committed
  358c09c

• Reorder search params

  tfeher

  committed
  64a898e

Commits on Apr 5, 2023

• Fix style errors

  tfeher

  committed
  e3639a6

• Fix style

  tfeher

  committed
  3983147

• Fix typo

  tfeher

  committed
  c52dd33

• Add resources arg to prune

  tfeher

  committed
  7a249b5

• Remove all cagra specializations

  tfeher

  committed
  4afb03e

• Remove unused cagra specialization header

  tfeher

  committed
  93c470a

• Make refine_rate arg std::optional

  tfeher

  committed
  f962f22

• Replace hashmap_mode string with enum

  tfeher

  committed
  ccbe925

• Only keep test file for float data type

  tfeher

  committed
  619666c

• Add constxpr

  tfeher

  committed
  1df4859

• Remove constexpr

  tfeher

  committed
  c9be192

Commits on Apr 6, 2023

• Merge branch 'branch-23.04' into cagra_experimental

  benfred

  authored
  7f73aa9