Code

Python API for brute-force KNN#1292

Code

Merged

rapids-bot[bot] merged 114 commits intorapidsai:branch-23.04rapidsai/raft:branch-23.04from cjnolet:fea-2304-bfknncjnolet/raft:fea-2304-bfknnCopy head branch name to clipboardMar 23, 2023

+395-41

Commits

Commits on Jan 27, 2023

• Replace faiss bfKnn

  Show description for 57cfa20
  benfred

  committed
  57cfa20

  Copy full SHA for 57cfa20

• Merge branch 'branch-23.02' into bfknn

  benfred

  committed
  984c641

  Copy full SHA for 984c641

• fix merge

  benfred

  committed
  805abc7

  Copy full SHA for 805abc7

• Merge branch 'branch-23.02' into bfknn

  cjnolet

  authored
  3e21478

  Copy full SHA for 3e21478

• Fix bug with col_tiles < K

  benfred

  committed
  74bd44f

  Copy full SHA for 74bd44f

• Merge branch 'bfknn' of github.com:benfred/raft into bfknn

  benfred

  committed
  c69054d

  Copy full SHA for c69054d

Commits on Jan 30, 2023

• Include metric_arg in bfknn

  benfred

  committed
  1d9581b

  Copy full SHA for 1d9581b

• speedup compile times

  benfred

  committed
  b4cf88c

  Copy full SHA for b4cf88c

Commits on Jan 31, 2023

• Merge branch 'branch-23.02' into bfknn

  benfred

  committed
  98ffb70

  Copy full SHA for 98ffb70

• Suggestions from code review

  benfred

  committed
  5442d31

  Copy full SHA for 5442d31

• fixes

  benfred

  committed
  0f5d206

  Copy full SHA for 0f5d206

Commits on Feb 3, 2023

• Merge branch 'branch-23.04' into bfknn

  benfred

  committed
  cb2b750

  Copy full SHA for cb2b750

Commits on Feb 7, 2023

• use pairwise_distance specialization to speed up compile times

  benfred

  committed
  e870eb3

  Copy full SHA for e870eb3

• Merge branch 'branch-23.04' into bfknn

  benfred

  committed
  cd84397

  Copy full SHA for cd84397

• Use distance specializations

  benfred

  committed
  8445aed

  Copy full SHA for 8445aed

Commits on Feb 9, 2023

• Merge branch 'branch-23.04' into bfknn

  cjnolet

  authored
  e87633a

  Copy full SHA for e87633a

Commits on Feb 11, 2023

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  52bf729

  Copy full SHA for 52bf729

• Merge branch 'branch-23.04' into bfknn

  cjnolet

  authored
  d97ddb8

  Copy full SHA for d97ddb8

Commits on Feb 14, 2023

• use specializations in RBC code

  benfred

  committed
  5905b2d

  Copy full SHA for 5905b2d

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  d905266

  Copy full SHA for d905266

• use pw specializations in rbc

  benfred

  committed
  8eaba84

  Copy full SHA for 8eaba84

• use matrix::select_k in bfknn call

  benfred

  committed
  fe728e9

  Copy full SHA for fe728e9

Commits on Feb 15, 2023

• expose bf detail specialization

  benfred

  committed
  96e05e1

  Copy full SHA for 96e05e1

• Revert "use pw specializations in rbc"

  Show description for 59060b2
  benfred

  committed
  59060b2

  Copy full SHA for 59060b2

• Add tests for other metrics

  Show description for c734bac
  benfred

  committed
  c734bac

  Copy full SHA for c734bac

Commits on Feb 16, 2023

• Fix parameter order

  benfred

  committed
  c65e4bb

  Copy full SHA for c65e4bb

Commits on Feb 17, 2023

• Fix Lp distance

  benfred

  committed
  3830e53

  Copy full SHA for 3830e53

• Revert "use matrix::select_k in bfknn call"

  Show description for 3f0b9a7
  benfred

  committed
  3f0b9a7

  Copy full SHA for 3f0b9a7

• re-enable failing tests

  benfred

  committed
  3900570

  Copy full SHA for 3900570

• fix cosine/innerproduct in bfknn

  benfred

  committed
  8e71915

  Copy full SHA for 8e71915

• Test JensenShannon distance

  benfred

  committed
  f806bf6

  Copy full SHA for f806bf6

Commits on Feb 18, 2023

• support k up to 2048 in faiss select

  Show description for 3315dca
  benfred

  committed
  3315dca

  Copy full SHA for 3315dca

• Merge remote-tracking branch 'origin/branch-23.04' into bfknn

  benfred

  committed
  1b6eda2

  Copy full SHA for 1b6eda2

• cmake format

  benfred

  committed
  a83bef3

  Copy full SHA for a83bef3

• support k up to 2048 in faiss select

  Show description for 3b811a1
  benfred

  committed
  3b811a1

  Copy full SHA for 3b811a1

• Merge branch 'branch-23.04' into faiss_largek

  cjnolet

  authored
  9a19456

  Copy full SHA for 9a19456

• style

  benfred

  committed
  c39dc65

  Copy full SHA for c39dc65

