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

fix: huge host and shared allocation

Browse Source 
Related-To: NEO-7229
Signed-off-by: Fabian Zwolinski <fabian.zwolinski@intel.com>
This commit is contained in:
Fabian Zwolinski
2023-07-05 19:33:33 +00:00
committed by Compute-Runtime-Automation
parent 83bd33befc
commit 1a398c7199
2 changed files with 44 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
shared/test/unit_test/os_interface/windows/wddm_memory_manager_tests.cpp
View File

@@ -57,6 +57,10 @@ class MockAllocateGraphicsMemoryWithAlignmentWddm : public MemoryManagerCreate<W
return true;
}
size_t getHugeGfxMemoryChunkSize(GfxMemoryAllocationMethod allocationMethod) const override {
return hugeGfxMemoryChunkSize;
}
size_t hugeGfxMemoryChunkSize = WddmMemoryManager::getHugeGfxMemoryChunkSize(preferredAllocationMethod);
};
class WddmMemoryManagerTests : public ::testing::Test {
@@ -291,6 +295,7 @@ TEST_F(WddmMemoryManagerAllocPathTests, GivenValidAllocationWithFailingCreateInt
memoryManager->freeGraphicsMemory(graphicsAllocation);
}
TEST_F(WddmMemoryManagerTests, GivenAllocationWhenAllocationIsFreeThenFreeToGmmClientContextCalled) {
NEO::AllocationData allocData = {};
allocData.type = NEO::AllocationType::BUFFER;
@@ -302,6 +307,22 @@ TEST_F(WddmMemoryManagerTests, GivenAllocationWhenAllocationIsFreeThenFreeToGmmC
auto graphicsAllocation = memoryManager->allocateGraphicsMemoryUsingKmdAndMapItToCpuVA(allocData, false);
auto gmmHelper = executionEnvironment->rootDeviceEnvironments[0]->getGmmHelper();
memoryManager->freeGraphicsMemory(graphicsAllocation);
EXPECT_GT(reinterpret_cast<MockGmmClientContextBase *>(gmmHelper->getClientContext())->freeGpuVirtualAddressCalled, 0u);
}
TEST_F(WddmMemoryManagerTests, givenAllocateGraphicsMemoryUsingKmdAndMapItToCpuVAWhenCreatingHostAllocationThenGpuAndCpuAddressesAreEqual) {
memoryManager->hugeGfxMemoryChunkSize = MemoryConstants::pageSize64k;
NEO::AllocationData allocData = {};
allocData.type = NEO::AllocationType::BUFFER_HOST_MEMORY;
allocData.size = 2ULL * MemoryConstants::pageSize64k;
allocData.forceKMDAllocation = false;
allocData.makeGPUVaDifferentThanCPUPtr = false;
memoryManager->callBaseAllocateGraphicsMemoryUsingKmdAndMapItToCpuVA = true;
auto graphicsAllocation = memoryManager->allocateGraphicsMemoryUsingKmdAndMapItToCpuVA(allocData, true);
auto gmmHelper = executionEnvironment->rootDeviceEnvironments[0]->getGmmHelper();
EXPECT_EQ(graphicsAllocation->getGpuAddress(), reinterpret_cast<uintptr_t>(graphicsAllocation->getUnderlyingBuffer()));
memoryManager->freeGraphicsMemory(graphicsAllocation);
EXPECT_GT(reinterpret_cast<MockGmmClientContextBase *>(gmmHelper->getClientContext())->freeGpuVirtualAddressCalled, 0u);
}