intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-24 21:18:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: initial support for single temporary allocations list

Browse Source 
Related-To: NEO-14641

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2025-05-21 11:35:07 +00:00
committed by Compute-Runtime-Automation
parent f25fa0ad27
commit 91940d2a79
14 changed files with 562 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
shared/source/memory_manager/internal_allocation_storage.cpp
View File

@@ -1,5 +1,5 @@
/*
* Copyright (C) 2018-2024 Intel Corporation
* Copyright (C) 2018-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
@@ -27,14 +27,22 @@ void InternalAllocationStorage::storeAllocation(std::unique_ptr<GraphicsAllocati
storeAllocationWithTaskCount(std::move(gfxAllocation), allocationUsage, taskCount);
}
void InternalAllocationStorage::storeAllocationWithTaskCount(std::unique_ptr<GraphicsAllocation> &&gfxAllocation, uint32_t allocationUsage, TaskCountType taskCount) {
auto memoryManager = commandStreamReceiver.getMemoryManager();
auto osContextId = commandStreamReceiver.getOsContext().getContextId();
if (allocationUsage == TEMPORARY_ALLOCATION && memoryManager->isSingleTemporaryAllocationsListEnabled()) {
memoryManager->storeTemporaryAllocation(std::move(gfxAllocation), osContextId, taskCount);
return;
}
if (allocationUsage == REUSABLE_ALLOCATION) {
if (debugManager.flags.DisableResourceRecycling.get()) {
commandStreamReceiver.getMemoryManager()->freeGraphicsMemory(gfxAllocation.release());
memoryManager->freeGraphicsMemory(gfxAllocation.release());
return;
}
}
auto &allocationsList = allocationLists[allocationUsage];
gfxAllocation->updateTaskCount(taskCount, commandStreamReceiver.getOsContext().getContextId());
gfxAllocation->updateTaskCount(taskCount, osContextId);
allocationsList.pushTailOne(*gfxAllocation.release());
}
@@ -44,6 +52,12 @@ void InternalAllocationStorage::cleanAllocationList(TaskCountType waitTaskCount,
void InternalAllocationStorage::freeAllocationsList(TaskCountType waitTaskCount, AllocationsList &allocationsList) {
auto memoryManager = commandStreamReceiver.getMemoryManager();
if (&allocationsList == &allocationLists[TEMPORARY_ALLOCATION] && memoryManager->isSingleTemporaryAllocationsListEnabled()) {
memoryManager->cleanTemporaryAllocations(commandStreamReceiver, waitTaskCount);
return;
}
auto lock = memoryManager->getHostPtrManager()->obtainOwnership();
GraphicsAllocation *curr = allocationsList.detachNodes();
@@ -70,10 +84,26 @@ std::unique_ptr<GraphicsAllocation> InternalAllocationStorage::obtainReusableAll
}
std::unique_ptr<GraphicsAllocation> InternalAllocationStorage::obtainTemporaryAllocationWithPtr(size_t requiredSize, const void *requiredPtr, AllocationType allocationType) {
auto memoryManager = commandStreamReceiver.getMemoryManager();
if (memoryManager->isSingleTemporaryAllocationsListEnabled()) {
return memoryManager->obtainTemporaryAllocationWithPtr(&commandStreamReceiver, requiredSize, requiredPtr, allocationType);
}
auto allocation = allocationLists[TEMPORARY_ALLOCATION].detachAllocation(requiredSize, requiredPtr, &commandStreamReceiver, allocationType);
return allocation;
}
AllocationsList &InternalAllocationStorage::getTemporaryAllocations() {
auto memoryManager = commandStreamReceiver.getMemoryManager();
if (memoryManager->isSingleTemporaryAllocationsListEnabled()) {
return memoryManager->getTemporaryAllocationsList();
}
return allocationLists[TEMPORARY_ALLOCATION];
}
DeviceBitfield InternalAllocationStorage::getDeviceBitfield() const {
return commandStreamReceiver.getOsContext().getDeviceBitfield();
}