|
| 1 | +// Copyright (C) 2024 Intel Corporation |
| 2 | +// Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT. |
| 3 | +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 4 | + |
| 5 | +#include "umf/pools/pool_jemalloc.h" |
| 6 | +#include "umf/pools/pool_scalable.h" |
| 7 | +#include "umf/providers/provider_coarse.h" |
| 8 | +#include "umf/providers/provider_file_memory.h" |
| 9 | + |
| 10 | +#include "pool.hpp" |
| 11 | +#include "poolFixtures.hpp" |
| 12 | + |
| 13 | +using umf_test::test; |
| 14 | +using namespace umf_test; |
| 15 | + |
| 16 | +#define INIT_BUFFER_SIZE 4096 |
| 17 | +#define FILE_PATH ((char *)"/tmp/file_provider") |
| 18 | + |
| 19 | +coarse_memory_provider_params_t getCoarseParams(size_t init_buffer_size) { |
| 20 | + coarse_memory_provider_params_t coarse_memory_provider_params; |
| 21 | + |
| 22 | + // make sure there are no undefined members - prevent a UB |
| 23 | + memset(&coarse_memory_provider_params, 0, |
| 24 | + sizeof(coarse_memory_provider_params)); |
| 25 | + |
| 26 | + // upstream_memory_provider will be set later in umfPoolTest |
| 27 | + coarse_memory_provider_params.upstream_memory_provider = nullptr; |
| 28 | + coarse_memory_provider_params.immediate_init_from_upstream = true; |
| 29 | + coarse_memory_provider_params.init_buffer = nullptr; |
| 30 | + coarse_memory_provider_params.init_buffer_size = init_buffer_size; |
| 31 | + coarse_memory_provider_params.destroy_upstream_memory_provider = true; |
| 32 | + |
| 33 | + return coarse_memory_provider_params; |
| 34 | +} |
| 35 | + |
| 36 | +auto coarseParams = getCoarseParams(INIT_BUFFER_SIZE); |
| 37 | +auto fileParams = umfFileMemoryProviderParamsDefault(FILE_PATH); |
| 38 | + |
| 39 | +INSTANTIATE_TEST_SUITE_P(jemallocCoarseFileTest, umfPoolTest, |
| 40 | + ::testing::Values(poolCreateExtParams{ |
| 41 | + umfJemallocPoolOps(), nullptr, |
| 42 | + umfFileMemoryProviderOps(), &fileParams, |
| 43 | + &coarseParams})); |
| 44 | + |
| 45 | +INSTANTIATE_TEST_SUITE_P(scalableCoarseFileTest, umfPoolTest, |
| 46 | + ::testing::Values(poolCreateExtParams{ |
| 47 | + umfScalablePoolOps(), nullptr, |
| 48 | + umfFileMemoryProviderOps(), &fileParams, |
| 49 | + &coarseParams})); |
0 commit comments