|
7 | 7 |
|
8 | 8 | #include "common.hpp" |
9 | 9 | #include <cstdint> |
| 10 | +#include <cuvs/core/export.hpp> |
10 | 11 | #include <cuvs/distance/distance.hpp> |
11 | 12 | #include <cuvs/neighbors/common.hpp> |
12 | 13 | #include <raft/core/device_mdarray.hpp> |
|
15 | 16 | #include <raft/core/mdspan.hpp> |
16 | 17 | #include <raft/util/integer_utils.hpp> |
17 | 18 |
|
18 | | -namespace cuvs::neighbors::ivf_sq { |
| 19 | +namespace CUVS_EXPORT cuvs { |
| 20 | +namespace neighbors { |
| 21 | +namespace ivf_sq { |
19 | 22 |
|
20 | 23 | /** |
21 | 24 | * @defgroup ivf_sq_cpp_index_params IVF-SQ index build parameters |
@@ -86,7 +89,7 @@ static_assert(std::is_aggregate_v<search_params>); |
86 | 89 | */ |
87 | 90 |
|
88 | 91 | template <typename SizeT, typename CodeT, typename IdxT> |
89 | | -struct list_spec { |
| 92 | +struct CUVS_EXPORT list_spec { |
90 | 93 | static_assert(std::is_same_v<CodeT, uint8_t>, "IVF-SQ code type CodeT must be uint8_t"); |
91 | 94 |
|
92 | 95 | using value_type = CodeT; |
@@ -166,7 +169,7 @@ using list_data = ivf::list<list_spec, SizeT, CodeT, IdxT>; |
166 | 169 | * |
167 | 170 | */ |
168 | 171 | template <typename CodeT> |
169 | | -struct index : cuvs::neighbors::index { |
| 172 | +struct CUVS_EXPORT index : cuvs::neighbors::index { |
170 | 173 | static_assert(std::is_same_v<CodeT, uint8_t>, "IVF-SQ code type CodeT must be uint8_t for now."); |
171 | 174 |
|
172 | 175 | using index_params_type = ivf_sq::index_params; |
@@ -625,4 +628,6 @@ void deserialize(raft::resources const& handle, |
625 | 628 | * @} |
626 | 629 | */ |
627 | 630 |
|
628 | | -} // namespace cuvs::neighbors::ivf_sq |
| 631 | +} // namespace ivf_sq |
| 632 | +} // namespace neighbors |
| 633 | +} // namespace CUVS_EXPORT cuvs |
0 commit comments