Code

[RELEASE] raft v24.02#2134

Code

Merged

raydouglass merged 81 commits intomainrapidsai/raft:mainfrom branch-24.02rapidsai/raft:branch-24.02Copy head branch name to clipboardFeb 12, 2024

+14,001-5,201

Commits

Commits on Nov 14, 2023

• v24.02 Updates [skip ci]

  raydouglass

  committed
  06979df

  Copy full SHA for 06979df

• Merge pull request #1992 from rapidsai/branch-23.12

  Show description for e875e04
  GPUtester

  authored
  e875e04

  Copy full SHA for e875e04

Commits on Nov 15, 2023

• Merge pull request #2001 from rapidsai/branch-23.12

  Show description for 8dd6a79
  GPUtester

  authored
  8dd6a79

  Copy full SHA for 8dd6a79

Commits on Nov 16, 2023

• Fix dynamic version in pyproject.toml files (#2002)

  Show description for d67c3b2
  vyasr

  authored
  d67c3b2

  Copy full SHA for d67c3b2

• Merge pull request #2004 from rapidsai/branch-23.12

  Show description for 5e6e50a
  GPUtester

  authored
  5e6e50a

  Copy full SHA for 5e6e50a

• Merge pull request #2007 from rapidsai/branch-23.12

  Show description for 7754040
  GPUtester

  authored
  7754040

  Copy full SHA for 7754040

Commits on Nov 17, 2023

• Merge pull request #2010 from rapidsai/branch-23.12

  Show description for fd1ac0c
  GPUtester

  authored
  fd1ac0c

  Copy full SHA for fd1ac0c

• Merge pull request #2012 from rapidsai/branch-23.12

  Show description for b5b5202
  GPUtester

  authored
  b5b5202

  Copy full SHA for b5b5202

• Fix ci/checks/copyright.py to mirror RAPIDS reference (#2008)

  Show description for 7e307b9
  divyegala

  authored
  7e307b9

  Copy full SHA for 7e307b9

• Merge pull request #2014 from rapidsai/branch-23.12

  Show description for bf16c50
  GPUtester

  authored
  bf16c50

  Copy full SHA for bf16c50

Commits on Nov 18, 2023

• Export Pareto frontier in raft-ann-bench.data_export (#2009)

  Show description for bdc5117
  divyegala

  authored
  bdc5117

  Copy full SHA for bdc5117

• Merge pull request #2015 from rapidsai/branch-23.12

  Show description for a03749b
  GPUtester

  authored
  a03749b

  Copy full SHA for a03749b

Commits on Nov 21, 2023

• Merge pull request #2019 from rapidsai/branch-23.12

  Show description for ea8c3ed
  GPUtester

  authored
  ea8c3ed

  Copy full SHA for ea8c3ed

Commits on Nov 28, 2023

• Enable host dataset for IVF-Flat (#1635)

  Show description for 88e9a55
  tfeher

  authored
  88e9a55

  Copy full SHA for 88e9a55

• Updating cagra build constraint (#2016)

  Show description for 736a8bf
  cjnolet

  authored
  736a8bf

  Copy full SHA for 736a8bf

Commits on Nov 29, 2023

• Fixing json parse error in raft-ann-bench.data_export (#2025)

  Show description for ed272c1
  cjnolet

  authored
  ed272c1

  Copy full SHA for ed272c1

Commits on Nov 30, 2023

• Add usage example for brute_force::build (#2029)

  Show description for 04fa426
  benfred

  authored
  04fa426

  Copy full SHA for 04fa426

• Update to fmt 10.1.1 and spdlog 1.12.0. (#1957)

  Show description for ff60d3a
  bdice

  authored
  ff60d3a

  Copy full SHA for ff60d3a

• Remove selection_faiss (#2027)

  Show description for 4ba0139
  benfred

  authored
  4ba0139

  Copy full SHA for 4ba0139

Commits on Dec 1, 2023

• Add filtering to vector search tutorial (#1996)

  Show description for b8c026b
  lowener

  authored
  b8c026b

  Copy full SHA for b8c026b

• Fix ucx-py version pinning in dependencies.yaml. (#2035)

  Show description for 712ba0e
  bdice

  authored
  712ba0e

  Copy full SHA for 712ba0e

Commits on Dec 4, 2023

• Merge pull request #2038 from rapidsai/branch-23.12

  Show description for 8e1c62c
  GPUtester

  authored
  8e1c62c

  Copy full SHA for 8e1c62c

Commits on Dec 5, 2023

• Make device_resources accessed from device_resources_manager thread-safe (#2030)

  Show description for 42e9f15
  wphicks

  authored
  42e9f15

  Copy full SHA for 42e9f15

Commits on Dec 6, 2023

• Fixing small bug in raft-ann-bench (#2041)

  Show description for c3a9c78
  cjnolet

  authored
  c3a9c78

  Copy full SHA for c3a9c78

• [REVIEW] Fix typos in parameter tuning guide (#2034)

  Show description for ecd292b
  abc99lr

  authored
  ecd292b

  Copy full SHA for ecd292b

• Merge pull request #2044 from rapidsai/branch-23.12

  Show description for b333b74
  GPUtester

  authored
  b333b74

  Copy full SHA for b333b74

• Remove RAFT_BUILD_WHEELS and standardize Python builds (#2040)

  Show description for 339878d
  vyasr

  authored
  339878d

  Copy full SHA for 339878d

Commits on Dec 7, 2023

• Update dependencies.yaml to new pip index (#2042)

  Show description for 9ce20bf
  vyasr

  authored
  9ce20bf

  Copy full SHA for 9ce20bf

• Correct function that wasn't returning a value (#2045)

  Show description for bc35f2b
  robertmaynard

  authored
  bc35f2b

  Copy full SHA for bc35f2b

• Add brute_force index serialization (#2036)

  Show description for d2210a2
  wphicks

  authored
  d2210a2

  Copy full SHA for d2210a2

Commits on Dec 8, 2023

• Add support for 1024+ dim vectors in CAGRA search (#1994)

  Show description for e999100
  enp1s0

  authored
  e999100

  Copy full SHA for e999100

• Use cuda::proclaim_return_type on device lambda. (#2048)

  Show description for 5e80c1d
  bdice

  authored
  5e80c1d

  Copy full SHA for 5e80c1d

Commits on Dec 9, 2023

• Removing code that explicitly compares equality of rmm memory resources (#2047)

  Show description for addb059
  cjnolet

  authored
  addb059

  Copy full SHA for addb059

Commits on Dec 13, 2023

• Fix ann-bench multithreading (#2021)

  Show description for d9a7290
  achirkin

  authored
  d9a7290

  Copy full SHA for d9a7290

• Fix a crash in FAISS benchmark wrapper introduced in #2021 (#2062)

  Show description for 6c95f9c
  achirkin

  authored
  6c95f9c

  Copy full SHA for 6c95f9c

• Switch to scikit-build-core (#2051)

  Show description for 80a48ca
  vyasr

  authored
  80a48ca

  Copy full SHA for 80a48ca

Commits on Dec 15, 2023

• [BUG] fix empty initialization of device_ndarray in pylibraft (#2061)

  Show description for 1beb556
  mfoerste4

  authored
  1beb556

  Copy full SHA for 1beb556

Commits on Dec 19, 2023

• Update to CCCL 2.2.0. (#2049)

  Show description for 2962169
  bdice

  authored
  2962169

  Copy full SHA for 2962169

Commits on Dec 20, 2023

• Benchmark brute force knn (#2063)

  Show description for 7e098b2
  benfred

  authored
  7e098b2

  Copy full SHA for 7e098b2

Commits on Dec 21, 2023

• Update raft-ann-bench output filenames and add features to plotting (#2043)

  Show description for bae049b
  divyegala

  authored
  bae049b

  Copy full SHA for bae049b

Commits on Jan 2, 2024

• Fix a typo (#2070)

  Show description for dc442d8
  narangvivek10

  authored
  dc442d8

  Copy full SHA for dc442d8

Commits on Jan 3, 2024

• Fix errors with ingroup exposed by doxygen 1.10 (#2079)

  Show description for 6d3572b
  wphicks

  authored
  6d3572b

  Copy full SHA for 6d3572b

Commits on Jan 4, 2024

• Implement maybe-owning multi-dimensional container (mdbuffer) (#1999)

  Show description for 0d65954
  wphicks

  authored
  0d65954

  Copy full SHA for 0d65954

• Remove hardcoded limit in print_results function (#2080)

  Show description for 6762fe5
  narangvivek10

  authored
  6762fe5

  Copy full SHA for 6762fe5

Commits on Jan 9, 2024

• Improve parallelism of refine host (#2059)

  Show description for 3b88d17
  anaruse

  authored
  3b88d17

  Copy full SHA for 3b88d17

• Fix max_queries for CAGRA (#2081)

  Show description for 1484a03
  lowener

  authored
  1484a03

  Copy full SHA for 1484a03

Commits on Jan 10, 2024

• Add public enum for select-k algorithm selection (#2046)

  Show description for 26d310b
  benfred

  authored
  26d310b

  Copy full SHA for 26d310b

Commits on Jan 11, 2024

• refactor CUDA versions in dependencies.yaml (#2086)

  Show description for 93a504e
  jameslamb

  authored
  93a504e

  Copy full SHA for 93a504e

• Add IVF-PQ example into the template project (#2091)

  Show description for 856288a
  achirkin

  authored
  856288a

  Copy full SHA for 856288a

Commits on Jan 12, 2024

• Properly taking ownership of nccl subcomm (and destroying it) (#2094)

  Show description for 7d5bb3c
  cjnolet

  authored
  7d5bb3c

  Copy full SHA for 7d5bb3c

• Remove usages of rapids-env-update (#2095)

  Show description for 0c75bb1
  KyleFromNVIDIA

  authored
  0c75bb1

  Copy full SHA for 0c75bb1

Commits on Jan 16, 2024

• Add AIR-Top-k reference (#2031)

  Show description for 1d9adab
  tfeher

  authored
  1d9adab

  Copy full SHA for 1d9adab

• [FEA] Add support for SDDMM by wrapping the cusparseSDDMM (#2067) (#2067)

  Show description for 3c7586f
  rhdong

  authored
  3c7586f

  Copy full SHA for 3c7586f

Commits on Jan 17, 2024

• Update to use rapids-cmake for all deps (#2096)

  Show description for b75fbd0
  robertmaynard

  authored
  b75fbd0

  Copy full SHA for b75fbd0

• Provide explicit pool size for pool_memory_resources and clean up includes (#2088)

  Show description for dbc33ea
  harrism

  authored
  dbc33ea

  Copy full SHA for dbc33ea

• [DOC] Documentation updates for release 24.02 (#2093)

  Show description for fafb8ed
  cjnolet

  authored
  fafb8ed

  Copy full SHA for fafb8ed

• Fix compile failure on RTX 4090 (#2076)

  Show description for 1e4961e
  JieFengWang

  authored
  1e4961e

  Copy full SHA for 1e4961e

Commits on Jan 18, 2024

• Replace call to mr::get_mem_info() (#2099)

  Show description for 7cab0c3
  harrism

  authored
  7cab0c3

  Copy full SHA for 7cab0c3

Commits on Jan 19, 2024

• Support for fp16 in CAGRA and IVF-PQ (#2085)

  Show description for 72f48ae
  achirkin

  authored
  72f48ae

  Copy full SHA for 72f48ae

• Remove unneeded CI symbol excludes (#2098)

  Show description for db8e21f
  robertmaynard

  authored
  db8e21f

  Copy full SHA for db8e21f

• Remove extraneous host pinnings from libraft-headers-only. (#2102)

  Show description for d7eebc6
  bdice

  authored
  d7eebc6

  Copy full SHA for d7eebc6

Commits on Jan 23, 2024

• Add random subsampling for IVF methods (#2077)

  Show description for 9c35f73
  tfeher

  authored
  9c35f73

  Copy full SHA for 9c35f73

• Replace GEMM backend: cublas.gemm -> cublaslt.matmul (#1736)

  Show description for 558dc8f
  achirkin

  authored
  558dc8f

  Copy full SHA for 558dc8f

• [BUG] Fix num_cta_per_query div (#2107)

  Show description for d89ab1b
  lowener

  authored
  d89ab1b

  Copy full SHA for d89ab1b

• Allow topk larger than 1024 in CAGRA (#2097)

  Show description for 0586fc3
  benfred

  authored
  0586fc3

  Copy full SHA for 0586fc3

Commits on Jan 24, 2024

• Fixing small bug in CUSPARSE spmm w/ CUDA 12.2 (#2117)

  Show description for 70e806a
  cjnolet

  authored
  70e806a

  Copy full SHA for 70e806a

• Add eps-neighbor search via RBC (#2028)

  Show description for ac2d1ae
  mfoerste4

  authored
  ac2d1ae

  Copy full SHA for ac2d1ae

• Remove get_mem_info functions from RAFT custom memory resources (#2108)

  Show description for 71fce1c
  harrism

  authored
  71fce1c

  Copy full SHA for 71fce1c

• add half/bfloat support to myInf and abs (#1592)

  Show description for dbb5e66
  Kh4ster

  authored
  dbb5e66

  Copy full SHA for dbb5e66

• [BUG] Fix SPMM strided view (#2124)

  Show description for 8743ac9
  lowener

  authored
  8743ac9

  Copy full SHA for 8743ac9

• ANN bench fix latency measurement overhead (#2084)

  Show description for 3ce00d3
  tfeher

  authored
  3ce00d3

  Copy full SHA for 3ce00d3

Commits on Jan 25, 2024

• Adding license info for wiki-all dataset (#2129)

  Show description for dbf3f74
  cjnolet

  authored
  dbf3f74

  Copy full SHA for dbf3f74

• Subsampling for IVF-PQ codebook generation (#2052)

  Show description for e272176
  abc99lr

  authored
  e272176

  Copy full SHA for e272176

Commits on Jan 26, 2024

• libraft and pylibraft API for CAGRA build and HNSW search (#2022)

  Show description for 2822532
  divyegala

  authored
  2822532

  Copy full SHA for 2822532

Commits on Jan 29, 2024

• conditionally include hnsw source (#2135)

  Show description for f876491
  divyegala

  authored
  f876491

  Copy full SHA for f876491

Commits on Jan 31, 2024

• Pin to pytest 7. (#2137)

  Show description for c70d17a
  bdice

  authored
  c70d17a

  Copy full SHA for c70d17a

Commits on Feb 1, 2024

• [HOTFIX] 24.02 Revert Random Sampling (#2144)

  Show description for 7edd372
  cjnolet

  authored
  7edd372

  Copy full SHA for 7edd372

Commits on Feb 7, 2024

• Exclude tests from builds (#2162)

  vyasr

  authored
  5bd12a5

  Copy full SHA for 5bd12a5

Commits on Feb 9, 2024

• Fix failing C++ tests and revert #2097, #2085. (#2168)

  Show description for 9446430
  cjnolet

  authored
  9446430

  Copy full SHA for 9446430

Commits on Feb 12, 2024

• fix is_row/col_order for strided layouts (#2173)

  Show description for dc75590
  mfoerste4

  authored
  dc75590

  Copy full SHA for dc75590

• Update Changelog [skip ci]

  raydouglass

  committed
  04a9c95

  Copy full SHA for 04a9c95