Code

[RELEASE] raft v23.04#1405

Code

Merged

raydouglass merged 110 commits intomainrapidsai/raft:mainfrom branch-23.04rapidsai/raft:branch-23.04Copy head branch name to clipboardApr 12, 2023

+38,703-23,118

Commits

Commits on Jan 23, 2023

• DOC

  raydouglass

  committed
  9963c12

  Copy full SHA for 9963c12

• Merge pull request #1163 from rapidsai/branch-23.02

  Show description for 001ad8e
  GPUtester

  authored
  001ad8e

  Copy full SHA for 001ad8e

• Merge pull request #1164 from rapidsai/branch-23.02

  Show description for 4f33ddf
  GPUtester

  authored
  4f33ddf

  Copy full SHA for 4f33ddf

Commits on Jan 25, 2023

• Merge pull request #1174 from rapidsai/branch-23.02

  Show description for 7ca4dcb
  GPUtester

  authored
  7ca4dcb

  Copy full SHA for 7ca4dcb

• Merge pull request #1180 from rapidsai/branch-23.02

  Show description for 6d35222
  GPUtester

  authored
  6d35222

  Copy full SHA for 6d35222

Commits on Jan 26, 2023

• Merge pull request #1183 from rapidsai/branch-23.02

  Show description for 6efe8f1
  GPUtester

  authored
  6efe8f1

  Copy full SHA for 6efe8f1

• Merge pull request #1186 from rapidsai/branch-23.02

  Show description for c94ea74
  GPUtester

  authored
  c94ea74

  Copy full SHA for c94ea74

