muda::DeviceSegmentedSort Class Reference

Inheritance diagram for muda::DeviceSegmentedSort:

Public Member Functions
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeys (const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeysDescending (const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeys (cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeysDescending (cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeys (const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeysDescending (const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeys (cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeysDescending (cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairs (const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairsDescending (const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairs (cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairsDescending (cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairs (const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairsDescending (const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairs (cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairsDescending (cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeys (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeysDescending (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeys (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortKeysDescending (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeys (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeysDescending (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeys (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortKeysDescending (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairs (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairsDescending (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairs (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	SortPairsDescending (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairs (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairsDescending (void *d_temp_storage, size_t &temp_storage_bytes, const KeyT *d_keys_in, KeyT *d_keys_out, const ValueT *d_values_in, ValueT *d_values_out, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairs (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
template<typename KeyT , typename ValueT , typename BeginOffsetIteratorT , typename EndOffsetIteratorT >
DeviceSegmentedSort &	StableSortPairsDescending (void *d_temp_storage, size_t &temp_storage_bytes, cub::DoubleBuffer< KeyT > &d_keys, cub::DoubleBuffer< ValueT > &d_values, int num_items, int num_segments, BeginOffsetIteratorT d_begin_offsets, EndOffsetIteratorT d_end_offsets)
Public Member Functions inherited from muda::CubWrapper< DeviceSegmentedSort >
	CubWrapper (Stream &stream=Stream::Default())
void	kernel_name (std::string_view)=delete
Public Member Functions inherited from muda::LaunchBase< T >
MUDA_GENERIC	LaunchBase (::cudaStream_t stream) MUDA_NOEXCEPT
T &	push_range (const std::string &name)
T &	pop_range ()
T &	kernel_name (std::string_view name)
T &	file_line (std::string_view file, int line)
T &	record (cudaEvent_t e, int flag=cudaEventRecordDefault)
T &	record (ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase * > &vars)
template<typename... ViewT>
T &	record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
T &	when (cudaEvent_t e, int flag=cudaEventWaitDefault)
T &	wait (cudaEvent_t e, int flag=cudaEventWaitDefault)
T &	wait (const ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase * > &vars)
template<typename... ViewT>
T &	wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
T &	wait ()
T &	callback (const std::function< void(::cudaStream_t, ::cudaError)> &callback)
template<typename Next >
Next	next (Next n)
template<typename Next , typename... Args>
Next	next (Args &&... args)
template<typename T >
MUDA_GENERIC	LaunchBase (cudaStream_t stream) MUDA_NOEXCEPT
Public Member Functions inherited from muda::LaunchCore
MUDA_GENERIC	LaunchCore (::cudaStream_t stream) MUDA_NOEXCEPT
void	init_stream (::cudaStream_t s)
void	push_range (const std::string &name)
void	pop_range ()
void	record (cudaEvent_t e, int flag=cudaEventRecordDefault)
void	record (ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase * > &vars)
template<typename... ViewT>
void	record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
void	when (cudaEvent_t e, int flag=cudaEventWaitDefault)
void	wait (cudaEvent_t e, int flag=cudaEventWaitDefault)
void	wait (const ComputeGraphVar< cudaEvent_t > &e, const std::vector< ComputeGraphVarBase * > &vars)
template<typename... ViewT>
void	wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
void	wait ()
void	callback (const std::function< void(::cudaStream_t, ::cudaError)> &callback)
template<typename... ViewT>
MUDA_INLINE void	record (ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)
template<typename... ViewT>
MUDA_INLINE void	wait (const ComputeGraphVar< cudaEvent_t > &e, ComputeGraphVar< ViewT > &... vars)

Additional Inherited Members
Public Types inherited from muda::LaunchBase< T >
using	derived_type = T
Static Public Member Functions inherited from muda::LaunchCore
static void	kernel_name (std::string_view name)
static void	file_line (std::string_view file, int line)
static void	wait_event (cudaEvent_t event)
static void	wait_stream (::cudaStream_t stream)
static void	wait_device ()
Public Attributes inherited from muda::CubWrapper< DeviceSegmentedSort >
Stream *	m_muda_stream
Protected Types inherited from muda::LaunchCore
template<typename T >
using	S = std::shared_ptr< T >
Protected Member Functions inherited from muda::CubWrapper< DeviceSegmentedSort >
std::byte *	prepare_buffer (size_t reqSize)
Protected Member Functions inherited from muda::LaunchBase< T >
T &	pop_kernel_label ()
Protected Member Functions inherited from muda::LaunchCore
MUDA_GENERIC::cudaStream_t	stream () const
MUDA_HOST void	pop_kernel_label ()
Protected Attributes inherited from muda::LaunchCore
::cudaStream_t	m_stream

---

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

• src/muda/cub/device/device_segmented_sort.h