mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-21 17:29:14 +08:00
54eee2a88b
With this commit OpenCL will track if external host memory is used from few threads and will secure to update task count in all threads before destroing allocation. Resolves: NEO-6807 Signed-off-by: Maciej Plewka <maciej.plewka@intel.com>
121 lines
4.1 KiB
C++
121 lines
4.1 KiB
C++
/*
|
|
* Copyright (C) 2018-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "graphics_allocation.h"
|
|
|
|
#include "shared/source/command_stream/command_stream_receiver.h"
|
|
#include "shared/source/gmm_helper/gmm.h"
|
|
#include "shared/source/gmm_helper/gmm_helper.h"
|
|
#include "shared/source/helpers/aligned_memory.h"
|
|
#include "shared/source/memory_manager/memory_manager.h"
|
|
#include "shared/source/utilities/logger.h"
|
|
|
|
namespace NEO {
|
|
void GraphicsAllocation::setAllocationType(AllocationType allocationType) {
|
|
this->allocationType = allocationType;
|
|
FileLoggerInstance().logAllocation(this);
|
|
}
|
|
|
|
GraphicsAllocation::GraphicsAllocation(uint32_t rootDeviceIndex, size_t numGmms, AllocationType allocationType, void *cpuPtrIn, uint64_t gpuAddress,
|
|
uint64_t baseAddress, size_t sizeIn, MemoryPool::Type pool, size_t maxOsContextCount)
|
|
: rootDeviceIndex(rootDeviceIndex),
|
|
gpuBaseAddress(baseAddress),
|
|
gpuAddress(GmmHelper::canonize(gpuAddress)),
|
|
size(sizeIn),
|
|
cpuPtr(cpuPtrIn),
|
|
memoryPool(pool),
|
|
allocationType(allocationType),
|
|
usageInfos(maxOsContextCount) {
|
|
gmms.resize(numGmms);
|
|
}
|
|
|
|
GraphicsAllocation::GraphicsAllocation(uint32_t rootDeviceIndex, size_t numGmms, AllocationType allocationType, void *cpuPtrIn, size_t sizeIn,
|
|
osHandle sharedHandleIn, MemoryPool::Type pool, size_t maxOsContextCount)
|
|
: rootDeviceIndex(rootDeviceIndex),
|
|
gpuAddress(GmmHelper::canonize(castToUint64(cpuPtrIn))),
|
|
size(sizeIn),
|
|
cpuPtr(cpuPtrIn),
|
|
memoryPool(pool),
|
|
allocationType(allocationType),
|
|
usageInfos(maxOsContextCount) {
|
|
sharingInfo.sharedHandle = sharedHandleIn;
|
|
gmms.resize(numGmms);
|
|
}
|
|
|
|
GraphicsAllocation::~GraphicsAllocation() = default;
|
|
|
|
void GraphicsAllocation::updateTaskCount(uint32_t newTaskCount, uint32_t contextId) {
|
|
if (usageInfos[contextId].taskCount == objectNotUsed) {
|
|
registeredContextsNum++;
|
|
}
|
|
if (newTaskCount == objectNotUsed) {
|
|
registeredContextsNum--;
|
|
}
|
|
usageInfos[contextId].taskCount = newTaskCount;
|
|
}
|
|
|
|
std::string GraphicsAllocation::getAllocationInfoString() const {
|
|
return "";
|
|
}
|
|
|
|
uint32_t GraphicsAllocation::getUsedPageSize() const {
|
|
switch (this->memoryPool) {
|
|
case MemoryPool::System64KBPages:
|
|
case MemoryPool::System64KBPagesWith32BitGpuAddressing:
|
|
case MemoryPool::LocalMemory:
|
|
return MemoryConstants::pageSize64k;
|
|
default:
|
|
return MemoryConstants::pageSize;
|
|
}
|
|
}
|
|
|
|
bool GraphicsAllocation::isAllocationLockable() const {
|
|
auto gmm = getDefaultGmm();
|
|
if (!gmm) {
|
|
return true;
|
|
}
|
|
return 0 == gmm->resourceParams.Flags.Info.NotLockable;
|
|
}
|
|
|
|
void GraphicsAllocation::setAubWritable(bool writable, uint32_t banks) {
|
|
UNRECOVERABLE_IF(banks == 0);
|
|
aubInfo.aubWritable = static_cast<uint32_t>(setBits(aubInfo.aubWritable, writable, banks));
|
|
}
|
|
|
|
bool GraphicsAllocation::isAubWritable(uint32_t banks) const {
|
|
return isAnyBitSet(aubInfo.aubWritable, banks);
|
|
}
|
|
|
|
void GraphicsAllocation::setTbxWritable(bool writable, uint32_t banks) {
|
|
UNRECOVERABLE_IF(banks == 0);
|
|
aubInfo.tbxWritable = static_cast<uint32_t>(setBits(aubInfo.tbxWritable, writable, banks));
|
|
}
|
|
|
|
bool GraphicsAllocation::isCompressionEnabled() const {
|
|
return (getDefaultGmm() && getDefaultGmm()->isCompressionEnabled);
|
|
}
|
|
|
|
bool GraphicsAllocation::isTbxWritable(uint32_t banks) const {
|
|
return isAnyBitSet(aubInfo.tbxWritable, banks);
|
|
}
|
|
|
|
void GraphicsAllocation::prepareHostPtrForResidency(CommandStreamReceiver *csr) {
|
|
if (hostPtrTaskCountAssignment > 0) {
|
|
auto allocTaskCount = getTaskCount(csr->getOsContext().getContextId());
|
|
auto currentTaskCount = *csr->getTagAddress() + 1;
|
|
if (currentTaskCount > allocTaskCount) {
|
|
updateTaskCount(currentTaskCount, csr->getOsContext().getContextId());
|
|
hostPtrTaskCountAssignment--;
|
|
}
|
|
}
|
|
}
|
|
|
|
constexpr uint32_t GraphicsAllocation::objectNotUsed;
|
|
constexpr uint32_t GraphicsAllocation::objectNotResident;
|
|
constexpr uint32_t GraphicsAllocation::objectAlwaysResident;
|
|
} // namespace NEO
|