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

refactor: extract method for getting CB peer device allocation

Browse Source 
Signed-off-by: Naklicki, Mateusz <mateusz.naklicki@intel.com>
This commit is contained in:
Naklicki, Mateusz
2025-03-20 17:51:39 +00:00
committed by Compute-Runtime-Automation
parent 1d07198a4c
commit 890e5c4a5a
4 changed files with 74 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
level_zero/core/test/unit_tests/mocks/mock_cmdlist.h
View File

@@ -616,6 +616,7 @@ class MockCommandListCoreFamily : public CommandListCoreFamily<gfxCoreFamily> {
using BaseClass::dummyBlitWa;
using BaseClass::enableInOrderExecution;
using BaseClass::encodeMiFlush;
using BaseClass::getDeviceCounterAllocForResidency;
using BaseClass::ownedPrivateAllocations;
using BaseClass::taskCountUpdateFenceRequired;

57
level_zero/core/test/unit_tests/sources/cmdlist/test_cmdlist_8.cpp
View File

@@ -1872,5 +1872,62 @@ HWTEST2_F(CommandListCreate, givenNewSizeDifferentThanSizesInMapWhenAllocatingPr
neoDevice->getMemoryManager()->freeGraphicsMemory(commandList->commandContainer.getResidencyContainer()[0]);
}
HWTEST2_F(CommandListCreate, givenCounterDeviceAllocWhenGetDeviceCounterAllocForResidencyThenReturnCorrectAllocation, MatchAny) {
auto commandList = std::make_unique<MockCommandListCoreFamily<gfxCoreFamily>>();
commandList->initialize(device, NEO::EngineGroupType::renderCompute, 0u);
MockGraphicsAllocation counterDeviceAlloc(this->device->getRootDeviceIndex(), nullptr, 0x0);
auto result = commandList->getDeviceCounterAllocForResidency(&counterDeviceAlloc);
EXPECT_EQ(result, &counterDeviceAlloc);
}
HWTEST2_F(CommandListCreate, givenCounterDeviceAllocFromDifferentRootDeviceWhenGetDeviceCounterAllocForResidencyThenReturnPeerAllocation, MatchAny) {
auto commandList = std::make_unique<MockCommandListCoreFamily<gfxCoreFamily>>();
commandList->initialize(device, NEO::EngineGroupType::renderCompute, 0u);
uint32_t baseDeviceIndex = this->device->getRootDeviceIndex();
uint32_t peerDeviceIndex = baseDeviceIndex + 1;
MockGraphicsAllocation counterDeviceAlloc(baseDeviceIndex, reinterpret_cast<void *>(0x1234), 0x0u);
MockGraphicsAllocation peerAllocation(peerDeviceIndex, reinterpret_cast<void *>(0x5678), 0x0u);
ASSERT_NE(counterDeviceAlloc.getRootDeviceIndex(), peerAllocation.getRootDeviceIndex());
ASSERT_NE(counterDeviceAlloc.getGpuAddress(), peerAllocation.getGpuAddress());
auto result = commandList->getDeviceCounterAllocForResidency(&peerAllocation);
EXPECT_EQ(result->getGpuAddress(), peerAllocation.getGpuAddress());
}
HWTEST2_F(CommandListCreate, givenCounterDeviceAllocFromDifferentRootDeviceWhenGetDeviceCounterAllocForResidencyAndAllocIsAlreadyImportedThenReturnPeerAllocation, MatchAny) {
auto commandList = std::make_unique<MockCommandListCoreFamily<gfxCoreFamily>>();
commandList->initialize(device, NEO::EngineGroupType::renderCompute, 0u);
uint32_t baseDeviceIndex = this->device->getRootDeviceIndex();
uint32_t peerDeviceIndex = baseDeviceIndex + 1;
MockGraphicsAllocation counterDeviceAlloc(baseDeviceIndex, reinterpret_cast<void *>(0x1234), 0x0u);
MockGraphicsAllocation peerAllocation(peerDeviceIndex, reinterpret_cast<void *>(0x5678), 0x0u);
ASSERT_NE(counterDeviceAlloc.getRootDeviceIndex(), peerAllocation.getRootDeviceIndex());
ASSERT_NE(counterDeviceAlloc.getGpuAddress(), peerAllocation.getGpuAddress());
auto deviceImp = static_cast<DeviceImp *>(this->device);
NEO::SvmAllocationData allocData(peerDeviceIndex);
allocData.gpuAllocations.addAllocation(&peerAllocation);
deviceImp->peerCounterAllocations.allocations.insert({reinterpret_cast<void *>(peerAllocation.getGpuAddress()), allocData});
auto result = commandList->getDeviceCounterAllocForResidency(&peerAllocation);
EXPECT_EQ(result->getGpuAddress(), peerAllocation.getGpuAddress());
deviceImp->peerCounterAllocations.remove(allocData);
}
HWTEST2_F(CommandListCreate, givenNullptrPeerAllocationWhenGetDeviceCounterAllocForResidencyThenAbortIsThrown, MatchAny) {
auto commandList = std::make_unique<MockCommandListCoreFamily<gfxCoreFamily>>();
commandList->initialize(device, NEO::EngineGroupType::renderCompute, 0u);
uint32_t baseDeviceIndex = this->device->getRootDeviceIndex();
uint32_t peerDeviceIndex = baseDeviceIndex + 1;
MockGraphicsAllocation counterDeviceAlloc(peerDeviceIndex, nullptr, 0x0);
EXPECT_ANY_THROW(commandList->getDeviceCounterAllocForResidency(&counterDeviceAlloc));
}
} // namespace ult
} // namespace L0