Skip to content

Commit

Permalink
Update cpp/include/raft/neighbors/ivf_flat-ext.cuh
Browse files Browse the repository at this point in the history
Co-authored-by: Tamas Bela Feher <[email protected]>
  • Loading branch information
ahendriksen and tfeher committed Apr 18, 2023
1 parent 35ed4df commit 65c1cba
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cpp/include/raft/neighbors/ivf_flat-ext.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ void extend(raft::device_resources const& handle,
* index_params.add_data_on_build = false; // don't populate index on build
* index_params.kmeans_trainset_fraction = 1.0; // use whole dataset for kmeans training
* // train the index from a [N, D] dataset
* auto index_empty = ivf_flat::build(handle, dataset, index_params, dataset);
* auto index_empty = ivf_flat::build(handle, index_params, dataset);
* // fill the index with the data
* std::optional<raft::device_vector_view<const IdxT, IdxT>> no_op = std::nullopt;
* ivf_flat::extend(handle, dataset, no_opt, &index_empty);
Expand Down

0 comments on commit 65c1cba

Please sign in to comment.