40 #include "dispatch/device_select_dispatch.cuh"
41 #include "../util_namespace.cuh"
122 typename InputIterator,
123 typename FlagIterator,
124 typename OutputIterator,
125 typename NumSelectedIterator>
126 CUB_RUNTIME_FUNCTION __forceinline__
128 void *d_temp_storage,
129 size_t &temp_storage_bytes,
131 FlagIterator d_flags,
132 OutputIterator d_out,
133 NumSelectedIterator d_num_selected,
135 cudaStream_t stream = 0,
136 bool debug_synchronous =
false)
142 return DeviceSelectDispatch<InputIterator, FlagIterator, OutputIterator, NumSelectedIterator, SelectOp, EqualityOp, Offset, true>::Dispatch(
231 typename InputIterator,
232 typename OutputIterator,
233 typename NumSelectedIterator,
235 CUB_RUNTIME_FUNCTION __forceinline__
236 static cudaError_t
If(
237 void *d_temp_storage,
238 size_t &temp_storage_bytes,
240 OutputIterator d_out,
241 NumSelectedIterator d_num_selected,
244 cudaStream_t stream = 0,
245 bool debug_synchronous =
false)
251 return DeviceSelectDispatch<InputIterator, FlagIterator, OutputIterator, NumSelectedIterator, SelectOp, EqualityOp, Offset, true>::Dispatch(