fix: add media context helper

Related-To: NEO-16154

Signed-off-by: Grochowski, Stanislaw <stanislaw.grochowski@intel.com>
This commit is contained in:
Grochowski, Stanislaw
2025-10-16 08:30:25 +00:00
committed by Compute-Runtime-Automation
parent ae2dce4070
commit 6a8b20edb8
6 changed files with 40 additions and 5 deletions

View File

@@ -2137,19 +2137,33 @@ HWTEST_TEMPLATED_F(DrmMemoryManagerTest, whenCallingCreateAndReleaseMediaContext
auto &drm = static_cast<DrmMockCustom &>(memoryManager->getDrm(rootDeviceIndex));
drm.ioctlHelper.reset(mockIoctlHelper);
auto &productHelper = executionEnvironment->rootDeviceEnvironments[0]->getHelper<ProductHelper>();
bool mediaContextSupported = productHelper.isMediaContextSupported();
void *handle = nullptr;
EXPECT_EQ(0u, mockIoctlHelper->createMediaContextCalled);
EXPECT_EQ(0u, mockIoctlHelper->releaseMediaContextCalled);
memoryManager->createMediaContext(rootDeviceIndex, nullptr, 0, nullptr, 0, handle);
EXPECT_EQ(1u, mockIoctlHelper->createMediaContextCalled);
if (mediaContextSupported) {
EXPECT_EQ(1u, mockIoctlHelper->createMediaContextCalled);
} else {
EXPECT_EQ(0u, mockIoctlHelper->createMediaContextCalled);
}
EXPECT_EQ(mock->getVirtualMemoryAddressSpace(0), mockIoctlHelper->mediaContextVmId);
EXPECT_EQ(0u, mockIoctlHelper->releaseMediaContextCalled);
memoryManager->releaseMediaContext(rootDeviceIndex, handle);
EXPECT_EQ(1u, mockIoctlHelper->createMediaContextCalled);
EXPECT_EQ(1u, mockIoctlHelper->releaseMediaContextCalled);
if (mediaContextSupported) {
EXPECT_EQ(1u, mockIoctlHelper->createMediaContextCalled);
EXPECT_EQ(1u, mockIoctlHelper->releaseMediaContextCalled);
} else {
EXPECT_EQ(0u, mockIoctlHelper->createMediaContextCalled);
EXPECT_EQ(0u, mockIoctlHelper->releaseMediaContextCalled);
}
}
HWTEST_TEMPLATED_F(DrmMemoryManagerTest, whenCallingGetNumMediaThenCallIoctlHelper) {

View File

@@ -1304,3 +1304,7 @@ HWTEST2_F(ProductHelperTest, givenDG1ProductHelperWhenCanShareMemoryWithoutNTHan
HWTEST2_F(ProductHelperTest, givenNonDG1ProductHelperWhenCanShareMemoryWithoutNTHandleIsCalledThenTrueIsReturned, IsNotDG1) {
EXPECT_EQ(productHelper->canShareMemoryWithoutNTHandle(), 1u);
}
HWTEST_F(ProductHelperTest, givenProductHelperWhenAskingIsMediaContextSupportedThenFalseReturned) {
EXPECT_FALSE(productHelper->isMediaContextSupported());
}