/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipcub/checkouts/docs-5.4.1/hipcub/include/hipcub/backend/cub/device/device_partition.hpp Source File

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

device_partition.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-2021, 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_CUB_DEVICE_DEVICE_PARTITION_HPP_

31 #define HIPCUB_CUB_DEVICE_DEVICE_PARTITION_HPP_

32 

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

34 

35 #include <cub/device/device_partition.cuh>

36 

37 BEGIN_HIPCUB_NAMESPACE

38 

39 struct DevicePartition

40 {

41  template <

42  typename InputIteratorT,

43  typename FlagIterator,

44  typename OutputIteratorT,

45  typename NumSelectedIteratorT>

46  HIPCUB_RUNTIME_FUNCTION __forceinline__

47  static hipError_t Flagged(

48  void* d_temp_storage,

49  size_t &temp_storage_bytes,

50  InputIteratorT d_in,

51  FlagIterator d_flags,

52  OutputIteratorT d_out,

53  NumSelectedIteratorT d_num_selected_out,

54  int num_items,

55  hipStream_t stream = 0,

56  bool debug_synchronous = false)

57  {

58  return hipCUDAErrorTohipError(

59  ::cub::DevicePartition::Flagged(

60  d_temp_storage,

61  temp_storage_bytes,

62  d_in,

63  d_flags,

64  d_out,

65  d_num_selected_out,

66  num_items,

67  stream,

68  debug_synchronous

69  )

70  );

71  }

72 

73  template <

74  typename InputIteratorT,

75  typename OutputIteratorT,

76  typename NumSelectedIteratorT,

77  typename SelectOp>

78  HIPCUB_RUNTIME_FUNCTION __forceinline__

79  static hipError_t If(

80  void* d_temp_storage,

81  size_t &temp_storage_bytes,

82  InputIteratorT d_in,

83  OutputIteratorT d_out,

84  NumSelectedIteratorT d_num_selected_out,

85  int num_items,

86  SelectOp select_op,

87  hipStream_t stream = 0,

88  bool debug_synchronous = false)

89  {

90  return hipCUDAErrorTohipError(

91  ::cub::DevicePartition::If(

92  d_temp_storage,

93  temp_storage_bytes,

94  d_in,

95  d_out,

96  d_num_selected_out,

97  num_items,

98  select_op,

99  stream,

100  debug_synchronous

101  )

102  );

103  }

104 

105  template <typename InputIteratorT,

106  typename FirstOutputIteratorT,

107  typename SecondOutputIteratorT,

108  typename UnselectedOutputIteratorT,

109  typename NumSelectedIteratorT,

110  typename SelectFirstPartOp,

111  typename SelectSecondPartOp>

112  HIPCUB_RUNTIME_FUNCTION __forceinline__ static hipError_t

113  If(void *d_temp_storage,

114  std::size_t &temp_storage_bytes,

115  InputIteratorT d_in,

116  FirstOutputIteratorT d_first_part_out,

117  SecondOutputIteratorT d_second_part_out,

118  UnselectedOutputIteratorT d_unselected_out,

119  NumSelectedIteratorT d_num_selected_out,

120  int num_items,

121  SelectFirstPartOp select_first_part_op,

122  SelectSecondPartOp select_second_part_op,

123  hipStream_t stream = 0,

124  bool debug_synchronous = false)

125  {

126  return hipCUDAErrorTohipError(

127  ::cub::DevicePartition::If(

128  d_temp_storage,

129  temp_storage_bytes,

130  d_in,

131  d_first_part_out,

132  d_second_part_out,

133  d_unselected_out,

134  d_num_selected_out,

135  num_items,

136  select_first_part_op,

137  select_second_part_op,

138  stream,

139  debug_synchronous

140  )

141  );

142  }

143 };

144 

145 END_HIPCUB_NAMESPACE

146 

147 #endif // HIPCUB_CUB_DEVICE_DEVICE_PARTITION_HPP_

hipcub::DevicePartition

Definition: device_partition.hpp:40

hipcub::DevicePartition::If

__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)

Definition: device_partition.hpp:79

hipcub::DevicePartition::Flagged

__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)

Definition: device_partition.hpp:47

hipcub::If

Definition: util_type.hpp:54