|
CUB
|
DevicePartition provides device-wide, parallel operations for partitioning sequences of data items residing within global memory.
CUB_CDP macro in your compiler's macro definitions.int32 items, where 50% of the items are randomly selected for the first partition. Performance plots for other scenarios can be found in the detailed method descriptions below.
Definition at line 73 of file device_partition.cuh.
Static Public Methods | |
| template<typename InputIterator , typename FlagIterator , typename OutputIterator , typename NumSelectedIterator > | |
| CUB_RUNTIME_FUNCTION static __forceinline__ cudaError_t | Flagged (void *d_temp_storage, size_t &temp_storage_bytes, InputIterator d_in, FlagIterator d_flags, OutputIterator d_out, NumSelectedIterator d_num_selected, int num_items, cudaStream_t stream=0, bool debug_synchronous=false) |
Uses the d_flags sequence to split the corresponding items from d_in into a partitioned sequence d_out. The total number of items copied into the first partition is written to d_num_selected.
. | |
| template<typename InputIterator , typename OutputIterator , typename NumSelectedIterator , typename SelectOp > | |
| CUB_RUNTIME_FUNCTION static __forceinline__ cudaError_t | If (void *d_temp_storage, size_t &temp_storage_bytes, InputIterator d_in, OutputIterator d_out, NumSelectedIterator d_num_selected, int num_items, SelectOp select_op, cudaStream_t stream=0, bool debug_synchronous=false) |
Uses the select_op functor to split the corresponding items from d_in into a partitioned sequence d_out. The total number of items copied into the first partition is written to d_num_selected.
. | |
|
inlinestatic |
Uses the d_flags sequence to split the corresponding items from d_in into a partitioned sequence d_out. The total number of items copied into the first partition is written to d_num_selected.
d_flags must be castable to bool (e.g., bool, char, int, etc.).d_out and maintain their original relative ordering, however copies of the unselected items are compacted into the rear of d_out in reverse order.d_temp_storage is NULL, no work is done and the required allocation size is returned in temp_storage_bytes.CUB_CDP macro in your compiler's macro definitions.int device vector. | InputIterator | [inferred] Random-access input iterator type for reading input items (may be a simple pointer type) |
| FlagIterator | [inferred] Random-access input iterator type for reading selection flags (may be a simple pointer type) |
| OutputIterator | [inferred] Random-access output iterator type for writing output items (may be a simple pointer type) |
| NumSelectedIterator | [inferred] Output iterator type for recording the number of items selected (may be a simple pointer type) |
| [in] | d_temp_storage | Device allocation of temporary storage. When NULL, the required allocation size is written to temp_storage_bytes and no work is done. |
| [in,out] | temp_storage_bytes | Reference to size in bytes of d_temp_storage allocation |
| [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 | 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] CUDA 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. |
Definition at line 127 of file device_partition.cuh.
|
inlinestatic |
Uses the select_op functor to split the corresponding items from d_in into a partitioned sequence d_out. The total number of items copied into the first partition is written to d_num_selected.
d_out and maintain their original relative ordering, however copies of the unselected items are compacted into the rear of d_out in reverse order.d_temp_storage is NULL, no work is done and the required allocation size is returned in temp_storage_bytes.CUB_CDP macro in your compiler's macro definitions.int32 and int64 items, respectively. Items are selected for the first partition with 50% probability.
int device vector. | InputIterator | [inferred] Random-access input iterator type for reading input items (may be a simple pointer type) |
| OutputIterator | [inferred] Random-access output iterator type for writing output items (may be a simple pointer type) |
| NumSelectedIterator | [inferred] Output iterator type for recording the number of items selected (may be a simple pointer type) |
| SelectOp | [inferred] Selection functor type having member bool operator()(const T &a) |
| [in] | d_temp_storage | Device allocation of temporary storage. When NULL, the required allocation size is written to temp_storage_bytes and no work is done. |
| [in,out] | temp_storage_bytes | Reference to size in bytes of d_temp_storage allocation |
| [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 | 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] CUDA 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. |
Definition at line 236 of file device_partition.cuh.
1.8.4