/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipcub/checkouts/docs-5.7.0/hipcub/include/hipcub/backend/rocprim/device/device_select.hpp Source File

• home
• docs
• checkouts
• readthedocs.org
• user_builds
• advanced-micro-devices-hipcub
• checkouts
• docs-5.7.0
• hipcub
• include
• hipcub
• backend
• rocprim
• device

device_select.hpp

1 /******************************************************************************

2  * Copyright (c) 2010-2011, Duane Merrill. All rights reserved.

3  * Copyright (c) 2011-2018, NVIDIA CORPORATION. All rights reserved.

4  * Modifications Copyright (c) 2017-2020, Advanced Micro Devices, Inc. All rights reserved.

5  *

6  * Redistribution and use in source and binary forms, with or without

7  * modification, are permitted provided that the following conditions are met:

8  * * Redistributions of source code must retain the above copyright

9  * notice, this list of conditions and the following disclaimer.

10  * * Redistributions in binary form must reproduce the above copyright

11  * notice, this list of conditions and the following disclaimer in the

12  * documentation and/or other materials provided with the distribution.

13  * * Neither the name of the NVIDIA CORPORATION nor the

14  * names of its contributors may be used to endorse or promote products

15  * derived from this software without specific prior written permission.

16  *

17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND

18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

20  * DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY

21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

26  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

27  *

28  ******************************************************************************/

29 

30 #ifndef HIPCUB_ROCPRIM_DEVICE_DEVICE_SELECT_HPP_

31 #define HIPCUB_ROCPRIM_DEVICE_DEVICE_SELECT_HPP_

32 

33 #include "../../../config.hpp"

34 

35 #include "../thread/thread_operators.hpp"

36 

37 #include <rocprim/device/device_select.hpp>

38 

39 BEGIN_HIPCUB_NAMESPACE

40 

41 class DeviceSelect

42 {

43 public:

44  template <

45  typename InputIteratorT,

46  typename FlagIterator,

47  typename OutputIteratorT,

48  typename NumSelectedIteratorT

49  >

50  HIPCUB_RUNTIME_FUNCTION static

51  hipError_t Flagged(void *d_temp_storage,

52  size_t &temp_storage_bytes,

53  InputIteratorT d_in,

54  FlagIterator d_flags,

55  OutputIteratorT d_out,

56  NumSelectedIteratorT d_num_selected_out,

57  int num_items,

58  hipStream_t stream = 0,

59  bool debug_synchronous = false)

60  {

61  return ::rocprim::select(

62  d_temp_storage, temp_storage_bytes,

63  d_in, d_flags, d_out, d_num_selected_out, num_items,

64  stream, debug_synchronous

65  );

66  }

67 

68  template <

69  typename InputIteratorT,

70  typename OutputIteratorT,

71  typename NumSelectedIteratorT,

72  typename SelectOp

73  >

74  HIPCUB_RUNTIME_FUNCTION static

75  hipError_t If(void *d_temp_storage,

76  size_t &temp_storage_bytes,

77  InputIteratorT d_in,

78  OutputIteratorT d_out,

79  NumSelectedIteratorT d_num_selected_out,

80  int num_items,

81  SelectOp select_op,

82  hipStream_t stream = 0,

83  bool debug_synchronous = false)

84  {

85  return ::rocprim::select(

86  d_temp_storage, temp_storage_bytes,

87  d_in, d_out, d_num_selected_out, num_items, select_op,

88  stream, debug_synchronous

89  );

90  }

91 

92  template <

93  typename InputIteratorT,

94  typename OutputIteratorT,

95  typename NumSelectedIteratorT

96  >

97  HIPCUB_RUNTIME_FUNCTION static

98  hipError_t Unique(void *d_temp_storage,

99  size_t &temp_storage_bytes,

100  InputIteratorT d_in,

101  OutputIteratorT d_out,

102  NumSelectedIteratorT d_num_selected_out,

103  int num_items,

104  hipStream_t stream = 0,

105  bool debug_synchronous = false)

106  {

107  return ::rocprim::unique(

108  d_temp_storage, temp_storage_bytes,

109  d_in, d_out, d_num_selected_out, num_items, hipcub::Equality(),

110  stream, debug_synchronous

111  );

112  }

113 

114  template <

115  typename KeyIteratorT,

116  typename ValueIteratorT,

117  typename OutputKeyIteratorT,

118  typename OutputValueIteratorT,

119  typename NumSelectedIteratorT

120  >

121  HIPCUB_RUNTIME_FUNCTION static

122  hipError_t UniqueByKey(void *d_temp_storage,

123  size_t &temp_storage_bytes,

124  KeyIteratorT d_keys_input,

125  ValueIteratorT d_values_input,

126  OutputKeyIteratorT d_keys_output,

127  OutputValueIteratorT d_values_output,

128  NumSelectedIteratorT d_num_selected_out,

129  int num_items,

130  hipStream_t stream = 0,

131  bool debug_synchronous = false)

132  {

133  return ::rocprim::unique_by_key(

134  d_temp_storage, temp_storage_bytes,

135  d_keys_input, d_values_input,

136  d_keys_output, d_values_output,

137  d_num_selected_out, num_items, hipcub::Equality(),

138  stream, debug_synchronous

139  );

140  }

141 };

142 

143 END_HIPCUB_NAMESPACE

144 

145 #endif // HIPCUB_ROCPRIM_DEVICE_DEVICE_SELECT_HPP_

hipcub::Equality

Definition: thread_operators.hpp:40