40 #include "dispatch/device_select_dispatch.cuh"
41 #include "../util_namespace.cuh"
129 typename InputIterator,
130 typename FlagIterator,
131 typename OutputIterator,
132 typename NumSelectedIterator>
133 CUB_RUNTIME_FUNCTION __forceinline__
135 void *d_temp_storage,
136 size_t &temp_storage_bytes,
138 FlagIterator d_flags,
139 OutputIterator d_out,
140 NumSelectedIterator d_num_selected,
142 cudaStream_t stream = 0,
143 bool debug_synchronous =
false)
149 return DeviceSelectDispatch<InputIterator, FlagIterator, OutputIterator, NumSelectedIterator, SelectOp, EqualityOp, Offset, false>::Dispatch(
236 typename InputIterator,
237 typename OutputIterator,
238 typename NumSelectedIterator,
240 CUB_RUNTIME_FUNCTION __forceinline__
241 static cudaError_t
If(
242 void *d_temp_storage,
243 size_t &temp_storage_bytes,
245 OutputIterator d_out,
246 NumSelectedIterator d_num_selected,
249 cudaStream_t stream = 0,
250 bool debug_synchronous =
false)
256 return DeviceSelectDispatch<InputIterator, FlagIterator, OutputIterator, NumSelectedIterator, SelectOp, EqualityOp, Offset, false>::Dispatch(
328 typename InputIterator,
329 typename OutputIterator,
330 typename NumSelectedIterator>
331 CUB_RUNTIME_FUNCTION __forceinline__
333 void *d_temp_storage,
334 size_t &temp_storage_bytes,
336 OutputIterator d_out,
337 NumSelectedIterator d_num_selected,
339 cudaStream_t stream = 0,
340 bool debug_synchronous =
false)
347 return DeviceSelectDispatch<InputIterator, FlagIterator, OutputIterator, NumSelectedIterator, SelectOp, EqualityOp, Offset, false>::Dispatch(