|
15 | 15 | #include "umf_pools/disjoint_pool_config_parser.hpp" |
16 | 16 | #include "usm.hpp" |
17 | 17 |
|
18 | | -#include <umf/pools/pool_disjoint.h> |
19 | | -#include <umf/pools/pool_proxy.h> |
20 | 18 | #include <umf/providers/provider_level_zero.h> |
21 | 19 |
|
22 | 20 | namespace umf { |
@@ -158,28 +156,6 @@ makeProvider(usm::pool_descriptor poolDescriptor) { |
158 | 156 | return std::move(provider); |
159 | 157 | } |
160 | 158 |
|
161 | | -static umf::pool_unique_handle_t |
162 | | -makeDisjointPool(umf::provider_unique_handle_t &&provider, |
163 | | - usm::umf_disjoint_pool_config_t &poolParams) { |
164 | | - auto umfParams = getUmfParamsHandle(poolParams); |
165 | | - auto [ret, poolHandle] = |
166 | | - umf::poolMakeUniqueFromOps(umfDisjointPoolOps(), std::move(provider), |
167 | | - static_cast<void *>(umfParams.get())); |
168 | | - if (ret != UMF_RESULT_SUCCESS) |
169 | | - throw umf::umf2urResult(ret); |
170 | | - return std::move(poolHandle); |
171 | | -} |
172 | | - |
173 | | -static umf::pool_unique_handle_t |
174 | | -makeProxyPool(umf::provider_unique_handle_t &&provider) { |
175 | | - auto [ret, poolHandle] = umf::poolMakeUniqueFromOps( |
176 | | - umfProxyPoolOps(), std::move(provider), nullptr); |
177 | | - if (ret != UMF_RESULT_SUCCESS) |
178 | | - throw umf::umf2urResult(ret); |
179 | | - |
180 | | - return std::move(poolHandle); |
181 | | -} |
182 | | - |
183 | 159 | ur_usm_pool_handle_t_::ur_usm_pool_handle_t_(ur_context_handle_t hContext, |
184 | 160 | ur_usm_pool_desc_t *pPoolDesc) |
185 | 161 | : hContext(hContext) { |
@@ -207,10 +183,10 @@ ur_usm_pool_handle_t_::ur_usm_pool_handle_t_(ur_context_handle_t hContext, |
207 | 183 | if (disjointPoolConfigs.has_value()) { |
208 | 184 | auto &poolConfig = |
209 | 185 | disjointPoolConfigs.value().Configs[descToDisjoinPoolMemType(desc)]; |
210 | | - poolManager.addPool(desc, |
211 | | - makeDisjointPool(makeProvider(desc), poolConfig)); |
| 186 | + poolManager.addPool( |
| 187 | + desc, usm::makeDisjointPool(makeProvider(desc), poolConfig)); |
212 | 188 | } else { |
213 | | - poolManager.addPool(desc, makeProxyPool(makeProvider(desc))); |
| 189 | + poolManager.addPool(desc, usm::makeProxyPool(makeProvider(desc))); |
214 | 190 | } |
215 | 191 | } |
216 | 192 | } |
|
0 commit comments