Files
compute-runtime/shared/source/dll/pool_info.cpp
Aleksander Czerwionka 0f639dc723 feature: set manager pool sizes based on isExtendedUsmPoolSizeEnabled
Related-To: NEO-16082

Signed-off-by: Aleksander Czerwionka <aleksander.czerwionka@intel.com>
2025-10-28 08:01:27 +01:00

42 lines
1.3 KiB
C++

/*
* Copyright (C) 2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#include "shared/source/memory_manager/pool_info.h"
#include "shared/source/helpers/constants.h"
#include "shared/source/helpers/gfx_core_helper.h"
namespace NEO {
static constexpr uint64_t KB = MemoryConstants::kiloByte; // NOLINT(readability-identifier-naming)
static constexpr uint64_t MB = MemoryConstants::megaByte; // NOLINT(readability-identifier-naming)
// clang-format off
const std::array<const PoolInfo, 3> PoolInfo::poolInfos = {
PoolInfo{ 0, 4 * KB, 2 * MB},
PoolInfo{ 4 * KB + 1, 64 * KB, 2 * MB},
PoolInfo{64 * KB + 1, 1 * MB, 2 * MB}};
const std::array<const PoolInfo, 3> PoolInfo::extendedPoolInfos = {
PoolInfo{ 0, 4 * KB, 2 * MB},
PoolInfo{ 4 * KB + 1, 64 * KB, 2 * MB},
PoolInfo{64 * KB + 1, 2 * MB, 16 * MB}};
// clang-format on
const std::array<const PoolInfo, 3> PoolInfo::getPoolInfos(const GfxCoreHelper &gfxCoreHelper) {
if (gfxCoreHelper.isExtendedUsmPoolSizeEnabled()) {
return extendedPoolInfos;
}
return poolInfos;
}
size_t PoolInfo::getMaxPoolableSize(const GfxCoreHelper &gfxCoreHelper) {
if (gfxCoreHelper.isExtendedUsmPoolSizeEnabled()) {
return 2 * MB;
}
return 1 * MB;
}
} // namespace NEO