注意

RAFT 中的向量搜索和聚类算法正在迁移到一个新的专门用于向量搜索的库，称为 cuVS。在此迁移期间，我们将继续支持 RAFT 中的向量搜索算法，但在 RAPIDS 24.06（六月）版本之后将不再更新它们。我们计划在 RAPIDS 24.10（十月）版本之前完成迁移，并在 24.12（十二月）版本中将它们从 RAFT 中完全移除。

数学函数

数学函数 API 保证兼容 CUDA 和 CPU，这使得编写 __host__ __device__ 函数更加直接，无需担心底层内联函数是否能构建和工作。

#include <raft/core/math.hpp>

namespace raft::core

template<typename T>
RAFT_INLINE_FUNCTION auto abs(T x) -> std::enable_if_t<std::is_same_v<float, T> || std::is_same_v<double, T> || std::is_same_v<int, T> || std::is_same_v<long int, T> || std::is_same_v<long long int, T>, T>