mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-23 11:03:02 +08:00
5e56c8e25d
when allocation was used but not by registered engines then we can destroy it Related-To: NEO-3857 Change-Id: Ibf6a92603e9317d3355283adecd5e156d7b5ab0a Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
42 lines
1.6 KiB
C++
42 lines
1.6 KiB
C++
/*
|
|
* Copyright (C) 2018-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "runtime/memory_manager/deferrable_allocation_deletion.h"
|
|
|
|
#include "runtime/command_stream/command_stream_receiver.h"
|
|
#include "runtime/helpers/engine_control.h"
|
|
#include "runtime/memory_manager/memory_manager.h"
|
|
#include "runtime/os_interface/os_context.h"
|
|
|
|
namespace NEO {
|
|
|
|
DeferrableAllocationDeletion::DeferrableAllocationDeletion(MemoryManager &memoryManager, GraphicsAllocation &graphicsAllocation) : memoryManager(memoryManager),
|
|
graphicsAllocation(graphicsAllocation) {}
|
|
bool DeferrableAllocationDeletion::apply() {
|
|
if (graphicsAllocation.isUsed()) {
|
|
bool isStillUsed = false;
|
|
for (auto &engine : memoryManager.getRegisteredEngines()) {
|
|
auto contextId = engine.osContext->getContextId();
|
|
if (graphicsAllocation.isUsedByOsContext(contextId)) {
|
|
auto currentContextTaskCount = *engine.commandStreamReceiver->getTagAddress();
|
|
if (graphicsAllocation.getTaskCount(contextId) <= currentContextTaskCount) {
|
|
graphicsAllocation.releaseUsageInOsContext(contextId);
|
|
} else {
|
|
isStillUsed = true;
|
|
engine.commandStreamReceiver->flushBatchedSubmissions();
|
|
}
|
|
}
|
|
}
|
|
if (isStillUsed) {
|
|
return false;
|
|
}
|
|
}
|
|
memoryManager.freeGraphicsMemory(&graphicsAllocation);
|
|
return true;
|
|
}
|
|
} // namespace NEO
|