mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
94 lines
2.9 KiB
C++
94 lines
2.9 KiB
C++
/*
|
|
* Copyright (C) 2017-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/command_stream/linear_stream.h"
|
|
#include "shared/source/helpers/aligned_memory.h"
|
|
#include "shared/source/helpers/basic_math.h"
|
|
#include "shared/source/helpers/constants.h"
|
|
#include "shared/source/helpers/ptr_math.h"
|
|
#include "shared/source/helpers/string.h"
|
|
#include "shared/source/memory_manager/graphics_allocation.h"
|
|
|
|
namespace NEO {
|
|
class GraphicsAllocation;
|
|
|
|
using HeapContainer = std::vector<GraphicsAllocation *>;
|
|
|
|
constexpr size_t defaultHeapSize = 64 * KB;
|
|
|
|
class IndirectHeap : public LinearStream {
|
|
typedef LinearStream BaseClass;
|
|
|
|
public:
|
|
enum Type {
|
|
DYNAMIC_STATE = 0,
|
|
INDIRECT_OBJECT,
|
|
SURFACE_STATE,
|
|
NUM_TYPES
|
|
};
|
|
|
|
IndirectHeap(void *graphicsAllocation, size_t bufferSize) : BaseClass(graphicsAllocation, bufferSize){};
|
|
IndirectHeap(GraphicsAllocation *graphicsAllocation) : BaseClass(graphicsAllocation) {}
|
|
IndirectHeap(GraphicsAllocation *graphicsAllocation, bool canBeUtilizedAs4GbHeap)
|
|
: BaseClass(graphicsAllocation), canBeUtilizedAs4GbHeap(canBeUtilizedAs4GbHeap) {}
|
|
|
|
// Disallow copy'ing
|
|
IndirectHeap(const IndirectHeap &) = delete;
|
|
IndirectHeap &operator=(const IndirectHeap &) = delete;
|
|
|
|
void align(size_t alignment);
|
|
uint64_t getHeapGpuStartOffset() const;
|
|
uint64_t getHeapGpuBase() const;
|
|
uint32_t getHeapSizeInPages() const;
|
|
uint32_t getBorderColorOffset() const;
|
|
void setBorderColor(void *borderColor, size_t size);
|
|
|
|
protected:
|
|
bool canBeUtilizedAs4GbHeap = false;
|
|
uint32_t borderColorOffset = std::numeric_limits<uint32_t>::max();
|
|
};
|
|
|
|
inline void IndirectHeap::align(size_t alignment) {
|
|
auto address = alignUp(ptrOffset(buffer, sizeUsed), alignment);
|
|
sizeUsed = ptrDiff(address, buffer);
|
|
}
|
|
|
|
inline uint32_t IndirectHeap::getHeapSizeInPages() const {
|
|
if (this->canBeUtilizedAs4GbHeap) {
|
|
return MemoryConstants::sizeOf4GBinPageEntities;
|
|
} else {
|
|
return (static_cast<uint32_t>(getMaxAvailableSpace()) + MemoryConstants::pageMask) / MemoryConstants::pageSize;
|
|
}
|
|
}
|
|
|
|
inline uint64_t IndirectHeap::getHeapGpuStartOffset() const {
|
|
if (this->canBeUtilizedAs4GbHeap) {
|
|
return this->graphicsAllocation->getGpuAddressToPatch();
|
|
} else {
|
|
return 0llu;
|
|
}
|
|
}
|
|
|
|
inline uint64_t IndirectHeap::getHeapGpuBase() const {
|
|
if (this->canBeUtilizedAs4GbHeap) {
|
|
return this->graphicsAllocation->getGpuBaseAddress();
|
|
} else {
|
|
return this->graphicsAllocation->getGpuAddress();
|
|
}
|
|
}
|
|
|
|
inline uint32_t IndirectHeap::getBorderColorOffset() const {
|
|
return borderColorOffset;
|
|
}
|
|
inline void IndirectHeap::setBorderColor(void *borderColor, size_t size) {
|
|
borderColorOffset = static_cast<uint32_t>(getUsed());
|
|
auto ptr = getSpace(size);
|
|
memcpy_s(ptr, size, borderColor, size);
|
|
}
|
|
} // namespace NEO
|