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

• home
• docs
• checkouts
• readthedocs.org
• user_builds
• advanced-micro-devices-hipcub
• checkouts
• docs-5.6.0
• hipcub
• include
• hipcub
• backend
• rocprim
• 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-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_PARTITION_HPP_

31 #define HIPCUB_ROCPRIM_DEVICE_PARTITION_HPP_

32 

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

34 

35 #include <rocprim/device/device_partition.hpp>

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 rocprim::partition(

59  d_temp_storage,

60  temp_storage_bytes,

61  d_in,

62  d_flags,

63  d_out,

64  d_num_selected_out,

65  num_items,

66  stream,

67  debug_synchronous);

68  }

69 

70  template <

71  typename InputIteratorT,

72  typename OutputIteratorT,

73  typename NumSelectedIteratorT,

74  typename SelectOp>

75  HIPCUB_RUNTIME_FUNCTION __forceinline__

76  static hipError_t If(

77  void* d_temp_storage,

78  size_t &temp_storage_bytes,

79  InputIteratorT d_in,

80  OutputIteratorT d_out,

81  NumSelectedIteratorT d_num_selected_out,

82  int num_items,

83  SelectOp select_op,

84  hipStream_t stream = 0,

85  bool debug_synchronous = false)

86  {

87  return rocprim::partition(

88  d_temp_storage,

89  temp_storage_bytes,

90  d_in,

91  d_out,

92  d_num_selected_out,

93  num_items,

94  select_op,

95  stream,

96  debug_synchronous);

97  }

98 

99  template <typename InputIteratorT,

100  typename FirstOutputIteratorT,

101  typename SecondOutputIteratorT,

102  typename UnselectedOutputIteratorT,

103  typename NumSelectedIteratorT,

104  typename SelectFirstPartOp,

105  typename SelectSecondPartOp>

106  HIPCUB_RUNTIME_FUNCTION __forceinline__ static hipError_t

107  If(void *d_temp_storage,

108  std::size_t &temp_storage_bytes,

109  InputIteratorT d_in,

110  FirstOutputIteratorT d_first_part_out,

111  SecondOutputIteratorT d_second_part_out,

112  UnselectedOutputIteratorT d_unselected_out,

113  NumSelectedIteratorT d_num_selected_out,

114  int num_items,

115  SelectFirstPartOp select_first_part_op,

116  SelectSecondPartOp select_second_part_op,

117  hipStream_t stream = 0,

118  bool debug_synchronous = false)

119  {

120  return rocprim::partition_three_way(

121  d_temp_storage,

122  temp_storage_bytes,

123  d_in,

124  d_first_part_out,

125  d_second_part_out,

126  d_unselected_out,

127  d_num_selected_out,

128  num_items,

129  select_first_part_op,

130  select_second_part_op,

131  stream,

132  debug_synchronous

133  );

134  }

135 };

136 

137 END_HIPCUB_NAMESPACE

138 

139 #endif

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:76

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