v24.02.00
🚨 Breaking Changes
- Switch to scikit-build-core (#2051) @vyasr
- Update to CCCL 2.2.0. (#2049) @bdice
- Update
raft-ann-bench
output filenames and add features to plotting (#2043) @divyegala - Remove selection_faiss (#2027) @benfred
🐛 Bug Fixes
- fix is_row/col_order for strided layouts (#2173) @mfoerste4
- Fix failing C++ tests and revert #2097, #2085. (#2168) @cjnolet
- Exclude tests from builds (#2162) @vyasr
- [HOTFIX] 24.02 Revert Random Sampling (#2144) @cjnolet
- Pin to pytest 7. (#2137) @bdice
- Conditionally include
hnsw
wrapper source in CMake (#2135) @divyegala - [BUG] Fix
SPMM
strided view (#2124) @lowener - Fixing small bug in CUSPARSE spmm w/ CUDA 12.2 (#2117) @cjnolet
- [BUG] Fix
num_cta_per_query
div (#2107) @lowener - Remove extraneous host pinnings from libraft-headers-only. (#2102) @bdice
- Remove unneeded CI symbol excludes (#2098) @robertmaynard
- Properly taking ownership of nccl subcomm (and destroying it) (#2094) @cjnolet
- Fix
max_queries
for CAGRA (#2081) @lowener - Fix compile failure on RTX 4090 (#2076) @JieFengWang
- Fix a crash in FAISS benchmark wrapper introduced in #2021 (#2062) @achirkin
- Correct function that wasn't returning a value (#2045) @robertmaynard
- Fixing small bug in raft-ann-bench (#2041) @cjnolet
- Make device_resources accessed from device_resources_manager thread-safe (#2030) @wphicks
- Fix ann-bench multithreading (#2021) @achirkin
- Fix
ci/checks/copyright.py
to mirror RAPIDS reference (#2008) @divyegala - Fix pyproject versions (#2002) @vyasr
📖 Documentation
- Adding license info for wiki-all dataset (#2129) @cjnolet
- [DOC] Documentation updates for release 24.02 (#2093) @cjnolet
- Fix errors with ingroup exposed by doxygen 1.10 (#2079) @wphicks
- Fix a typo (#2070) @narangvivek10
- Add usage example for brute_force::build (#2029) @benfred
- Add filtering to vector search tutorial (#1996) @lowener
🚀 New Features
- Update to use rapids-cmake for all deps (#2096) @robertmaynard
- Add IVF-PQ example into the template project (#2091) @achirkin
- Support for fp16 in CAGRA and IVF-PQ (#2085) @achirkin
- Add random subsampling for IVF methods (#2077) @tfeher
- Update
raft-ann-bench
output filenames and add features to plotting (#2043) @divyegala - Add brute_force index serialization (#2036) @wphicks
- Add eps-neighbor search via RBC (#2028) @mfoerste4
libraft
andpylibraft
API for CAGRA build and HNSW search (#2022) @divyegala- Export Pareto frontier in
raft-ann-bench.data_export
(#2009) @divyegala - Implement maybe-owning multi-dimensional container (mdbuffer) (#1999) @wphicks
- Add support for 1024+ dim vectors in CAGRA search (#1994) @enp1s0
- Replace GEMM backend: cublas.gemm -> cublaslt.matmul (#1736) @achirkin
🛠️ Improvements
- Remove get_mem_info functions from RAFT custom memory resources (#2108) @harrism
- Replace call to mr::get_mem_info() (#2099) @harrism
- Allow topk larger than 1024 in CAGRA (#2097) @benfred
- Remove usages of rapids-env-update (#2095) @KyleFromNVIDIA
- Provide explicit pool size for pool_memory_resources and clean up includes (#2088) @harrism
- refactor CUDA versions in dependencies.yaml (#2086) @jameslamb
- ANN bench fix latency measurement overhead (#2084) @tfeher
- Remove hardcoded limit in
print_results
function (#2080) @narangvivek10 - [FEA] Add support for SDDMM by wrapping the cusparseSDDMM (#2067) (#2067) @rhdong
- Benchmark brute force knn (#2063) @benfred
- [BUG] fix empty initialization of device_ndarray in pylibraft (#2061) @mfoerste4
- Improve parallelism of refine host (#2059) @anaruse
- Subsampling for IVF-PQ codebook generation (#2052) @abc99lr
- Switch to scikit-build-core (#2051) @vyasr
- Update to CCCL 2.2.0. (#2049) @bdice
- Use cuda::proclaim_return_type on device lambda. (#2048) @bdice
- Removing code that explicitly compares equality of rmm memory resources (#2047) @cjnolet
- Add public enum for select-k algorithm selection (#2046) @benfred
- Update dependencies.yaml to new pip index (#2042) @vyasr
- Remove RAFT_BUILD_WHEELS and standardize Python builds (#2040) @vyasr
- Fix ucx-py version pinning in dependencies.yaml. (#2035) @bdice
- [REVIEW] Fix typos in parameter tuning guide (#2034) @abc99lr
- Add AIR-Top-k reference (#2031) @tfeher
- Remove selection_faiss (#2027) @benfred
- Fixing json parse error in
raft-ann-bench.data_export
(#2025) @cjnolet - Updating cagra build constraint (#2016) @cjnolet
- Update to fmt 10.1.1 and spdlog 1.12.0. (#1957) @bdice
- Enable host dataset for IVF-Flat (#1635) @tfeher
- add half/bfloat support to myInf and abs (#1592) @Kh4ster