• Fix ucx-py version (#1184)

  Show description for 427c6d3
  ajschmidt8

  authored
  427c6d3

  Copy full SHA for 427c6d3

• Merge pull request #1193 from rapidsai/branch-23.02

  Show description for b0f99cd
  GPUtester

  authored
  b0f99cd

  Copy full SHA for b0f99cd

Commits on Jan 27, 2023

• Merge pull request #1199 from rapidsai/branch-23.02

  Show description for 47ecc63
  GPUtester

  authored
  47ecc63

  Copy full SHA for 47ecc63

• Merge pull request #1203 from rapidsai/branch-23.02

  Show description for 43f56b6
  GPUtester

  authored
  43f56b6

  Copy full SHA for 43f56b6

Commits on Jan 31, 2023

• Merge pull request #1214 from rapidsai/branch-23.02

  Show description for 4b4377e
  GPUtester

  authored
  4b4377e

  Copy full SHA for 4b4377e

• Merge pull request #1216 from rapidsai/branch-23.02

  Show description for 429d0dd
  GPUtester

  authored
  429d0dd

  Copy full SHA for 429d0dd

• Merge pull request #1217 from rapidsai/branch-23.02

  Show description for f40c86b
  GPUtester

  authored
  f40c86b

  Copy full SHA for f40c86b

• Merge pull request #1218 from rapidsai/branch-23.02

  Show description for 6307c86
  GPUtester

  authored
  6307c86

  Copy full SHA for 6307c86

Commits on Feb 1, 2023

• Merge pull request #1221 from rapidsai/branch-23.02

  Show description for af93c05
  GPUtester

  authored
  af93c05

  Copy full SHA for af93c05

Commits on Feb 2, 2023

• Merge pull request #1227 from rapidsai/branch-23.02

  Show description for 40bea04
  GPUtester

  authored
  40bea04

  Copy full SHA for 40bea04

• Merge pull request #1231 from rapidsai/branch-23.02

  Show description for d15fbe5
  GPUtester

  authored
  d15fbe5

  Copy full SHA for d15fbe5

Commits on Feb 3, 2023

• Merge pull request #1234 from rapidsai/branch-23.02

  Show description for f8d3f5b
  GPUtester

  authored
  f8d3f5b

  Copy full SHA for f8d3f5b

• Remove uint32 distance idx specializations (#1235)

  Show description for a850726
  cjnolet

  authored
  a850726

  Copy full SHA for a850726

• Merge pull request #1244 from rapidsai/branch-23.02

  Show description for 88cb31d
  GPUtester

  authored
  88cb31d

  Copy full SHA for 88cb31d

Commits on Feb 4, 2023

• Expose KMeans init_plus_plus in pylibraft (#1198)

  Show description for a39237c
  betatim

  authored
  a39237c

  Copy full SHA for a39237c

• Expose raft::handle_t in the public header (#1192)

  Show description for a426bc9
  vyasr

  authored
  a426bc9

  Copy full SHA for a426bc9

• Fix for svd API (#1190)

  Show description for bc764b8
  lowener

  authored
  bc764b8

  Copy full SHA for bc764b8

Commits on Feb 6, 2023

• Move date to build string in conda recipe (#1223)

  Show description for 16746ee
  ajschmidt8

  authored
  16746ee

  Copy full SHA for 16746ee

• Merge branch-23.02 into branch-23.04

  galipremsagar

  committed
  47961a0

  Copy full SHA for 47961a0

• Merge pull request #1251 from galipremsagar/branch-23.04-merge-branch-23.02

  Show description for 09f0fd9
  raydouglass

  authored
  09f0fd9

  Copy full SHA for 09f0fd9

Commits on Feb 8, 2023

• Consolidating ANN benchmarks and tests (#1243)

  Show description for ab11cba
  cjnolet

  authored
  ab11cba

  Copy full SHA for ab11cba

• Keeping only compute similarity specializations for uint64_t for now (#1255)

  Show description for ca34321
  cjnolet

  authored
  ca34321

  Copy full SHA for ca34321

• Update shared workflow branches (#1256)

  Show description for 9ca33f3
  ajschmidt8

  authored
  9ca33f3

  Copy full SHA for 9ca33f3

Commits on Feb 9, 2023

• Fix compile time explosion for minkowski distance (#1254)

  Show description for cd32d6f
  ahendriksen

  authored
  cd32d6f

  Copy full SHA for cd32d6f

• Merge pull request #1263 from rapidsai/branch-23.02

  Show description for d1e0d6c
  GPUtester

  authored
  d1e0d6c

  Copy full SHA for d1e0d6c

Commits on Feb 10, 2023

• Bump pinned deps to 23.4 (#1266)

  Show description for 611bda2
  vyasr

  authored
  611bda2

  Copy full SHA for 611bda2

• Remove duplicate librmm runtime dependency (#1264)

  Show description for bc3a9c5
  ajschmidt8

  authored
  bc3a9c5

  Copy full SHA for bc3a9c5

• Reduce error handling verbosity in CI tests scripts (#1259)

  Show description for c976be6
  AjayThorve

  authored
  c976be6

  Copy full SHA for c976be6

Commits on Feb 11, 2023

• Remove gpuCI scripts. (#1252)

  Show description for 5fdce85
  bdice

  authored
  5fdce85

  Copy full SHA for 5fdce85

• Remove no longer instantiated templates from list of extern template declarations (#1272)

  Show description for 4440c39
  vyasr

  authored
  4440c39

  Copy full SHA for 4440c39

• Fixign a couple small kmeans bugs (#1274)

  Show description for a572397
  cjnolet

  authored
  a572397

  Copy full SHA for a572397

• Unpin dask and distributed for development (#1253)

  Show description for 1cb4c34
  galipremsagar

  authored
  1cb4c34

  Copy full SHA for 1cb4c34

Commits on Feb 13, 2023

• Add innerproduct to the pairwise distance api (#1226)

  Show description for cca81c4
  benfred

  authored
  cca81c4

  Copy full SHA for cca81c4

• Remove deprecated headers (#1145)

  Show description for 2347e06
  lowener

  authored
  2347e06

  Copy full SHA for 2347e06

Commits on Feb 14, 2023

• fix build on cuda 11.5 (#1277)

  Show description for 7d6e4dc
  benfred

  authored
  7d6e4dc

  Copy full SHA for 7d6e4dc

Commits on Feb 15, 2023

• Fix the destruction of interruptible token registry (#1229)

  Show description for 27ca9b9
  achirkin

  authored
  27ca9b9

  Copy full SHA for 27ca9b9

Commits on Feb 16, 2023

• Fix inconsistency between the building doc and CMakeLists.txt (#1248)

  Show description for e14ec63
  yong-wang

  authored
  e14ec63

  Copy full SHA for e14ec63

Commits on Feb 17, 2023

• Add docs build job (#1168)

  Show description for 013da20
  AyodeAwe

  authored
  013da20

  Copy full SHA for 013da20

Commits on Feb 18, 2023

• IVF-PQ: store cluster data in individual lists and reduce templates (#1249)

  Show description for 3414616
  achirkin

  authored
  3414616

  Copy full SHA for 3414616

• CI: Remove specification of manual stage for check_style.sh script. (#1283)

  Show description for 02cfacf
  csadorf

  authored
  02cfacf

  Copy full SHA for 02cfacf

• matrix::select_k specializations (#1268)

  Show description for a277388
  achirkin

  authored
  a277388

  Copy full SHA for a277388

• support k up to 2048 in faiss select (#1287)

  Show description for d45eb1e
  benfred

  authored
  d45eb1e

  Copy full SHA for d45eb1e

Commits on Feb 21, 2023

• Initial port of auto-find-k (#1070)

  Show description for bbfb869
  cjnolet

  authored
  bbfb869

  Copy full SHA for bbfb869

Commits on Feb 23, 2023

• Adding util to get the device id for a pointer address (#1297)

  Show description for ad768f4
  cjnolet

  authored
  ad768f4

  Copy full SHA for ad768f4

Commits on Feb 24, 2023

• Fix nightly docs build script (#1305)

  Show description for 9ace862
  AyodeAwe

  authored
  9ace862

  Copy full SHA for 9ace862

• Enable dfg in pre-commit. (#1293)

  Show description for 90c8da1
  vyasr

  authored
  90c8da1

  Copy full SHA for 90c8da1

Commits on Feb 28, 2023

• Stop using versioneer to manage versions (#1301)

  Show description for 754ebc3
  vyasr

  authored
  754ebc3

  Copy full SHA for 754ebc3

• Make docs builds less verbose (#1302)

  Show description for ae15be7
  AyodeAwe

  authored
  ae15be7

  Copy full SHA for ae15be7

Commits on Mar 2, 2023

• Fix docs build to be pydata-sphinx-theme=0.13.0 compatible (#1311)

  Show description for 004f508
  galipremsagar

  authored
  004f508

  Copy full SHA for 004f508

Commits on Mar 6, 2023

• Update to GCC 11 (#1309)

  Show description for e7f0268
  bdice

  authored
  e7f0268

  Copy full SHA for e7f0268

Commits on Mar 7, 2023

• Add stream overloads to ivf_pq serialize/deserialize methods (#1315)

  Show description for 3ca7eac
  divyegala

  authored
  3ca7eac

  Copy full SHA for 3ca7eac

Commits on Mar 8, 2023

• Temporary buffer to view host or device memory in device (#1313)

  Show description for 206611c
  divyegala

  authored
  206611c

  Copy full SHA for 206611c

• Pass AWS_SESSION_TOKEN and SCCACHE_S3_USE_SSL vars to conda build (#1321)

  Show description for cf083e4
  ajschmidt8

  authored
  cf083e4

  Copy full SHA for cf083e4

Commits on Mar 9, 2023

• mdspan view for IVF-PQ API (#1236)

  Show description for 8a22373
  viclafargue

  authored
  8a22373

  Copy full SHA for 8a22373

• Numerical stability fixes for l2 pairwise distance (#1319)

  Show description for 0f774b9
  benfred

  authored
  0f774b9

  Copy full SHA for 0f774b9

• Consolidate linter configuration into pyproject.toml (#1317)

  Show description for 362dc93
  vyasr

  authored
  362dc93

  Copy full SHA for 362dc93

Commits on Mar 10, 2023

• Using int64_t specializations for ivf_pq and refine (#1325)

  Show description for 6b30de9
  cjnolet

  authored
  6b30de9

  Copy full SHA for 6b30de9

• Fixing remaining bug in ann_quantized (#1327)

  Show description for bcb0976
  cjnolet

  authored
  bcb0976

  Copy full SHA for bcb0976

• Simplify distance/detail to make is easier to dispatch to different kernel implementations (#1142)

  Show description for e4aec7b
  ahendriksen

  authored
  e4aec7b

  Copy full SHA for e4aec7b

Commits on Mar 11, 2023

• Fixing select_k specializations (#1330)

  Show description for c71673c
  cjnolet

  authored
  c71673c

  Copy full SHA for c71673c

Commits on Mar 12, 2023

• Small follow-up to specializations cleanup (#1332)

  Show description for b12fc82
  cjnolet

  authored
  b12fc82

  Copy full SHA for b12fc82

Commits on Mar 13, 2023

• Generic linalg::map (#1329)

  Show description for cb78f5f
  achirkin

  authored
  cb78f5f

  Copy full SHA for cb78f5f

Commits on Mar 14, 2023

• Migrate as much as possible to pyproject.toml (#1324)

  Show description for f8708e1
  vyasr

  authored
  f8708e1

  Copy full SHA for f8708e1

• Revert "Generic linalg::map (#1329)" (#1336)

  Show description for 7e328ff
  cjnolet

  authored
  7e328ff

  Copy full SHA for 7e328ff

• Generic linalg::map (#1337)

  Show description for 1053f29
  achirkin

  authored
  1053f29

  Copy full SHA for 1053f29

• Small updates to docs (#1339)

  Show description for 667d873
  cjnolet

  authored
  667d873

  Copy full SHA for 667d873

Commits on Mar 15, 2023

• IVF-Flat index splitting (#1271)

  Show description for 03e26b5
  lowener

  authored
  03e26b5

  Copy full SHA for 03e26b5

• Pass minimum target compile options through raft::raft (#1341)

  Show description for 8b4d2e5
  cjnolet

  authored
  8b4d2e5

  Copy full SHA for 8b4d2e5

• New Sparse Matrix APIs (#1279)

  Show description for fb84190
  cjnolet

  authored
  fb84190

  Copy full SHA for fb84190

Commits on Mar 16, 2023

• Update and standardize IVF indexes API (#1328)

  Show description for 9f2a64f
  viclafargue

  authored
  9f2a64f

  Copy full SHA for 9f2a64f

• Rename optional helper function (#1345)

  Show description for 8386807
  viclafargue

  authored
  8386807

  Copy full SHA for 8386807

Commits on Mar 17, 2023

• IVF-Flat Python wrappers (#1316)

  Show description for 7074010
  tfeher

  authored
  7074010

  Copy full SHA for 7074010

• Fix serialize documentation of ivf_flat (#1347)

  Show description for 4577cbe
  lowener

  authored
  4577cbe

  Copy full SHA for 4577cbe

• Remove MANIFEST.in use auto-generated one for sdists and package_data for wheels (#1348)

  Show description for 40f246d
  vyasr

  authored
  40f246d

  Copy full SHA for 40f246d

• Add dispatch based on compute architecture (#1335)

  Show description for e661035
  ahendriksen

  authored
  e661035

  Copy full SHA for e661035

• Replace faiss bfKnn (#1202)

  Show description for 1a3c028
  benfred

  authored
  1a3c028

  Copy full SHA for 1a3c028

Commits on Mar 18, 2023

• Remove faiss_mr.hpp (#1351)

  Show description for 2586f80
  benfred

  authored
  2586f80

  Copy full SHA for 2586f80

• Removing FAISS from build (#1340)

  Show description for c4671ed
  cjnolet

  authored
  c4671ed

  Copy full SHA for c4671ed

Commits on Mar 20, 2023

• Adding small readme image (#1354)

  Show description for 81fd1e0
  cjnolet

  authored
  81fd1e0

  Copy full SHA for 81fd1e0

• Consolidate pre-compiled specializations into single libraft binary (#1333)

  Show description for 97f8ad7
  cjnolet

  authored
  97f8ad7

  Copy full SHA for 97f8ad7

• Fix ivf flat specialization header IdxT from uint64_t -> int64_t (#1358)

  Show description for 56ac43a
  ahendriksen

  authored
  56ac43a

  Copy full SHA for 56ac43a

Commits on Mar 21, 2023

• Stop setting package version attribute in wheels (#1359)

  Show description for 05d899b
  vyasr

  authored
  05d899b

  Copy full SHA for 05d899b

Commits on Mar 22, 2023

• Remove usage of Dask's get_worker (#1365)

  Show description for a7e619c
  pentschev

  authored
  a7e619c

  Copy full SHA for a7e619c

Commits on Mar 23, 2023

• Reduce compile times of distance specializations (#1307)

  Show description for 08e7012
  ahendriksen

  authored
  08e7012

  Copy full SHA for 08e7012

• Add nccl to dependencies.yaml (#1361)

  Show description for 7c73f23
  benfred

  authored
  7c73f23

  Copy full SHA for 7c73f23

• Generate pyproject dependencies with dfg (#1364)

  Show description for 419f0c2
  vyasr

  authored
  419f0c2

  Copy full SHA for 419f0c2

• Python API for brute-force KNN (#1292)

  Show description for 31847af
  cjnolet

  authored
  31847af

  Copy full SHA for 31847af

• Relax UCX pin to allow 1.14 (#1366)

  Show description for 0df5cee
  pentschev

  authored
  0df5cee

  Copy full SHA for 0df5cee

Commits on Mar 24, 2023

• Adding architecture diagram to README.md (#1370)

  Show description for 1b18d1f
  cjnolet

  authored
  1b18d1f

  Copy full SHA for 1b18d1f

• Improve the performance of radix top-k (#1175)

  Show description for 8f1fa07
  yong-wang

  authored
  8f1fa07

  Copy full SHA for 8f1fa07

• RAFT skeleton project template (#1312)

  Show description for 9389108
  cjnolet

  authored
  9389108

  Copy full SHA for 9389108

• Adding some functions back in that seem to be a copy/paste error (#1373)

  Show description for f4c7f1f
  cjnolet

  authored
  f4c7f1f

  Copy full SHA for f4c7f1f

Commits on Mar 25, 2023

• Add extern template for ivfflat_interleaved_scan (#1360)

  Show description for 76c828d
  ahendriksen

  authored
  76c828d

  Copy full SHA for 76c828d

Commits on Mar 28, 2023

• add a distance epilogue function to the bfknn call (#1371)

  Show description for 0d3bd3d
  benfred

  authored
  0d3bd3d

  Copy full SHA for 0d3bd3d

• Add end-to-end CUDA ann-benchmarks to raft (#1304)

  Show description for 22ebc72
  cjnolet

  authored
  22ebc72

  Copy full SHA for 22ebc72

Commits on Mar 29, 2023

• Fixing index-url link on pip install docs (#1378)

  Show description for c2cb779
  cjnolet

  authored
  c2cb779

  Copy full SHA for c2cb779

• build.sh: Add option to log nvcc compile times (#1262)

  Show description for e963f5a
  ahendriksen

  authored
  e963f5a

  Copy full SHA for e963f5a

Commits on Mar 30, 2023

• Add Options to Generate Build Metrics Report (#1369)

  Show description for e456207
  divyegala

  authored
  e456207

  Copy full SHA for e456207

• Pin cupy in wheel tests to supported versions (#1383)

  Show description for e874921
  vyasr

  authored
  e874921

  Copy full SHA for e874921

Commits on Mar 31, 2023

• Fix primitives benchmarks (#1389)

  Show description for 09720d8
  ahendriksen

  authored
  09720d8

  Copy full SHA for 09720d8

• Use rapids-cmake new COMPONENT exporting feature (#1154)

  Show description for 7cd298d
  robertmaynard

  authored
  7cd298d

  Copy full SHA for 7cd298d

Commits on Apr 6, 2023

• Pin dask and distributed for release (#1399)

  Show description for 790dba6
  galipremsagar

  authored
  790dba6

  Copy full SHA for 790dba6

• CAGRA (#1375)

  Show description for a6e0e6b
  tfeher

  authored
  a6e0e6b

  Copy full SHA for a6e0e6b