From fb856578a5d0ab1cd86977a20f19b8fde6b7cc07 Mon Sep 17 00:00:00 2001 From: Maciej Bielski Date: Wed, 12 Jun 2024 15:57:11 +0000 Subject: [PATCH] fix: restore overlooked debug flag usage Overlooked during recent refactor. ec21260fdafa76b926d9f213b1a80464e25f7ccf Related-To: NEO-9754 Signed-off-by: Maciej Bielski --- .../memory_manager/definitions/storage_info.cpp | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/shared/source/memory_manager/definitions/storage_info.cpp b/shared/source/memory_manager/definitions/storage_info.cpp index 7947c6d05a..e4cfcad69d 100644 --- a/shared/source/memory_manager/definitions/storage_info.cpp +++ b/shared/source/memory_manager/definitions/storage_info.cpp @@ -206,19 +206,24 @@ DeviceBitfield MemoryManager::computeStorageInfoMemoryBanks(const AllocationProp memoryBanks = (properties.flags.multiOsContextCapable ? allBanks : preferredBank); break; case AllocationType::buffer: - case AllocationType::svmGpu: + case AllocationType::svmGpu: { + auto granularity = properties.colouringGranularity; + if (debugManager.flags.MultiStorageGranularity.get() != -1) { + granularity = debugManager.flags.MultiStorageGranularity.get() * MemoryConstants::kiloByte; + } - DEBUG_BREAK_IF(properties.colouringPolicy == ColouringPolicy::deviceCountBased && properties.colouringGranularity != MemoryConstants::pageSize64k); + DEBUG_BREAK_IF(properties.colouringPolicy == ColouringPolicy::deviceCountBased && granularity != MemoryConstants::pageSize64k); if (this->supportsMultiStorageResources && properties.multiStorageResource && - properties.size >= deviceCount * properties.colouringGranularity && + properties.size >= deviceCount * granularity && properties.subDevicesBitfield.count() != 1u) { memoryBanks = (forcedMultiStoragePlacement == -1 ? allBanks : forcedMultiStoragePlacement); } memoryBanks = (properties.flags.readOnlyMultiStorage ? allBanks : memoryBanks); break; + } case AllocationType::unifiedSharedMemory: memoryBanks = (forcedMultiStoragePlacement == -1 ? allBanks : forcedMultiStoragePlacement); break;