Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add IVF-Flat C++ example #1828

Merged
merged 13 commits into from
Sep 26, 2023
Merged

Add IVF-Flat C++ example #1828

Changes from 1 commit
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
074353e
Add IVF-Flat C++ example
tfeher Sep 18, 2023
c44d751
Fix dataset indices, Sync and print results
tfeher Sep 18, 2023
f126e51
Fix typos
tfeher Sep 18, 2023
5efa8db
Merge branch 'branch-23.10' into ivf_flat_cpp_example
tfeher Sep 18, 2023
d7e94db
Merge branch 'branch-23.10' into ivf_flat_cpp_example
tfeher Sep 19, 2023
1f3e827
Rename CAGRA example, and refactor it
tfeher Sep 21, 2023
c5c723e
Merge remote-tracking branch 'origin/branch-23.10' into ivf_flat_cpp_…
tfeher Sep 21, 2023
f683a85
Merge branch 'branch-23.10' into ivf_flat_cpp_example
tfeher Sep 22, 2023
e9c5b34
Remove RAFT_ENABLE_EXPLICIT_INSTANTIATE_ONLY flag
tfeher Sep 22, 2023
8e0742e
Merge remote-tracking branch 'origin/branch-23.10' into ivf_flat_cpp_…
tfeher Sep 22, 2023
d680cf0
Merge branch 'branch-23.10' into ivf_flat_cpp_example
tfeher Sep 23, 2023
95e5402
Merge branch 'branch-23.10' into ivf_flat_cpp_example
tfeher Sep 25, 2023
f7649d0
Merge branch 'branch-23.10' into ivf_flat_cpp_example
tfeher Sep 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix typos
  • Loading branch information
@tfeher
tfeher committed Sep 18, 2023
commit f126e51636bf013e1a06fdcb1fb9b73ae8d528a4
6 changes: 3 additions & 3 deletions cpp/template/src/ivf_flat_example.cu
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ void print_results(raft::device_resources const& dev_resources,
raft::copy(neighbors_host.data_handle(), neighbors.data_handle(), neighbors.size(), stream);
raft::copy(distances_host.data_handle(), distances.data_handle(), distances.size(), stream);

// The calls to ivf_flat::search and raft::copy is asyncronous.
// The calls to ivf_flat::search and raft::copy is asynchronous.
// We need to sync the stream before accessing the data.
raft::resource::sync_stream(dev_resources, stream);

Expand Down Expand Up @@ -86,7 +86,7 @@ void ivf_flat_build_search_simple(raft::device_resources const& dev_resources,
ivf_flat::search(
dev_resources, search_params, index, queries, neighbors.view(), distances.view());

// The call to ivf_flat::search is asyncronous. Before accessing the data, sync by calling
// The call to ivf_flat::search is asynchronous. Before accessing the data, sync by calling
// raft::resource::sync_stream(dev_resources);

print_results(dev_resources, neighbors.view(), distances.view());
Expand Down Expand Up @@ -168,7 +168,7 @@ void ivf_flat_build_extend_search(raft::device_resources const& dev_resources,
ivf_flat::search(
dev_resources, search_params, index, queries, neighbors.view(), distances.view());

// The call to ivf_flat::search is asyncronous. Before accessing the data, sync using:
// The call to ivf_flat::search is asynchronous. Before accessing the data, sync using:
// raft::resource::sync_stream(dev_resources);

print_results(dev_resources, neighbors.view(), distances.view());
Expand Down