• Merge branch 'faiss_largek' of github.com:benfred/raft into faiss_largek

  benfred

  committed
  c60e17f

  Copy full SHA for c60e17f

• code review suggestions

  benfred

  committed
  2752294

  Copy full SHA for 2752294

• Merge remote-tracking branch 'bf/faiss_largek' into bfknn

  benfred

  committed
  84f7a42

  Copy full SHA for 84f7a42

Commits on Feb 20, 2023

• Merge branch 'branch-23.04' into bfknn

  cjnolet

  authored
  901b898

  Copy full SHA for 901b898

Commits on Feb 21, 2023

• Remove ENABLE_NN_DEPENDENCIES option

  benfred

  committed
  1548a78

  Copy full SHA for 1548a78

• Merge branch 'branch-23.04' into bfknn

  benfred

  committed
  3fdc712

  Copy full SHA for 3fdc712

• Adding brute-force knn api to pylibraft

  cjnolet

  committed
  642f87d

  Copy full SHA for 642f87d

• Merge branch 'branch-23.04' into fea-2304-bfknn

  cjnolet

  authored
  1ac42af

  Copy full SHA for 1ac42af

• temporarily re-add faiss build targets

  Show description for 31c9cf2
  benfred

  committed
  31c9cf2

  Copy full SHA for 31c9cf2

• couple more files to re-add faiss

  benfred

  committed
  f7fd6a7

  Copy full SHA for f7fd6a7

Commits on Feb 22, 2023

• re-add faiss_mr

  benfred

  committed
  37d66d2

  Copy full SHA for 37d66d2

Commits on Feb 23, 2023

• explicitly include faiss_mr

  benfred

  committed
  a61c92f

  Copy full SHA for a61c92f

Commits on Feb 24, 2023

• Allow col_major input to bfknn

  benfred

  committed
  dbd31b2

  Copy full SHA for dbd31b2

• fix faiss queryempty test

  benfred

  committed
  fddecc3

  Copy full SHA for fddecc3

Commits on Feb 27, 2023

• exclude LP from fused

  benfred

  committed
  4687144

  Copy full SHA for 4687144

• Merge branch 'branch-23.04' into bfknn

  benfred

  committed
  bd3ff51

  Copy full SHA for bd3ff51

• use metric processor for cosine/correlation

  Show description for 06c8674
  benfred

  committed
  06c8674

  Copy full SHA for 06c8674

Commits on Feb 28, 2023

• exclude cosine

  benfred

  committed
  b44d15c

  Copy full SHA for b44d15c

Commits on Mar 1, 2023

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  5a582cd

  Copy full SHA for 5a582cd

• avoid l2expanded distance

  Show description for eb0271a
  benfred

  committed
  eb0271a

  Copy full SHA for eb0271a

• Merge branch 'bfknn' of github.com:benfred/raft into bfknn

  benfred

  committed
  616455c

  Copy full SHA for 616455c

Commits on Mar 6, 2023

• Expanded L2 Changes

  Show description for 4c41c63
  benfred

  committed
  4c41c63

  Copy full SHA for 4c41c63

Commits on Mar 7, 2023

• correct for small instabilities in l2sqrtexpanded distance

  benfred

  committed
  cdf1962

  Copy full SHA for cdf1962

• warp divergence

  benfred

  committed
  6a1e2d8

  Copy full SHA for 6a1e2d8

• clamp to 0

  benfred

  committed
  1e2817c

  Copy full SHA for 1e2817c

• threshold

  benfred

  committed
  4b56fac

  Copy full SHA for 4b56fac

Commits on Mar 8, 2023

• Transpose for fusedl2knn as well

  benfred

  committed
  4b41e2c

  Copy full SHA for 4b41e2c

• fix

  benfred

  committed
  455c952

  Copy full SHA for 455c952

• Fix stream handling on col-major inputs

  Show description for df46b65
  benfred

  committed
  df46b65

  Copy full SHA for df46b65

• Merge branch 'bfknn' of github.com:benfred/raft into bfknn

  benfred

  committed
  97a3c01

  Copy full SHA for 97a3c01

Commits on Mar 9, 2023

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  105bc96

  Copy full SHA for 105bc96

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  2828c3b

  Copy full SHA for 2828c3b

Commits on Mar 11, 2023

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  6e45267

  Copy full SHA for 6e45267

• Merge branch 'branch-23.04' into fea-2304-bfknn

  benfred

  authored
  f426510

  Copy full SHA for f426510

• Merge remote-tracking branch 'ben/bfknn' into fea-2304-bfknn

  cjnolet

  committed
  754ab9a

  Copy full SHA for 754ab9a

• Including specializations in bfknn runtime API

  cjnolet

  committed
  a35ec7b

  Copy full SHA for a35ec7b

Commits on Mar 13, 2023

• fix build for missing symbols

  benfred

  committed
  28ebeef

  Copy full SHA for 28ebeef

• Merge branch 'branch-23.04' into bfknn

  cjnolet

  authored
  9917324

  Copy full SHA for 9917324

