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

test: add test for drm residency handler in multi root devices case

Browse Source 
Related-To: NEO-7920

Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2023-04-25 11:59:05 +00:00
committed by Compute-Runtime-Automation
parent a5bc67f51a
commit e48bb5ad6a
1 changed files with 31 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
shared/test/unit_test/os_interface/linux/drm_residency_handler_prelim_tests.cpp
View File

@@ -52,6 +52,10 @@ struct MockDrmMemoryOperationsHandlerBind : public DrmMemoryOperationsHandlerBin
return MemoryOperationsStatus::SUCCESS;
}
int evictImpl(OsContext *osContext, GraphicsAllocation &gfxAllocation, DeviceBitfield deviceBitfield) override {
EXPECT_EQ(this->rootDeviceIndex, gfxAllocation.getRootDeviceIndex());
return DrmMemoryOperationsHandlerBind::evictImpl(osContext, gfxAllocation, deviceBitfield);
}
};
template <uint32_t numRootDevices>
@@ -208,7 +212,8 @@ struct DrmMemoryOperationsHandlerBindFixture2 : public ::testing::Test {
memoryManager = std::make_unique<TestedDrmMemoryManager>(*executionEnvironment);
deviceDefault = devices[0].get();
device = devices[1].get();
mock = executionEnvironment->rootDeviceEnvironments[0]->osInterface->getDriverModel()->as<DrmQueryMock>();
mockDefault = executionEnvironment->rootDeviceEnvironments[0]->osInterface->getDriverModel()->as<DrmQueryMock>();
mock = executionEnvironment->rootDeviceEnvironments[1]->osInterface->getDriverModel()->as<DrmQueryMock>();
operationHandlerDefault = static_cast<DrmMemoryOperationsHandlerDefault *>(executionEnvironment->rootDeviceEnvironments[0]->memoryOperationsInterface.get());
operationHandler = static_cast<MockDrmMemoryOperationsHandlerBind *>(executionEnvironment->rootDeviceEnvironments[1]->memoryOperationsInterface.get());
memoryManagerBackup = executionEnvironment->memoryManager.release();
@@ -227,27 +232,28 @@ struct DrmMemoryOperationsHandlerBindFixture2 : public ::testing::Test {
protected:
ExecutionEnvironment *executionEnvironment = nullptr;
MockDevice *device;
MockDevice *deviceDefault;
MockDevice *device = nullptr;
MockDevice *deviceDefault = nullptr;
std::vector<std::unique_ptr<MockDevice>> devices;
std::unique_ptr<TestedDrmMemoryManager> memoryManager;
DrmMemoryOperationsHandlerDefault *operationHandlerDefault = nullptr;
MockDrmMemoryOperationsHandlerBind *operationHandler = nullptr;
DebugManagerStateRestore restorer;
DrmQueryMock *mock;
MemoryManager *memoryManagerBackup;
DrmQueryMock *mock = nullptr;
DrmQueryMock *mockDefault = nullptr;
MemoryManager *memoryManagerBackup = nullptr;
};
using DrmMemoryOperationsHandlerBindMultiRootDeviceTest2 = DrmMemoryOperationsHandlerBindFixture2<2u>;
TEST_F(DrmMemoryOperationsHandlerBindMultiRootDeviceTest2, givenOperationHandlersWhenRootDeviceIndexIsChangedThenEvictSucceeds) {
auto allocation = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{device->getRootDeviceIndex(), MemoryConstants::pageSize});
auto allocationDefult = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{deviceDefault->getRootDeviceIndex(), MemoryConstants::pageSize});
auto allocationDefault = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{deviceDefault->getRootDeviceIndex(), MemoryConstants::pageSize});
EXPECT_EQ(operationHandlerDefault->getRootDeviceIndex(), 0u);
EXPECT_EQ(operationHandler->getRootDeviceIndex(), 1u);
operationHandlerDefault->makeResident(device, ArrayRef<GraphicsAllocation *>(&allocationDefult, 1));
operationHandlerDefault->makeResident(device, ArrayRef<GraphicsAllocation *>(&allocationDefault, 1));
operationHandler->makeResident(device, ArrayRef<GraphicsAllocation *>(&allocation, 1));
operationHandlerDefault->setRootDeviceIndex(1u);
@@ -255,28 +261,38 @@ TEST_F(DrmMemoryOperationsHandlerBindMultiRootDeviceTest2, givenOperationHandler
EXPECT_EQ(operationHandlerDefault->getRootDeviceIndex(), 1u);
EXPECT_EQ(operationHandler->getRootDeviceIndex(), 0u);
EXPECT_EQ(operationHandlerDefault->evict(device, *allocationDefult), MemoryOperationsStatus::SUCCESS);
EXPECT_EQ(operationHandler->evict(device, *allocation), MemoryOperationsStatus::SUCCESS);
EXPECT_EQ(operationHandlerDefault->evict(device, *allocation), MemoryOperationsStatus::SUCCESS);
EXPECT_EQ(operationHandler->evict(device, *allocationDefault), MemoryOperationsStatus::SUCCESS);
memoryManager->freeGraphicsMemory(allocationDefult);
operationHandlerDefault->setRootDeviceIndex(0u);
operationHandler->setRootDeviceIndex(1u);
memoryManager->freeGraphicsMemory(allocationDefault);
memoryManager->freeGraphicsMemory(allocation);
}
using DrmMemoryOperationsHandlerBindTest = DrmMemoryOperationsHandlerBindFixture<1u>;
TEST_F(DrmMemoryOperationsHandlerBindTest, whenNoSpaceLeftOnDeviceThenEvictUnusedAllocations) {
TEST_F(DrmMemoryOperationsHandlerBindMultiRootDeviceTest2, whenNoSpaceLeftOnDeviceThenEvictUnusedAllocations) {
auto allocation = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{device->getRootDeviceIndex(), MemoryConstants::pageSize});
auto allocationDefault = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{deviceDefault->getRootDeviceIndex(), MemoryConstants::pageSize});
mock->context.vmBindReturn = -1;
mock->baseErrno = false;
mock->errnoRetVal = ENOSPC;
operationHandler->useBaseEvictUnused = false;
operationHandler->useBaseEvictUnused = true;
auto registeredAllocations = memoryManager->getSysMemAllocs();
EXPECT_EQ(2u, registeredAllocations.size());
EXPECT_EQ(allocation, registeredAllocations[0]);
EXPECT_EQ(allocationDefault, registeredAllocations[1]);
EXPECT_EQ(operationHandler->evictUnusedCalled, 0u);
operationHandler->makeResident(device, ArrayRef<GraphicsAllocation *>(&allocation, 1));
EXPECT_EQ(operationHandler->evictUnusedCalled, 1u);
memoryManager->freeGraphicsMemory(allocation);
memoryManager->freeGraphicsMemory(allocationDefault);
}
using DrmMemoryOperationsHandlerBindTest = DrmMemoryOperationsHandlerBindFixture<1u>;
TEST_F(DrmMemoryOperationsHandlerBindTest, givenObjectAlwaysResidentAndNotUsedWhenRunningOutOfMemoryThenUnusedAllocationIsNotUnbound) {
auto allocation = memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{device->getRootDeviceIndex(), MemoryConstants::pageSize});
@@ -1432,4 +1448,4 @@ TEST(DrmResidencyHandlerTests, whenQueryingForSetPairAvailableWithDebugKeySetToZ
EXPECT_FALSE(drm.isSetPairAvailable());
EXPECT_FALSE(drm.setPairAvailable);
EXPECT_EQ(0u, drm.context.setPairQueryCalled);
}
}