Skip to content

Commit 6f232ab

Browse files
committed
move batchsize from public header to internal
1 parent 95daa58 commit 6f232ab

4 files changed

Lines changed: 17 additions & 3 deletions

File tree

cpp/include/cuvs/neighbors/cagra.hpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3300,9 +3300,6 @@ void optimize(raft::resources const& handle,
33003300
raft::host_matrix_view<uint32_t, int64_t, raft::row_major> knn_graph,
33013301
raft::host_matrix_view<uint32_t, int64_t, raft::row_major> new_graph);
33023302

3303-
/** The batch size for the CAGRA optimize stage. */
3304-
constexpr static size_t kOptimizeBatchSize = 256 * 1024;
3305-
33063303
} // namespace helpers
33073304
} // namespace cagra
33083305
} // namespace neighbors

cpp/src/neighbors/detail/cagra/cagra_helpers.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
* SPDX-License-Identifier: Apache-2.0
44
*/
55

6+
#include "cagra_helpers.hpp"
7+
68
#include <algorithm>
79
#include <cstdint>
810
#include <cuvs/neighbors/cagra.hpp>
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
/*
2+
* SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION. All rights reserved.
3+
* SPDX-License-Identifier: Apache-2.0
4+
*/
5+
#pragma once
6+
7+
#include <cstddef>
8+
9+
namespace cuvs::neighbors::cagra::helpers {
10+
11+
/** The batch size for the CAGRA optimize stage. */
12+
constexpr static size_t kOptimizeBatchSize = 256 * 1024;
13+
14+
} // namespace cuvs::neighbors::cagra::helpers

cpp/src/neighbors/detail/cagra/graph_core.cuh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
*/
55
#pragma once
66

7+
#include "cagra_helpers.hpp"
78
#include "utils.hpp"
89

910
#include <raft/core/copy.cuh>

0 commit comments

Comments
 (0)