raft
raft copied to clipboard
rapidsai
RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing hig...
**This is not ready for review.** Contributes to https://github.com/rapidsai/build-planning/issues/54. The `libraft-headers` and `libraft-headers-only` conda packages are bundling `rmm`'s headers. I believe that's because the `librmm` conda package isn't available in...
### Brief Add another workspace memory resource that does not have the explicit memory limit. That is, after the change we have the following: 1. `rmm::mr::get_current_device_resource()` is default for all...
Contributes to rapidsai/build-planning#15.
Benchmark CAGRA build + DiskANN (in memory) search on the graph.
This PR introduces the new vector addition feature to CAGRA. Rel: https://github.com/rapidsai/raft/issues/1775 CAGRA-Q is not supported ## Usage ```cpp auto additional_dataset = raft::make_host_matrix(res,updated_dataset_size, dim); raft::neighbors::cagra::extend(handle, raft::make_const_mdspan(additiona_dataset.view()), cagra_index); ``` ## Algorithm...
Looking through logs on https://github.com/rapidsai/raft/pull/2279, I noticed this warning: ```text test/test_comms.py:267 /__w/raft/raft/python/raft-dask/raft_dask/test/test_comms.py:267: PytestUnknownMarkWarning: Unknown pytest.mark.ucx - is this a typo? You can register custom marks to avoid this warning -...
## Description Contributes to: * https://github.com/rapidsai/build-planning/issues/54 * https://github.com/rapidsai/build-planning/issues/56 * https://github.com/rapidsai/rapids-cmake/issues/387 Testing with: * https://github.com/rapidsai/rapids-cmake/pull/592 * https://github.com/rapidsai/rmm/pull/1544 ## Notes for Reviewers Will put up a better description and revert testing-only changes...
- Link to `nvtx3` instead of deprecated `nvToolsExt` - Use `LAZY` cmake install message, to avoid flooding the console with all the headers name when they are left untouched.
