Media/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-09-15 13:01:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle not lockable allocations in HW with AUB mode

Browse Source 
don't lock not lockable allocation

Related-To: NEO-5733
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2021-05-14 16:34:47 +00:00
committed by Compute-Runtime-Automation
parent 058679b5c2
commit 8e691711e9
3 changed files with 72 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
opencl/test/unit_test/command_stream/aub_command_stream_receiver_2_tests.cpp
View File

@ -263,6 +263,76 @@ HWTEST_F(AubCommandStreamReceiverTests, givenAubCommandStreamReceiverWhenDefault
aubCsr->flush(batchBuffer, allocationsForResidency);
}
HWTEST_F(AubCommandStreamReceiverTests, givenNoCpuPtrAndNotLockableAllocationWhenGettingParametersForWriteThenLockResourceIsNotCalled) {
auto aubExecutionEnvironment = getEnvironment<MockAubCsr<FamilyType>>(true, true, true);
auto aubCsr = aubExecutionEnvironment->template getCsr<MockAubCsr<FamilyType>>();
auto mockMemoryManager = new MockMemoryManager();
auto memoryManagerBackup = aubExecutionEnvironment->executionEnvironment->memoryManager.release();
aubExecutionEnvironment->executionEnvironment->memoryManager.reset(mockMemoryManager);
EXPECT_EQ(0u, mockMemoryManager->lockResourceCalled);
constexpr uint64_t initGpuAddress = 1234;
constexpr size_t initSize = 10;
MockGraphicsAllocation allocation(nullptr, initGpuAddress, initSize);
allocation.setAllocationType(GraphicsAllocation::AllocationType::BUFFER);
allocation.overrideMemoryPool(MemoryPool::LocalMemory);
aubExecutionEnvironment->executionEnvironment->rootDeviceEnvironments[0]->initGmm();
MockGmm mockGmm(aubExecutionEnvironment->executionEnvironment->rootDeviceEnvironments[0]->getGmmClientContext(), nullptr, initSize, initSize, false, false, false, {});
mockGmm.resourceParams.Flags.Info.NotLockable = true;
allocation.setDefaultGmm(&mockGmm);
uint64_t gpuAddress{};
void *cpuAddress{};
size_t size{};
aubCsr->getParametersForWriteMemory(allocation, gpuAddress, cpuAddress, size);
EXPECT_EQ(nullptr, cpuAddress);
EXPECT_EQ(initGpuAddress, gpuAddress);
EXPECT_EQ(initSize, size);
EXPECT_EQ(0u, mockMemoryManager->lockResourceCalled);
aubExecutionEnvironment->executionEnvironment->memoryManager.reset(memoryManagerBackup);
}
HWTEST_F(AubCommandStreamReceiverTests, givenNoCpuPtrAndLockableAllocationWhenGettingParametersForWriteThenLockResourceIsCalled) {
auto aubExecutionEnvironment = getEnvironment<MockAubCsr<FamilyType>>(true, true, true);
auto aubCsr = aubExecutionEnvironment->template getCsr<MockAubCsr<FamilyType>>();
auto mockMemoryManager = new MockMemoryManager();
auto memoryManagerBackup = aubExecutionEnvironment->executionEnvironment->memoryManager.release();
aubExecutionEnvironment->executionEnvironment->memoryManager.reset(mockMemoryManager);
EXPECT_EQ(0u, mockMemoryManager->lockResourceCalled);
constexpr uint64_t initGpuAddress = 1234;
constexpr size_t initSize = 10;
MockGraphicsAllocation allocation(nullptr, initGpuAddress, initSize);
allocation.setAllocationType(GraphicsAllocation::AllocationType::BUFFER_HOST_MEMORY);
allocation.overrideMemoryPool(MemoryPool::LocalMemory);
aubExecutionEnvironment->executionEnvironment->rootDeviceEnvironments[0]->initGmm();
MockGmm mockGmm(aubExecutionEnvironment->executionEnvironment->rootDeviceEnvironments[0]->getGmmClientContext(), nullptr, initSize, initSize, false, false, false, {});
mockGmm.resourceParams.Flags.Info.NotLockable = false;
allocation.setDefaultGmm(&mockGmm);
uint64_t gpuAddress{};
void *cpuAddress{};
size_t size{};
aubCsr->getParametersForWriteMemory(allocation, gpuAddress, cpuAddress, size);
EXPECT_EQ(nullptr, cpuAddress);
EXPECT_EQ(initGpuAddress, gpuAddress);
EXPECT_EQ(initSize, size);
EXPECT_EQ(1u, mockMemoryManager->lockResourceCalled);
aubExecutionEnvironment->executionEnvironment->memoryManager.reset(memoryManagerBackup);
}
HWTEST_F(AubCommandStreamReceiverTests, givenAubCommandStreamReceiverWhenForcedFlattenBatchBufferAndImmediateDispatchModeThenExpectFlattenBatchBufferIsCalled) {
DebugManagerStateRestore dbgRestore;
DebugManager.flags.FlattenBatchBufferForAUBDump.set(true);