Skip to content

Commit 08f72ba

Browse files
authored
Merge branch 'main' into c_abi_checker
2 parents 1aec04d + 862259a commit 08f72ba

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • python/cuvs_bench/cuvs_bench/generate_groundtruth

python/cuvs_bench/cuvs_bench/generate_groundtruth/__main__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/usr/bin/env python
22
#
3-
# SPDX-FileCopyrightText: Copyright (c) 2024, NVIDIA CORPORATION.
3+
# SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION.
44
# SPDX-License-Identifier: Apache-2.0
55
#
66
import argparse
@@ -190,7 +190,7 @@ def calc_truth(dataset, queries, k, metric="sqeuclidean"):
190190
D, Ind = search(index, queries, k, resources=resources)
191191
resources.sync()
192192
else:
193-
D, Ind = cpu_search(X, queries, metric=metric)
193+
D, Ind = cpu_search(X, queries, k, metric=metric)
194194

195195
D, Ind = xp.asarray(D), xp.asarray(Ind)
196196
Ind += i # shift neighbor index by offset i

0 commit comments

Comments
 (0)