Classes | Functions
raft_proto::detail Namespace Reference

Classes

struct  device_id
 
struct  device_id< device_type::cpu >
 
struct  device_id< device_type::gpu >
 
struct  device_setter
 
struct  device_setter< device_type::gpu >
 
struct  host_only_throw
 
struct  host_only_throw< T, true >
 
struct  non_owning_buffer
 
struct  owning_buffer
 
struct  owning_buffer< device_type::cpu, T >
 
struct  owning_buffer< device_type::gpu, T >
 

Functions

template<device_type dst_type, device_type src_type, typename T >
std::enable_if_t< std::conjunction_v< std::bool_constant< dst_type==device_type::cpu >, std::bool_constant< src_type==device_type::cpu > >, void > copy (T *dst, T const *src, uint32_t size, cuda_stream stream)
 
template<device_type dst_type, device_type src_type, typename T >
std::enable_if_t< std::conjunction_v< std::disjunction< std::bool_constant< dst_type !=device_type::cpu >, std::bool_constant< src_type !=device_type::cpu > >, std::bool_constant<!GPU_ENABLED > >, void > copy (T *dst, T const *src, uint32_t size, cuda_stream stream)
 
template<device_type dst_type, device_type src_type, typename T >
std::enable_if_t< std::conjunction_v< std::disjunction< std::bool_constant< dst_type==device_type::gpu >, std::bool_constant< src_type==device_type::gpu > >, std::bool_constant< GPU_ENABLED > >, void > copy (T *dst, T const *src, uint32_t size, cuda_stream stream)
 
template<device_type D, typename error_t >
void cuda_check (error_t const &err)
 
template<>
void cuda_check< device_type::gpu, cudaError_t > (cudaError_t const &err) noexcept(false)
 

Function Documentation

◆ copy() [1/3]

template<device_type dst_type, device_type src_type, typename T >
std::enable_if_t<std::conjunction_v<std::bool_constant<dst_type == device_type::cpu>, std::bool_constant<src_type == device_type::cpu> >, void> raft_proto::detail::copy ( T *  dst,
T const *  src,
uint32_t  size,
cuda_stream  stream 
)

◆ copy() [2/3]

template<device_type dst_type, device_type src_type, typename T >
std::enable_if_t< std::conjunction_v<std::disjunction<std::bool_constant<dst_type != device_type::cpu>, std::bool_constant<src_type != device_type::cpu> >, std::bool_constant<!GPU_ENABLED> >, void> raft_proto::detail::copy ( T *  dst,
T const *  src,
uint32_t  size,
cuda_stream  stream 
)

◆ copy() [3/3]

template<device_type dst_type, device_type src_type, typename T >
std::enable_if_t< std::conjunction_v<std::disjunction<std::bool_constant<dst_type == device_type::gpu>, std::bool_constant<src_type == device_type::gpu> >, std::bool_constant<GPU_ENABLED> >, void> raft_proto::detail::copy ( T *  dst,
T const *  src,
uint32_t  size,
cuda_stream  stream 
)

◆ cuda_check()

template<device_type D, typename error_t >
void raft_proto::detail::cuda_check ( error_t const &  err)

◆ cuda_check< device_type::gpu, cudaError_t >()

template<>
void raft_proto::detail::cuda_check< device_type::gpu, cudaError_t > ( cudaError_t const &  err)
inlinenoexcept