intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-03 14:55:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: enable usm reuse limit based on memory usage

Browse Source 
Related-To: NEO-14160, NEO-6893

Signed-off-by: Dominik Dabek <dominik.dabek@intel.com>
This commit is contained in:
Dominik Dabek
2025-03-27 13:38:30 +00:00
committed by Compute-Runtime-Automation
parent 915d657420
commit c76edaba4e
3 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
shared/test/unit_test/memory_manager/unified_memory_manager_cache_tests.cpp
View File

@@ -242,22 +242,22 @@ TEST_F(SvmDeviceAllocationCacheTest, givenReuseLimitFlagWhenInitUsmReuseLimitCal
DebugManagerStateRestore restore;
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(0);
device->initUsmReuseLimits();
const auto expectedLimitThreshold = std::numeric_limits<uint64_t>::max();
EXPECT_EQ(expectedLimitThreshold, device->usmReuseInfo.getLimitAllocationsReuseThreshold());
EXPECT_EQ(UsmReuseInfo::notLimited, device->usmReuseInfo.getLimitAllocationsReuseThreshold());
}
{
DebugManagerStateRestore restore;
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(80);
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(70);
device->initUsmReuseLimits();
const auto totalDeviceMemory = device->getGlobalMemorySize(static_cast<uint32_t>(device->getDeviceBitfield().to_ulong()));
const auto expectedLimitThreshold = static_cast<uint64_t>(0.8 * totalDeviceMemory);
const auto expectedLimitThreshold = static_cast<uint64_t>(0.7 * totalDeviceMemory);
EXPECT_EQ(expectedLimitThreshold, device->usmReuseInfo.getLimitAllocationsReuseThreshold());
}
{
DebugManagerStateRestore restore;
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(-1);
device->initUsmReuseLimits();
const auto expectedLimitThreshold = std::numeric_limits<uint64_t>::max();
const auto totalDeviceMemory = device->getGlobalMemorySize(static_cast<uint32_t>(device->getDeviceBitfield().to_ulong()));
const auto expectedLimitThreshold = static_cast<uint64_t>(0.8 * totalDeviceMemory);
EXPECT_EQ(expectedLimitThreshold, device->usmReuseInfo.getLimitAllocationsReuseThreshold());
}
}
@@ -1091,22 +1091,22 @@ TEST_F(SvmHostAllocationCacheTest, givenReuseLimitFlagWhenInitUsmReuseLimitCalle
DebugManagerStateRestore restore;
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(0);
memoryManager->initUsmReuseLimits();
const auto expectedLimitThreshold = std::numeric_limits<uint64_t>::max();
EXPECT_EQ(expectedLimitThreshold, memoryManager->usmReuseInfo.getLimitAllocationsReuseThreshold());
EXPECT_EQ(UsmReuseInfo::notLimited, memoryManager->usmReuseInfo.getLimitAllocationsReuseThreshold());
}
{
DebugManagerStateRestore restore;
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(80);
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(70);
memoryManager->initUsmReuseLimits();
const auto systemSharedMemory = memoryManager->getSystemSharedMemory(device->getRootDeviceIndex());
const auto expectedLimitThreshold = static_cast<uint64_t>(0.8 * systemSharedMemory);
const auto expectedLimitThreshold = static_cast<uint64_t>(0.7 * systemSharedMemory);
EXPECT_EQ(expectedLimitThreshold, memoryManager->usmReuseInfo.getLimitAllocationsReuseThreshold());
}
{
DebugManagerStateRestore restore;
debugManager.flags.ExperimentalUSMAllocationReuseLimitThreshold.set(-1);
memoryManager->initUsmReuseLimits();
const auto expectedLimitThreshold = std::numeric_limits<uint64_t>::max();
const auto systemSharedMemory = memoryManager->getSystemSharedMemory(device->getRootDeviceIndex());
const auto expectedLimitThreshold = static_cast<uint64_t>(0.8 * systemSharedMemory);
EXPECT_EQ(expectedLimitThreshold, memoryManager->usmReuseInfo.getLimitAllocationsReuseThreshold());
}
}