Add unit test for append memory prefetch for regular command list

Related-To: NEO-6740

Signed-off-by: Milczarek, Slawomir <slawomir.milczarek@intel.com>
This commit is contained in:
Milczarek, Slawomir
2022-08-04 01:45:19 +00:00
committed by Compute-Runtime-Automation
parent 3de620af15
commit 601ace6a25
3 changed files with 78 additions and 1 deletions

View File

@ -8,10 +8,18 @@
#pragma once
#include "shared/source/memory_manager/prefetch_manager.h"
#include "shared/source/memory_manager/unified_memory_manager.h"
using namespace NEO;
class MockPrefetchManager : public PrefetchManager {
public:
using PrefetchManager::allocations;
void migrateAllocationsToGpu(SVMAllocsManager &unifiedMemoryManager, Device &device) override {
PrefetchManager::migrateAllocationsToGpu(unifiedMemoryManager, device);
migrateAllocationsToGpuCalled = true;
}
bool migrateAllocationsToGpuCalled = false;
};