Free allocations in 2MB heaps

Related-To: NEO-5750
Signed-off-by: Maciej Dziuban <maciej.dziuban@intel.com>
This commit is contained in:
Maciej Dziuban
2021-04-19 18:58:10 +00:00
committed by Compute-Runtime-Automation
parent 0607118f15
commit 495841812f
5 changed files with 57 additions and 5 deletions

View File

@ -48,8 +48,6 @@ class MockGfxPartition : public GfxPartition {
}
}
MOCK_METHOD(void, freeGpuAddressRange, (uint64_t gpuAddress, size_t size), (override));
static std::array<HeapIndex, static_cast<uint32_t>(HeapIndex::TOTAL_HEAPS)> allHeapNames;
OSMemory::ReservedCpuAddressRange reservedCpuAddressRange;
@ -58,6 +56,11 @@ class MockGfxPartition : public GfxPartition {
const uint64_t mockGpuVa = std::numeric_limits<uint64_t>::max();
};
struct GmockGfxPartition : MockGfxPartition {
using MockGfxPartition::MockGfxPartition;
MOCK_METHOD(void, freeGpuAddressRange, (uint64_t gpuAddress, size_t size), (override));
};
class MockGfxPartitionBasic : public GfxPartition {
public:
MockGfxPartitionBasic() : GfxPartition(reservedCpuAddressRange) {}