fix: use correct memoryBanks when makeResident is called in aub mode

Related-To: NEO-12869

Signed-off-by: Mateusz Hoppe <[email protected]>

Author: HoppeMateusz

shared/source/os_interface/aub_memory_operations_handler.cpp

@@ -42,11 +42,12 @@ MemoryOperationsStatus AubMemoryOperationsHandler::makeResident(Device *device,
             continue;
         }
 
+        auto memoryBanks = static_cast<uint32_t>(getMemoryBanksBitfield(allocation, device).to_ulong());
         uint64_t gpuAddress = device ? device->getGmmHelper()->decanonize(allocation->getGpuAddress()) : allocation->getGpuAddress();
         aub_stream::AllocationParams params(gpuAddress,
                                             allocation->getUnderlyingBuffer(),
                                             allocation->getUnderlyingBufferSize(),
-                                            allocation->storageInfo.getMemoryBanks(),
+                                            memoryBanks,
                                             hint,
                                             allocation->getUsedPageSize());
 

shared/test/unit_test/os_interface/aub_memory_operations_handler_tests.cpp

@@ -305,6 +305,27 @@ TEST_F(AubMemoryOperationsHandlerTests, givenLocalMemoryAndNonLocalMemoryAllocat
     device->getDefaultEngine().commandStreamReceiver = oldCsr;
 }
 
+TEST_F(AubMemoryOperationsHandlerTests, givenNonLocalMemoryAllocationWithStorageInfoNonZeroWhenMakeResidentCalledThenSystemMemoryBankIsPassedToWriteMemory) {
+    auto executionEnvironment = device->getExecutionEnvironment();
+    executionEnvironment->prepareRootDeviceEnvironments(1);
+    executionEnvironment->rootDeviceEnvironments[0]->setHwInfoAndInitHelpers(&hardwareInfo);
+    executionEnvironment->initializeMemoryManager();
+
+    MemoryAllocation allocation(0, 1u /*num gmms*/, AllocationType::unknown, nullptr, reinterpret_cast<void *>(0x1000), 0x1000u,
+                                MemoryConstants::pageSize, 0, MemoryPool::system64KBPages, false, false, MemoryManager::maxOsContextCount);
+    allocation.storageInfo.memoryBanks = 0x3u;
+    MockAubManager aubManager;
+    aubManager.storeAllocationParams = true;
+    getMemoryOperationsHandler()->setAubManager(&aubManager);
+    auto memoryOperationsInterface = getMemoryOperationsHandler();
+
+    allocPtr = &allocation;
+    memoryOperationsInterface->makeResident(device.get(), ArrayRef<GraphicsAllocation *>(&allocPtr, 1), false);
+
+    EXPECT_TRUE(aubManager.writeMemory2Called);
+    EXPECT_EQ(0u, aubManager.storedAllocationParams[0].memoryBanks);
+}
+
 TEST_F(AubMemoryOperationsHandlerTests, givenLocalMemoryNoncloneableAllocationWithManyBanksWhenGetMemoryBanksBitfieldThenSingleMemoryBankIsReturned) {
     auto executionEnvironment = device->getExecutionEnvironment();
     executionEnvironment->prepareRootDeviceEnvironments(1);