Add ULTs for OsAgnosticMemoryManager

Related-To: NEO-2877

Change-Id: Id80fd66ced9d711ff74d85fa48741c95f9f750bb
Signed-off-by: Igor Venevtsev <igor.venevtsev@intel.com>

Author: ivvenevt

unit_tests/memory_manager/memory_manager_tests.cpp

@@ -579,6 +579,43 @@ TEST(OsAgnosticMemoryManager, givenDefaultMemoryManagerWhenForce32bitallocationI
     EXPECT_TRUE(memoryManager.peekForce32BitAllocations());
 }
 
+TEST(OsAgnosticMemoryManager, givenMemoryManagerWhenAskedFor32BitAllocationWhenLimitedAllocationIsEnabledThenGpuRangeFromExternalHeapIsAllocatiedAndBaseAddressIsSet) {
+    if (is32bit) {
+        GTEST_SKIP();
+    }
+
+    ExecutionEnvironment *executionEnvironment = platformImpl->peekExecutionEnvironment();
+    MockMemoryManager memoryManager(*executionEnvironment);
+    memoryManager.setForce32BitAllocations(true);
+    memoryManager.forceLimitedRangeAllocator(0xFFFFFFFFF);
+
+    AllocationData allocationData;
+    MockMemoryManager::getAllocationData(allocationData, {MemoryConstants::pageSize, GraphicsAllocation::AllocationType::BUFFER}, nullptr, StorageInfo{});
+    auto gfxAllocation = memoryManager.allocateGraphicsMemoryWithAlignment(allocationData);
+    ASSERT_NE(gfxAllocation, nullptr);
+    EXPECT_NE(gfxAllocation->getGpuBaseAddress(), 0ull);
+    EXPECT_EQ(gfxAllocation->getGpuBaseAddress(), memoryManager.getExternalHeapBaseAddress());
+    memoryManager.freeGraphicsMemory(gfxAllocation);
+}
+
+TEST(OsAgnosticMemoryManager, givenMemoryManagerWhenAskedForNon32BitAllocationWhenLimitedAllocationIsEnabledThenGpuRangeFromiStandardHeapIsAllocatiedAndBaseAddressIsNotSet) {
+    if (is32bit) {
+        GTEST_SKIP();
+    }
+
+    ExecutionEnvironment *executionEnvironment = platformImpl->peekExecutionEnvironment();
+    MockMemoryManager memoryManager(*executionEnvironment);
+    memoryManager.forceLimitedRangeAllocator(0xFFFFFFFFF);
+
+    AllocationData allocationData;
+    MockMemoryManager::getAllocationData(allocationData, {MemoryConstants::pageSize, GraphicsAllocation::AllocationType::BUFFER}, nullptr, StorageInfo{});
+    auto gfxAllocation = memoryManager.allocateGraphicsMemoryWithAlignment(allocationData);
+    ASSERT_NE(gfxAllocation, nullptr);
+    EXPECT_EQ(gfxAllocation->getGpuBaseAddress(), 0ull);
+    EXPECT_EQ(gfxAllocation->getGpuAddress(), memoryManager.gfxPartition->getHeapLimit(HeapIndex::HEAP_STANDARD) + 1 - GfxPartition::heapGranularity - MemoryConstants::pageSize);
+    memoryManager.freeGraphicsMemory(gfxAllocation);
+}
+
 TEST(OsAgnosticMemoryManager, givenDefaultMemoryManagerWhenAllocateGraphicsMemoryForImageIsCalledThenGraphicsAllocationIsReturned) {
     ExecutionEnvironment *executionEnvironment = platformImpl->peekExecutionEnvironment();
     MockMemoryManager memoryManager(*executionEnvironment);

unit_tests/mocks/mock_memory_manager.h

@@ -31,6 +31,7 @@ class MockMemoryManager : public MemoryManagerCreate<OsAgnosticMemoryManager> {
   public:
     using MemoryManager::allocateGraphicsMemoryForNonSvmHostPtr;
     using MemoryManager::allocateGraphicsMemoryInPreferredPool;
+    using MemoryManager::allocateGraphicsMemoryWithAlignment;
     using MemoryManager::allocateGraphicsMemoryWithProperties;
     using MemoryManager::AllocationData;
     using MemoryManager::createGraphicsAllocation;
@@ -100,6 +101,8 @@ class MockMemoryManager : public MemoryManagerCreate<OsAgnosticMemoryManager> {
     GraphicsAllocation *allocate32BitGraphicsMemory(size_t size, const void *ptr, GraphicsAllocation::AllocationType allocationType);
     GraphicsAllocation *allocate32BitGraphicsMemoryImpl(const AllocationData &allocationData) override;
 
+    void forceLimitedRangeAllocator(uint64_t range) { gfxPartition->init(range, 0); }
+
     uint32_t freeGraphicsMemoryCalled = 0u;
     uint32_t unlockResourceCalled = 0u;
     uint32_t lockResourceCalled = 0u;