/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipcub/checkouts/docs-5.4.3/hipcub/include/hipcub/backend/cub/util_allocator.hpp Source File

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

util_allocator.hpp

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

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

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

4  * Modifications Copyright (c) 2019-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_UTIL_ALLOCATOR_HPP_

31 #define HIPCUB_CUB_UTIL_ALLOCATOR_HPP_

32 

33 #if defined(HIPCUB_STDERR) && !defined(CUB_STDERR)

34  #define CUB_STDERR

35 #endif

36 

37 #include "../../config.hpp"

38 

39 #include <cub/util_allocator.cuh>

40 

41 BEGIN_HIPCUB_NAMESPACE

42 

43 struct CachingDeviceAllocator : public ::cub::CachingDeviceAllocator

44 {

45  hipError_t SetMaxCachedBytes(

46  size_t max_cached_bytes)

47  {

48  return hipCUDAErrorTohipError(

49  ::cub::CachingDeviceAllocator::SetMaxCachedBytes(max_cached_bytes)

50  );

51  }

52 

53  hipError_t DeviceAllocate(

54  int device,

55  void **d_ptr,

56  size_t bytes,

57  hipStream_t active_stream = 0)

58  {

59  return hipCUDAErrorTohipError(

60  ::cub::CachingDeviceAllocator::DeviceAllocate(device, d_ptr, bytes, active_stream)

61  );

62  }

63 

64  hipError_t DeviceAllocate(

65  void **d_ptr,

66  size_t bytes,

67  hipStream_t active_stream = 0)

68  {

69  return hipCUDAErrorTohipError(

70  ::cub::CachingDeviceAllocator::DeviceAllocate(d_ptr, bytes, active_stream)

71  );

72  }

73 

74  hipError_t DeviceFree(

75  int device,

76  void* d_ptr)

77  {

78  return hipCUDAErrorTohipError(

79  ::cub::CachingDeviceAllocator::DeviceFree(device, d_ptr)

80  );

81  }

82 

83  hipError_t DeviceFree(

84  void* d_ptr)

85  {

86  return hipCUDAErrorTohipError(

87  ::cub::CachingDeviceAllocator::DeviceFree(d_ptr)

88  );

89  }

90 

91  hipError_t FreeAllCached()

92  {

93  return hipCUDAErrorTohipError(

94  ::cub::CachingDeviceAllocator::FreeAllCached()

95  );

96  }

97 };

98 

99 END_HIPCUB_NAMESPACE

100 

101 #endif // HIPCUB_CUB_UTIL_ALLOCATOR_HPP_

hipcub::CachingDeviceAllocator

Definition: util_allocator.hpp:44