| 117 | } |
| 118 | |
| 119 | inline auto make_and_set_managed_pool |
| 120 | ( |
| 121 | std::size_t initial_size, |
| 122 | std::size_t maximum_size |
| 123 | ) |
| 124 | { |
| 125 | // std::cout<< " make_managed_pool called with init_size" |
| 126 | // <<initial_size<<" max_size "<<maximum_size<<"\n"; |
| 127 | |
| 128 | auto resource = rmm::mr::make_owning_wrapper<rmm::mr::pool_memory_resource> |
| 129 | ( make_managed(), initial_size, maximum_size ) ; |
| 130 | rmm::mr::set_current_device_resource( resource.get()) ; |
| 131 | return resource; |
| 132 | } |
| 133 | |
| 134 | inline std::shared_ptr<rmm::cuda_stream_pool> make_and_set_cuda_stream_pool |
| 135 | ( |
no test coverage detected