DevicePartition Struct Reference#
hipcub::DevicePartition Struct Reference
Static Public Member Functions | |
| template<typename InputIteratorT , typename FlagIterator , typename OutputIteratorT , typename NumSelectedIteratorT > | |
| __host__ static __forceinline__ hipError_t | 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, hipStream_t stream=0, bool debug_synchronous=false) |
| template<typename InputIteratorT , typename OutputIteratorT , typename NumSelectedIteratorT , typename SelectOp > | |
| __host__ static __forceinline__ hipError_t | 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, hipStream_t stream=0, bool debug_synchronous=false) |
| template<typename InputIteratorT , typename FlagIterator , typename OutputIteratorT , typename NumSelectedIteratorT > | |
| __host__ static __forceinline__ hipError_t | 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, hipStream_t stream=0, bool debug_synchronous=false) |
| template<typename InputIteratorT , typename OutputIteratorT , typename NumSelectedIteratorT , typename SelectOp > | |
| __host__ static __forceinline__ hipError_t | 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, hipStream_t stream=0, bool debug_synchronous=false) |
Member Function Documentation
◆ Flagged() [1/2]
template<typename InputIteratorT , typename FlagIterator , typename OutputIteratorT , typename NumSelectedIteratorT >
|
inlinestatic |
- Parameters
-
[in] d_temp_storage Device-accessible allocation of temporary storage. When NULL, the required allocation size is written to temp_storage_bytesand no work is done.[in,out] temp_storage_bytes Reference to size in bytes of d_temp_storageallocation[in] d_in Pointer to the input sequence of data items [in] d_flags Pointer to the input sequence of selection flags [out] d_out Pointer to the output sequence of partitioned data items [out] d_num_selected_out Pointer to the output total number of items selected (i.e., the offset of the unselected partition) [in] num_items Total number of items to select from [in] stream [optional] hip stream to launch kernels within. Default is stream0. [in] debug_synchronous [optional] Whether or not to synchronize the stream after every kernel launch to check for errors. May cause significant slowdown. Default is false.
◆ If() [1/2]
template<typename InputIteratorT , typename OutputIteratorT , typename NumSelectedIteratorT , typename SelectOp >
|
inlinestatic |
- Parameters
-
[in] d_temp_storage Device-accessible allocation of temporary storage. When NULL, the required allocation size is written to temp_storage_bytesand no work is done.[in,out] temp_storage_bytes Reference to size in bytes of d_temp_storageallocation[in] d_in Pointer to the input sequence of data items [out] d_out Pointer to the output sequence of partitioned data items [out] d_num_selected_out Pointer to the output total number of items selected (i.e., the offset of the unselected partition) [in] num_items Total number of items to select from [in] select_op Unary selection operator [in] stream [optional] hip stream to launch kernels within. Default is stream0. [in] debug_synchronous [optional] Whether or not to synchronize the stream after every kernel launch to check for errors. May cause significant slowdown. Default is false.
◆ Flagged() [2/2]
template<typename InputIteratorT , typename FlagIterator , typename OutputIteratorT , typename NumSelectedIteratorT >
|
inlinestatic |
- Parameters
-
[in] d_temp_storage Device-accessible allocation of temporary storage. When NULL, the required allocation size is written to temp_storage_bytesand no work is done.[in,out] temp_storage_bytes Reference to size in bytes of d_temp_storageallocation[in] d_in Pointer to the input sequence of data items [in] d_flags Pointer to the input sequence of selection flags [out] d_out Pointer to the output sequence of partitioned data items [out] d_num_selected_out Pointer to the output total number of items selected (i.e., the offset of the unselected partition) [in] num_items Total number of items to select from [in] stream [optional] hip stream to launch kernels within. Default is stream0. [in] debug_synchronous [optional] Whether or not to synchronize the stream after every kernel launch to check for errors. May cause significant slowdown. Default is false.
◆ If() [2/2]
template<typename InputIteratorT , typename OutputIteratorT , typename NumSelectedIteratorT , typename SelectOp >
|
inlinestatic |
- Parameters
-
[in] d_temp_storage Device-accessible allocation of temporary storage. When NULL, the required allocation size is written to temp_storage_bytesand no work is done.[in,out] temp_storage_bytes Reference to size in bytes of d_temp_storageallocation[in] d_in Pointer to the input sequence of data items [out] d_out Pointer to the output sequence of partitioned data items [out] d_num_selected_out Pointer to the output total number of items selected (i.e., the offset of the unselected partition) [in] num_items Total number of items to select from [in] select_op Unary selection operator [in] stream [optional] hip stream to launch kernels within. Default is stream0. [in] debug_synchronous [optional] Whether or not to synchronize the stream after every kernel launch to check for errors. May cause significant slowdown. Default is false.
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipcub/checkouts/docs-5.0.0/hipcub/include/hipcub/backend/cub/device/device_partition.hpp