rocPRIM
Classes | Public Types | Public Member Functions | List of all members
detail::warp_reduce_shared_mem< T, WarpSize, UseAllReduce > Class Template Reference

Public Types

using storage_type = detail::raw_storage< storage_type_ >
 

Public Member Functions

template<class BinaryFunction >
ROCPRIM_DEVICE ROCPRIM_INLINE void reduce (T input, T &output, storage_type &storage, BinaryFunction reduce_op)
 
template<class BinaryFunction >
ROCPRIM_DEVICE ROCPRIM_INLINE void reduce (T input, T &output, unsigned int valid_items, storage_type &storage, BinaryFunction reduce_op)
 
template<class Flag , class BinaryFunction >
ROCPRIM_DEVICE ROCPRIM_INLINE void head_segmented_reduce (T input, T &output, Flag flag, storage_type &storage, BinaryFunction reduce_op)
 
template<class Flag , class BinaryFunction >
ROCPRIM_DEVICE ROCPRIM_INLINE void tail_segmented_reduce (T input, T &output, Flag flag, storage_type &storage, BinaryFunction reduce_op)
 

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