fix: add missing AlignLocalMemoryVaTo2MB flag support

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2024-08-01 11:49:25 +00:00
committed by Compute-Runtime-Automation
parent 7ca0210ef2
commit 5ccfd6f2be
3 changed files with 23 additions and 8 deletions

View File

@@ -1,5 +1,5 @@
/*
* Copyright (C) 2019-2023 Intel Corporation
* Copyright (C) 2019-2024 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
@@ -21,6 +21,10 @@ GraphicsAllocation *HeapHelper::getHeapAllocation(uint32_t heapType, size_t heap
allocationType = AllocationType::internalHeap;
}
if (NEO::debugManager.flags.AlignLocalMemoryVaTo2MB.get() == 1) {
alignment = MemoryConstants::pageSize2M;
}
auto allocation = this->storageForReuse->obtainReusableAllocation(heapSize, allocationType);
if (allocation) {
return allocation.release();