|
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