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

fix: pass vm_id to createMediaContext

Browse Source 
Related-To: NEO-9777

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2024-08-23 11:03:22 +00:00
committed by Compute-Runtime-Automation
parent b8f181d50e
commit aeb5e0d5ad
4 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
shared/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp
View File

@@ -1757,7 +1757,18 @@ TEST_F(DrmMemoryManagerTest, whenCallingAllocateAndReleaseInterruptThenCallIoctl
}
TEST_F(DrmMemoryManagerTest, whenCallingCreateAndReleaseMediaContextThenCallIoctlHelper) {
auto mockIoctlHelper = new MockIoctlHelper(*mock);
class MyMockIoctlHelper : public MockIoctlHelper {
public:
using MockIoctlHelper::MockIoctlHelper;
bool createMediaContext(uint32_t vmId, void *controlSharedMemoryBuffer, uint32_t controlSharedMemoryBufferSize, void *controlBatchBuffer, uint32_t controlBatchBufferSize, uint64_t &outDoorbell) override {
mediaContextVmId = vmId;
return MockIoctlHelper::createMediaContext(vmId, controlSharedMemoryBuffer, controlSharedMemoryBufferSize, controlBatchBuffer, controlBatchBufferSize, outDoorbell);
}
uint32_t mediaContextVmId = 0;
};
auto mockIoctlHelper = new MyMockIoctlHelper(*mock);
auto &drm = static_cast<DrmMockCustom &>(memoryManager->getDrm(rootDeviceIndex));
drm.ioctlHelper.reset(mockIoctlHelper);
@@ -1769,6 +1780,7 @@ TEST_F(DrmMemoryManagerTest, whenCallingCreateAndReleaseMediaContextThenCallIoct
memoryManager->createMediaContext(rootDeviceIndex, nullptr, 0, nullptr, 0, handle);
EXPECT_EQ(1u, mockIoctlHelper->createMediaContextCalled);
EXPECT_EQ(mock->getVirtualMemoryAddressSpace(0), mockIoctlHelper->mediaContextVmId);
EXPECT_EQ(0u, mockIoctlHelper->releaseMediaContextCalled);
memoryManager->releaseMediaContext(rootDeviceIndex, handle);