diff --git a/cpp/include/raft/neighbors/detail/refine_device.cuh b/cpp/include/raft/neighbors/detail/refine_device.cuh index 337318f791..5c9f1459e7 100644 --- a/cpp/include/raft/neighbors/detail/refine_device.cuh +++ b/cpp/include/raft/neighbors/detail/refine_device.cuh @@ -1,5 +1,5 @@ /* - * Copyright (c) 2023, NVIDIA CORPORATION. + * Copyright (c) 2023-2024, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/cpp/template/src/common.cuh b/cpp/template/src/common.cuh index 0b72d3bf3b..c2cb15bcf3 100644 --- a/cpp/template/src/common.cuh +++ b/cpp/template/src/common.cuh @@ -1,5 +1,5 @@ /* - * Copyright (c) 2023, NVIDIA CORPORATION. + * Copyright (c) 2023-2024, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -42,7 +42,7 @@ void generate_dataset(raft::device_resources const& dev_resources, 1.0f); } -// Copy the results to host and print a few samples +// Copy the results to host and print them template void print_results(raft::device_resources const& dev_resources, raft::device_matrix_view neighbors, @@ -61,7 +61,7 @@ void print_results(raft::device_resources const& dev_resources, // We need to sync the stream before accessing the data. raft::resource::sync_stream(dev_resources, stream); - for (int query_id = 0; query_id < 2; query_id++) { + for (int query_id = 0; query_id < neighbors.extent(0); query_id++) { std::cout << "Query " << query_id << " neighbor indices: "; raft::print_host_vector("", &neighbors_host(query_id, 0), topk, std::cout); std::cout << "Query " << query_id << " neighbor distances: ";