[13/n] Internal 4GB allocator.

- Add common function for all memory managers to obtain internal heap base.

Change-Id: Iacdaaf598e8786dad046b3f4550f9c80ce3f15d9

runtime/memory_manager/memory_manager.h

@@ -144,6 +144,8 @@ class MemoryManager {
 
     virtual uint64_t getMaxApplicationAddress() = 0;
 
+    virtual uint64_t getInternalHeapBaseAddress() = 0;
+
     virtual bool cleanAllocationList(uint32_t waitTaskCount, uint32_t allocationType);
 
     void freeAllocationsList(uint32_t waitTaskCount, AllocationsList &allocationsList);

runtime/memory_manager/os_agnostic_memory_manager.cpp

@@ -167,6 +167,10 @@ uint64_t OsAgnosticMemoryManager::getMaxApplicationAddress() {
     return MemoryConstants::max32BitAppAddress + static_cast<uint64_t>(is64bit) * (MemoryConstants::max64BitAppAddress - MemoryConstants::max32BitAppAddress);
 }
 
+uint64_t OsAgnosticMemoryManager::getInternalHeapBaseAddress() {
+    return this->allocator32Bit->getBase();
+}
+
 GraphicsAllocation *OsAgnosticMemoryManager::createGraphicsAllocation(OsHandleStorage &handleStorage, size_t hostPtrSize, const void *hostPtr) {
     auto allocation = new MemoryAllocation(false, 0, const_cast<void *>(hostPtr), reinterpret_cast<uint64_t>(hostPtr), hostPtrSize, counter++);
     allocation->fragmentsStorage = handleStorage;

runtime/memory_manager/os_agnostic_memory_manager.h

@@ -76,6 +76,7 @@ class OsAgnosticMemoryManager : public MemoryManager {
 
     uint64_t getSystemSharedMemory() override;
     uint64_t getMaxApplicationAddress() override;
+    uint64_t getInternalHeapBaseAddress() override;
 
     GraphicsAllocation *createGraphicsAllocation(OsHandleStorage &handleStorage, size_t hostPtrSize, const void *hostPtr) override;
 

runtime/os_interface/linux/drm_memory_manager.cpp

@@ -462,6 +462,10 @@ uint64_t DrmMemoryManager::getMaxApplicationAddress() {
     return MemoryConstants::max32BitAppAddress + (uint64_t)is64bit * (MemoryConstants::max64BitAppAddress - MemoryConstants::max32BitAppAddress);
 }
 
+uint64_t DrmMemoryManager::getInternalHeapBaseAddress() {
+    return this->internal32bitAllocator->getBase();
+}
+
 bool DrmMemoryManager::populateOsHandles(OsHandleStorage &handleStorage) {
     for (unsigned int i = 0; i < max_fragments_count; i++) {
         // If there is no fragment it means it already exists.

runtime/os_interface/linux/drm_memory_manager.h

@@ -62,6 +62,7 @@ class DrmMemoryManager : public MemoryManager {
 
     uint64_t getSystemSharedMemory() override;
     uint64_t getMaxApplicationAddress() override;
+    uint64_t getInternalHeapBaseAddress() override;
 
     bool populateOsHandles(OsHandleStorage &handleStorage) override;
     void cleanOsHandles(OsHandleStorage &handleStorage) override;

runtime/os_interface/windows/wddm_memory_manager.cpp

@@ -395,6 +395,10 @@ uint64_t WddmMemoryManager::getMaxApplicationAddress() {
     return wddm->getMaxApplicationAddress();
 }
 
+uint64_t WddmMemoryManager::getInternalHeapBaseAddress() {
+    return this->wddm->getGfxPartition().Heap32[1].Base;
+}
+
 bool WddmMemoryManager::makeResidentResidencyAllocations(ResidencyContainer *allocationsForResidency) {
 
     auto &residencyAllocations = allocationsForResidency ? *allocationsForResidency : this->residencyAllocations;

runtime/os_interface/windows/wddm_memory_manager.h

@@ -67,6 +67,7 @@ class WddmMemoryManager : public MemoryManager {
 
     uint64_t getSystemSharedMemory() override;
     uint64_t getMaxApplicationAddress() override;
+    uint64_t getInternalHeapBaseAddress() override;
 
     static void APIENTRY trimCallback(_Inout_ D3DKMT_TRIMNOTIFICATION *trimNotification);
 

unit_tests/memory_manager/memory_manager_tests.cpp

@@ -1133,6 +1133,11 @@ TEST(OsAgnosticMemoryManager, givenPointerAndSizeWhenCreateInternalAllocationIsC
     EXPECT_EQ(allocationSize, graphicsAllocation->getUnderlyingBufferSize());
     memoryManager.freeGraphicsMemory(graphicsAllocation);
 }
+TEST(OsAgnosticMemoryManager, givenDefaultOsAgnosticMemoryManagerWhenItIsQueriedForInternalHeapBaseThen32BitAllocatorBaseIsReturned) {
+    OsAgnosticMemoryManager memoryManager;
+    auto heapBase = memoryManager.allocator32Bit->getBase();
+    EXPECT_EQ(heapBase, memoryManager.getInternalHeapBaseAddress());
+}
 
 TEST_F(MemoryAllocatorTest, GivenSizeWhenGmmIsCreatedThenSuccess) {
     Gmm *gmm = Gmm::create(nullptr, 65536, false);

unit_tests/os_interface/linux/drm_memory_manager_tests.cpp

@@ -2115,3 +2115,10 @@ TEST(DrmMemoryManager, givenDisabledAsyncDeleterFlagWhenMemoryManagerIsCreatedTh
     EXPECT_FALSE(memoryManager.isAsyncDeleterEnabled());
     EXPECT_EQ(nullptr, memoryManager.getDeferredDeleter());
 }
+
+TEST(DrmMemoryManager, givenDefaultDrmMemoryManagerWhenItIsQueriedForInternalHeapBaseThenInternalHeapBaseIsReturned) {
+    std::unique_ptr<TestedDrmMemoryManager> memoryManager(new (std::nothrow) TestedDrmMemoryManager(Drm::get(0), true));
+    auto internalAllocator = memoryManager->getDrmInternal32BitAllocator();
+    auto heapBase = internalAllocator->getBase();
+    EXPECT_EQ(heapBase, memoryManager->getInternalHeapBaseAddress());
+}

unit_tests/os_interface/windows/wddm_memory_manager_tests.cpp

@@ -1787,6 +1787,14 @@ TEST(WddmMemoryManagerWithAsyncDeleterTest, givenMemoryManagerWithoutAsyncDelete
     EXPECT_EQ(1u, wddm->createAllocationResult.called);
 }
 
+TEST(WddmMemoryManagerDefaults, givenDefaultWddmMemoryManagerWhenItIsQueriedForInternalHeapBaseThenHeap1BaseIsReturned) {
+    WddmMock *wddm = new WddmMock;
+    wddm->callBaseDestroyAllocations = false;
+    MockWddmMemoryManager memoryManager(wddm);
+    auto heapBase = wddm->getGfxPartition().Heap32[1].Base;
+    EXPECT_EQ(heapBase, memoryManager.getInternalHeapBaseAddress());
+}
+
 HWTEST_F(MockWddmMemoryManagerTest, givenValidateAllocationFunctionWhenItIsCalledWithTripleAllocationThenSuccessIsReturned) {
     WddmMock *wddm = new WddmMock;
     EXPECT_TRUE(wddm->init<FamilyType>());