Don't allow 32bit allocation for SVM allocation type

Change-Id: I2fbae4ce3be956a386bdc22c9b129f37d75c8e8f
Signed-off-by: Mateusz Jablonski <[email protected]>

Author: JablonskiMateusz

runtime/memory_manager/memory_manager.cpp

@@ -205,14 +205,21 @@ bool MemoryManager::getAllocationData(AllocationData &allocationData, const Allo
     case GraphicsAllocation::AllocationType::PRINTF_SURFACE:
     case GraphicsAllocation::AllocationType::CONSTANT_SURFACE:
     case GraphicsAllocation::AllocationType::GLOBAL_SURFACE:
-    case GraphicsAllocation::AllocationType::SVM:
         allow64KbPages = true;
         allow32Bit = true;
         break;
     default:
         break;
     }
 
+    switch (properties.allocationType) {
+    case GraphicsAllocation::AllocationType::SVM:
+        allow64KbPages = true;
+        break;
+    default:
+        break;
+    }
+
     switch (properties.allocationType) {
     case GraphicsAllocation::AllocationType::BUFFER:
     case GraphicsAllocation::AllocationType::BUFFER_HOST_MEMORY:

unit_tests/memory_manager/memory_manager_allocate_in_preferred_pool_tests.inl

@@ -354,6 +354,13 @@ TEST(MemoryManagerTest, givenSvmAllocationTypeWhenGetAllocationDataIsCalledThenZ
     EXPECT_TRUE(allocData.flags.allocateMemory);
 }
 
+TEST(MemoryManagerTest, givenSvmAllocationTypeWhenGetAllocationDataIsCalledThen64kbPagesAreAllowedAnd32BitAllocationIsDisallowed) {
+    AllocationData allocData;
+    MockMemoryManager::getAllocationData(allocData, {1, GraphicsAllocation::AllocationType::SVM}, 0, nullptr);
+    EXPECT_TRUE(allocData.flags.allow64kbPages);
+    EXPECT_FALSE(allocData.flags.allow32Bit);
+}
+
 TEST(MemoryManagerTest, givenUndecidedTypeWhenGetAllocationDataIsCalledThenSystemMemoryIsRequested) {
     AllocationData allocData;
     MockMemoryManager::getAllocationData(allocData, {1, GraphicsAllocation::AllocationType::UNDECIDED}, 0, nullptr);