From e7573af28f1c0cd9553ce8607bcab057844b63f7 Mon Sep 17 00:00:00 2001 From: "Milczarek, Slawomir" Date: Tue, 28 May 2019 16:24:26 +0200 Subject: [PATCH] Add freeMemory to AubManager Related-To: NEO-3231 Change-Id: Iaf7015cfa2eacd5f2eb0537e10c1c1950d52e1eb Signed-off-by: Milczarek, Slawomir --- third_party/aub_stream/headers/aub_manager.h | 1 + unit_tests/mocks/mock_aub_manager.h | 2 ++ 2 files changed, 3 insertions(+) diff --git a/third_party/aub_stream/headers/aub_manager.h b/third_party/aub_stream/headers/aub_manager.h index 6f73aa4e1a..57096b4f18 100644 --- a/third_party/aub_stream/headers/aub_manager.h +++ b/third_party/aub_stream/headers/aub_manager.h @@ -26,6 +26,7 @@ class AubManager { virtual void addComment(const char *message) = 0; virtual void writeMemory(uint64_t gfxAddress, const void *memory, size_t size, uint32_t memoryBanks, int hint, size_t pageSize) = 0; + virtual void freeMemory(uint64_t gfxAddress, size_t size) = 0; static AubManager *create(uint32_t productFamily, uint32_t devicesCount, uint64_t memoryBankSize, bool localMemorySupported, uint32_t streamMode); }; diff --git a/unit_tests/mocks/mock_aub_manager.h b/unit_tests/mocks/mock_aub_manager.h index 386f6abb34..66a62741ee 100644 --- a/unit_tests/mocks/mock_aub_manager.h +++ b/unit_tests/mocks/mock_aub_manager.h @@ -89,6 +89,8 @@ class MockAubManager : public aub_stream::AubManager { hintToWriteMemory = hint; } + void freeMemory(uint64_t gfxAddress, size_t size) override {} + uint32_t openCalledCnt = 0; std::string fileName = ""; bool closeCalled = false;