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

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

device_run_length_encode.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_CUB_DEVICE_DEVICE_RUN_LENGTH_ENCODE_HPP_

31 #define HIPCUB_CUB_DEVICE_DEVICE_RUN_LENGTH_ENCODE_HPP_

32 

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

34 

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

36 

37 BEGIN_HIPCUB_NAMESPACE

38 

39 class DeviceRunLengthEncode

40 {

41 public:

42  template<

43  typename InputIteratorT,

44  typename UniqueOutputIteratorT,

45  typename LengthsOutputIteratorT,

46  typename NumRunsOutputIteratorT

47  >

48  HIPCUB_RUNTIME_FUNCTION static

49  hipError_t Encode(void * d_temp_storage,

50  size_t& temp_storage_bytes,

51  InputIteratorT d_in,

52  UniqueOutputIteratorT d_unique_out,

53  LengthsOutputIteratorT d_counts_out,

54  NumRunsOutputIteratorT d_num_runs_out,

55  int num_items,

56  hipStream_t stream = 0,

57  bool debug_synchronous = false)

58  {

59  return hipCUDAErrorTohipError(

60  ::cub::DeviceRunLengthEncode::Encode(

61  d_temp_storage, temp_storage_bytes,

62  d_in,

63  d_unique_out, d_counts_out, d_num_runs_out,

64  num_items,

65  stream, debug_synchronous

66  )

67  );

68  }

69 

70  template<

71  typename InputIteratorT,

72  typename OffsetsOutputIteratorT,

73  typename LengthsOutputIteratorT,

74  typename NumRunsOutputIteratorT

75  >

76  HIPCUB_RUNTIME_FUNCTION static

77  hipError_t NonTrivialRuns(void * d_temp_storage,

78  size_t& temp_storage_bytes,

79  InputIteratorT d_in,

80  OffsetsOutputIteratorT d_offsets_out,

81  LengthsOutputIteratorT d_lengths_out,

82  NumRunsOutputIteratorT d_num_runs_out,

83  int num_items,

84  hipStream_t stream = 0,

85  bool debug_synchronous = false)

86  {

87  return hipCUDAErrorTohipError(

88  ::cub::DeviceRunLengthEncode::NonTrivialRuns(

89  d_temp_storage, temp_storage_bytes,

90  d_in,

91  d_offsets_out, d_lengths_out, d_num_runs_out,

92  num_items,

93  stream, debug_synchronous

94  )

95  );

96  }

97 };

98 

99 END_HIPCUB_NAMESPACE

100 

101 #endif // HIPCUB_CUB_DEVICE_DEVICE_RUN_LENGTH_ENCODE_HPP_

hipcub::DeviceRunLengthEncode

Definition: device_run_length_encode.hpp:40