Commits on Mar 14, 2023

• code review feedback

  benfred

  committed
  65d7725

  Copy full SHA for 65d7725

• matrix::fill and linalg::map_offset

  benfred

  committed
  e41ff88

  Copy full SHA for e41ff88

• build fix

  benfred

  committed
  5eb7d22

  Copy full SHA for 5eb7d22

• fix

  benfred

  committed
  e36d089

  Copy full SHA for e36d089

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  50e366f

  Copy full SHA for 50e366f

• move faiss_select into raft::neighbors namespace

  benfred

  committed
  e8f9c55

  Copy full SHA for e8f9c55

• move knn_merge parts to its own file

  benfred

  committed
  9f211a0

  Copy full SHA for 9f211a0

• Merge branch 'branch-23.04' into fea-2304-bfknn

  cjnolet

  authored
  f1011dc

  Copy full SHA for f1011dc

• Merge branch 'branch-23.04' into bfknn

  cjnolet

  authored
  9cbac3c

  Copy full SHA for 9cbac3c

Commits on Mar 15, 2023

• Merge remote-tracking branch 'origin/branch-23.04' into bfknn

  benfred

  committed
  0667526

  Copy full SHA for 0667526

• Merge remote-tracking branch 'ben/bfknn' into fea-2304-bfknn

  cjnolet

  committed
  b48ad31

  Copy full SHA for b48ad31

• Use stream pool

  benfred

  committed
  9593ae1

  Copy full SHA for 9593ae1

• Merge branch 'bfknn' of github.com:benfred/raft into bfknn

  benfred

  committed
  76d2b19

  Copy full SHA for 76d2b19

• use right handle for transpose

  benfred

  committed
  97753b0

  Copy full SHA for 97753b0

• set blas stream

  benfred

  committed
  a534538

  Copy full SHA for a534538

• error handling

  benfred

  committed
  237b7e1

  Copy full SHA for 237b7e1

Commits on Mar 16, 2023

• try to isolate stream failure

  Show description for 07290bc
  benfred

  committed
  07290bc

  Copy full SHA for 07290bc

• Move transpose code out of loop

  Show description for 2671a0e
  benfred

  committed
  2671a0e

  Copy full SHA for 2671a0e

• fix

  benfred

  committed
  7e0bb9b

  Copy full SHA for 7e0bb9b

• try transpose inside streampool again

  benfred

  committed
  b4c3284

  Copy full SHA for b4c3284

Commits on Mar 17, 2023

• one more try with cublasSetStream

  benfred

  committed
  92d82db

  Copy full SHA for 92d82db

• Merge branch 'branch-23.04' into bfknn

  benfred

  authored
  c84e560

  Copy full SHA for c84e560

• fix

  benfred

  committed
  21a1953

  Copy full SHA for 21a1953

• Merge branch 'bfknn' of github.com:benfred/raft into bfknn

  benfred

  committed
  80fb76c

  Copy full SHA for 80fb76c

• Merge remote-tracking branch 'rapidsai/branch-23.04' into fea-2304-bfknn

  cjnolet

  committed
  2d89994

  Copy full SHA for 2d89994

• Merge remote-tracking branch 'ben/bfknn' into fea-2304-bfknn

  cjnolet

  committed
  1b89ed1

  Copy full SHA for 1b89ed1

• Getting some tests running (no assertions yet)

  cjnolet

  committed
  5f743f1

  Copy full SHA for 5f743f1

• Merge branch 'branch-23.04' into fea-2304-bfknn

  cjnolet

  authored
  fa93cf7

  Copy full SHA for fa93cf7

Commits on Mar 22, 2023

• Adding pytest assertions and baseline for comparison

  cjnolet

  committed
  accfc3b

  Copy full SHA for accfc3b

• Merge branch 'fea-2304-bfknn' of github.com:cjnolet/raft into fea-2304-bfknn

  cjnolet

  committed
  0d7eba7

  Copy full SHA for 0d7eba7

Commits on Mar 23, 2023

• Merge remote-tracking branch 'rapidsai/branch-23.04' into fea-2304-bfknn

  cjnolet

  committed
  1c9ba9d

  Copy full SHA for 1c9ba9d

• Using correct size of indices

  cjnolet

  committed
  1e28f9f

  Copy full SHA for 1e28f9f

• Using correct number of distances too

  cjnolet

  committed
  e8879a3

  Copy full SHA for e8879a3

• Limiting the correct value

  cjnolet

  committed
  fb90502

  Copy full SHA for fb90502

• Merge branch 'branch-23.04' into fea-2304-bfknn

  cjnolet

  authored
  5b407b9

  Copy full SHA for 5b407b9

• Enabling more distances and fixing assertion

  cjnolet

  committed
  7c5c87b

  Copy full SHA for 7c5c87b

• Tests are passing.

  cjnolet

  committed
  4c9f85b

  Copy full SHA for 4c9f85b

• Review feedback

  cjnolet

  committed
  7400402

  Copy full SHA for 7400402

• Using proper import

  cjnolet

  committed
  d9016f2

  Copy full SHA for d9016f2