Class muda::DevicePartition
ClassList > muda > DevicePartition
Inherits the following classes: muda::CubWrapper
Public Types inherited from muda::LaunchBase
See muda::LaunchBase
Type | Name |
---|---|
typedef T | derived_type |
Public Attributes inherited from muda::CubWrapper
See muda::CubWrapper
Type | Name |
---|---|
Stream * | m_muda_stream = = nullptr |
Public Functions
Type | Name |
---|---|
DevicePartition & | Flagged (InputIteratorT d_in, FlagIterator d_flags, OutputIteratorT d_out, NumSelectedIteratorT d_num_selected_out, int num_items) |
DevicePartition & | Flagged (void * d_temp_storage, size_t & temp_storage_bytes, InputIteratorT d_in, FlagIterator d_flags, OutputIteratorT d_out, NumSelectedIteratorT d_num_selected_out, int num_items) |
DevicePartition & | If (InputIteratorT d_in, OutputIteratorT d_out, NumSelectedIteratorT d_num_selected_out, int num_items, SelectOp select_op) |
DevicePartition & | If (InputIteratorT d_in, FirstOutputIteratorT d_first_part_out, SecondOutputIteratorT d_second_part_out, UnselectedOutputIteratorT d_unselected_out, NumSelectedIteratorT d_num_selected_out, int num_items, SelectFirstPartOp select_first_part_op, SelectSecondPartOp select_second_part_op) |
DevicePartition & | If (void * d_temp_storage, size_t & temp_storage_bytes, InputIteratorT d_in, OutputIteratorT d_out, NumSelectedIteratorT d_num_selected_out, int num_items, SelectOp select_op) |
DevicePartition & | If (void * d_temp_storage, size_t & temp_storage_bytes, InputIteratorT d_in, FirstOutputIteratorT d_first_part_out, SecondOutputIteratorT d_second_part_out, UnselectedOutputIteratorT d_unselected_out, NumSelectedIteratorT d_num_selected_out, int num_items, SelectFirstPartOp select_first_part_op, SelectSecondPartOp select_second_part_op) |
Public Functions inherited from muda::CubWrapper
See muda::CubWrapper
Type | Name |
---|---|
CubWrapper (Stream & stream=Stream::Default()) |
|
void | kernel_name (std::string_view) = delete |
Public Functions inherited from muda::LaunchBase
See muda::LaunchBase
Type | Name |
---|---|
MUDA_GENERIC | LaunchBase (::cudaStream_t stream) |
T & | callback (const std::function< void(::cudaStream_t, ::cudaError)> & callback) |
T & | file_line (std::string_view file, int line) |
T & | kernel_name (std::string_view name) |
Next | next (Next n) |
Next | next (Args &&... args) |
T & | pop_range () |
T & | push_range (const std::string & name) |
T & | record (cudaEvent_t e, int flag=cudaEventRecordDefault) |
T & | record (ComputeGraphVar< cudaEvent_t > & e, const std::vector< ComputeGraphVarBase * > & vars) |
T & | record (ComputeGraphVar< cudaEvent_t > & e, ComputeGraphVar< ViewT > &... vars) |
T & | wait (cudaEvent_t e, int flag=cudaEventWaitDefault) |
T & | wait (const ComputeGraphVar< cudaEvent_t > & e, const std::vector< ComputeGraphVarBase * > & vars) |
T & | wait (const ComputeGraphVar< cudaEvent_t > & e, ComputeGraphVar< ViewT > &... vars) |
T & | wait () |
T & | when (cudaEvent_t e, int flag=cudaEventWaitDefault) |
~LaunchBase () |
Public Functions inherited from muda::LaunchCore
See muda::LaunchCore
Type | Name |
---|---|
MUDA_GENERIC | LaunchCore (::cudaStream_t stream) |
void | callback (const std::function< void(::cudaStream_t, ::cudaError)> & callback) |
void | init_stream (::cudaStream_t s) |
void | pop_range () |
void | push_range (const std::string & name) |
void | record (cudaEvent_t e, int flag=cudaEventRecordDefault) |
void | record (ComputeGraphVar< cudaEvent_t > & e, const std::vector< ComputeGraphVarBase * > & vars) |
void | record (ComputeGraphVar< cudaEvent_t > & e, ComputeGraphVar< ViewT > &... vars) |
void | wait (cudaEvent_t e, int flag=cudaEventWaitDefault) |
void | wait (const ComputeGraphVar< cudaEvent_t > & e, const std::vector< ComputeGraphVarBase * > & vars) |
void | wait (const ComputeGraphVar< cudaEvent_t > & e, ComputeGraphVar< ViewT > &... vars) |
void | wait () |
void | when (cudaEvent_t e, int flag=cudaEventWaitDefault) |
~LaunchCore () |
Public Static Functions inherited from muda::LaunchCore
See muda::LaunchCore
Type | Name |
---|---|
void | file_line (std::string_view file, int line) |
void | kernel_name (std::string_view name) |
void | wait_device () |
void | wait_event (cudaEvent_t event) |
void | wait_stream (::cudaStream_t stream) |
Protected Types inherited from muda::LaunchCore
See muda::LaunchCore
Type | Name |
---|---|
typedef std::shared_ptr< T > | S |
Protected Attributes inherited from muda::LaunchCore
See muda::LaunchCore
Type | Name |
---|---|
::cudaStream_t | m_stream |
Protected Functions inherited from muda::CubWrapper
See muda::CubWrapper
Type | Name |
---|---|
std::byte * | prepare_buffer (size_t reqSize) |
Protected Functions inherited from muda::LaunchBase
See muda::LaunchBase
Type | Name |
---|---|
T & | pop_kernel_label () |
Protected Functions inherited from muda::LaunchCore
See muda::LaunchCore
Type | Name |
---|---|
MUDA_HOST void | pop_kernel_label () |
MUDA_GENERIC::cudaStream_t | stream () const |
Public Functions Documentation
function Flagged [1/2]
template<typename InputIteratorT, typename FlagIterator, typename OutputIteratorT, typename NumSelectedIteratorT>
inline DevicePartition & muda::DevicePartition::Flagged (
InputIteratorT d_in,
FlagIterator d_flags,
OutputIteratorT d_out,
NumSelectedIteratorT d_num_selected_out,
int num_items
)
function Flagged [2/2]
template<typename InputIteratorT, typename FlagIterator, typename OutputIteratorT, typename NumSelectedIteratorT>
inline DevicePartition & muda::DevicePartition::Flagged (
void * d_temp_storage,
size_t & temp_storage_bytes,
InputIteratorT d_in,
FlagIterator d_flags,
OutputIteratorT d_out,
NumSelectedIteratorT d_num_selected_out,
int num_items
)
function If [1/4]
template<typename InputIteratorT, typename OutputIteratorT, typename NumSelectedIteratorT, typename SelectOp>
inline DevicePartition & muda::DevicePartition::If (
InputIteratorT d_in,
OutputIteratorT d_out,
NumSelectedIteratorT d_num_selected_out,
int num_items,
SelectOp select_op
)
function If [2/4]
template<typename InputIteratorT, typename FirstOutputIteratorT, typename SecondOutputIteratorT, typename UnselectedOutputIteratorT, typename NumSelectedIteratorT, typename SelectFirstPartOp, typename SelectSecondPartOp>
inline DevicePartition & muda::DevicePartition::If (
InputIteratorT d_in,
FirstOutputIteratorT d_first_part_out,
SecondOutputIteratorT d_second_part_out,
UnselectedOutputIteratorT d_unselected_out,
NumSelectedIteratorT d_num_selected_out,
int num_items,
SelectFirstPartOp select_first_part_op,
SelectSecondPartOp select_second_part_op
)
function If [3/4]
template<typename InputIteratorT, typename OutputIteratorT, typename NumSelectedIteratorT, typename SelectOp>
inline DevicePartition & muda::DevicePartition::If (
void * d_temp_storage,
size_t & temp_storage_bytes,
InputIteratorT d_in,
OutputIteratorT d_out,
NumSelectedIteratorT d_num_selected_out,
int num_items,
SelectOp select_op
)
function If [4/4]
template<typename InputIteratorT, typename FirstOutputIteratorT, typename SecondOutputIteratorT, typename UnselectedOutputIteratorT, typename NumSelectedIteratorT, typename SelectFirstPartOp, typename SelectSecondPartOp>
inline DevicePartition & muda::DevicePartition::If (
void * d_temp_storage,
size_t & temp_storage_bytes,
InputIteratorT d_in,
FirstOutputIteratorT d_first_part_out,
SecondOutputIteratorT d_second_part_out,
UnselectedOutputIteratorT d_unselected_out,
NumSelectedIteratorT d_num_selected_out,
int num_items,
SelectFirstPartOp select_first_part_op,
SelectSecondPartOp select_second_part_op
)
The documentation for this class was generated from the following file src/muda/cub/device/device_partition.h