intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-21 09:14:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

feature: set manager pool sizes based on isExtendedUsmPoolSizeEnabled

Browse Source 
Related-To: NEO-16082

Signed-off-by: Aleksander Czerwionka <aleksander.czerwionka@intel.com>
This commit is contained in:
Aleksander Czerwionka
2025-10-27 11:26:29 +00:00
committed by Compute-Runtime-Automation
parent 4087c80f37
commit 0f639dc723
9 changed files with 122 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
shared/source/memory_manager/pool_info.h
View File

@@ -10,6 +10,7 @@
#include <cstddef>
namespace NEO {
class GfxCoreHelper;
class PoolInfo {
public:
size_t minServicedSize;
@@ -19,13 +20,11 @@ class PoolInfo {
return this->minServicedSize < rhs.minServicedSize;
}
static const std::array<const PoolInfo, 3> getPoolInfos() {
return poolInfos;
}
static size_t getMaxPoolableSize();
static const std::array<const PoolInfo, 3> getPoolInfos(const GfxCoreHelper &gfxCoreHelper);
static size_t getMaxPoolableSize(const GfxCoreHelper &gfxCoreHelper);
private:
static const std::array<const PoolInfo, 3> poolInfos;
static const std::array<const PoolInfo, 3> extendedPoolInfos;
};
} // namespace NEO

8
shared/source/memory_manager/unified_memory_pooling.cpp
View File

@@ -202,7 +202,7 @@ bool UsmMemAllocPoolsManager::initialize(SVMAllocsManager *svmMemoryManager) {
poolMemoryType != InternalMemoryType::hostUnifiedMemory);
DEBUG_BREAK_IF(device == nullptr && poolMemoryType == InternalMemoryType::deviceUnifiedMemory);
this->svmMemoryManager = svmMemoryManager;
for (const auto &poolInfo : PoolInfo::getPoolInfos()) {
for (const auto &poolInfo : PoolInfo::getPoolInfos(device->getGfxCoreHelper())) {
this->pools[poolInfo] = std::vector<std::unique_ptr<UsmMemAllocPool>>();
auto pool = tryAddPool(poolInfo);
if (nullptr == pool) {
@@ -234,7 +234,7 @@ void *UsmMemAllocPoolsManager::createUnifiedMemoryAllocation(size_t size, const
}
std::unique_lock<std::mutex> lock(mtx);
void *ptr = nullptr;
for (const auto &poolInfo : PoolInfo::getPoolInfos()) {
for (const auto &poolInfo : PoolInfo::getPoolInfos(device->getGfxCoreHelper())) {
if (size <= poolInfo.maxServicedSize) {
for (auto &pool : this->pools[poolInfo]) {
if (nullptr != (ptr = pool->createUnifiedMemoryAllocation(size, memoryProperties))) {
@@ -273,7 +273,7 @@ bool UsmMemAllocPoolsManager::canAddPool(PoolInfo poolInfo) {
}
bool UsmMemAllocPoolsManager::canBePooled(size_t size, const UnifiedMemoryProperties &memoryProperties) {
return size <= PoolInfo::getMaxPoolableSize() &&
return size <= PoolInfo::getMaxPoolableSize(device->getGfxCoreHelper()) &&
UsmMemAllocPool::alignmentIsAllowed(memoryProperties.alignment) &&
UsmMemAllocPool::flagsAreAllowed(memoryProperties);
}
@@ -328,7 +328,7 @@ size_t UsmMemAllocPoolsManager::getOffsetInPool(const void *ptr) {
UsmMemAllocPool *UsmMemAllocPoolsManager::getPoolContainingAlloc(const void *ptr) {
std::unique_lock<std::mutex> lock(mtx);
for (const auto &poolInfo : PoolInfo::getPoolInfos()) {
for (const auto &poolInfo : PoolInfo::getPoolInfos(device->getGfxCoreHelper())) {
for (auto &pool : this->pools[poolInfo]) {
if (pool->isInPool(ptr)) {
return pool.get();