| 105 | // of each allocation |
| 106 | |
| 107 | inline auto make_and_set_device_pool |
| 108 | ( |
| 109 | std::size_t initial_size, |
| 110 | std::size_t maximum_size |
| 111 | ) |
| 112 | { |
| 113 | auto resource = rmm::mr::make_owning_wrapper<rmm::mr::pool_memory_resource> |
| 114 | ( make_cuda(), initial_size, maximum_size ) ; |
| 115 | rmm::mr::set_current_device_resource( resource.get()) ; |
| 116 | return resource; |
| 117 | } |
| 118 | |
| 119 | inline auto make_and_set_managed_pool |
| 120 | ( |
no test coverage detected