Mathematical Functions

The math functions APIs guarantee both CUDA and CPU compatibility, making it more straightforward to write __host__ __device__ functions without being concerned whether the underlying intrinsics will build and work.

#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>