xpu
Loading...
Searching...
No Matches
xpu::block_reduce< T, BlockSize, Impl > Class Template Reference

#include <device.h>

Classes

struct  storage_t
 

Public Member Functions

template<typename ContextT >
XPU_D block_reduce (ContextT &ctx, storage_t &storage)
 
XPU_D block_reduce (tpos &, storage_t &)
 
XPU_D T sum (T input)
 
template<typename ReduceOp >
XPU_D T reduce (T input, ReduceOp reduce_op)
 

Constructor & Destructor Documentation

◆ block_reduce() [1/2]

template<typename T , int BlockSize, xpu::driver_t Impl = XPU_COMPILATION_TARGET>
template<typename ContextT >
XPU_D xpu::block_reduce< T, BlockSize, Impl >::block_reduce ( ContextT &  ctx,
storage_t &  storage 
)

◆ block_reduce() [2/2]

template<typename T , int BlockSize, xpu::driver_t Impl = XPU_COMPILATION_TARGET>
XPU_D xpu::block_reduce< T, BlockSize, Impl >::block_reduce ( tpos &  ,
storage_t &   
)

Member Function Documentation

◆ reduce()

template<typename T , int BlockSize, xpu::driver_t Impl = XPU_COMPILATION_TARGET>
template<typename ReduceOp >
XPU_D T xpu::block_reduce< T, BlockSize, Impl >::reduce ( T  input,
ReduceOp  reduce_op 
)

◆ sum()

template<typename T , int BlockSize, xpu::driver_t Impl = XPU_COMPILATION_TARGET>
XPU_D T xpu::block_reduce< T, BlockSize, Impl >::sum ( T  input)

The documentation for this class was generated from the following file: