mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-22 10:17:01 +08:00
32 lines
1.1 KiB
C++
32 lines
1.1 KiB
C++
/*
|
|
* Copyright (C) 2019-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
namespace NEO {
|
|
|
|
class MemoryManager;
|
|
class GraphicsAllocation;
|
|
class InternalAllocationStorage;
|
|
|
|
class HeapHelper {
|
|
public:
|
|
HeapHelper(MemoryManager *memManager, InternalAllocationStorage *storageForReuse, bool isMultiOsContextCapable) : storageForReuse(storageForReuse),
|
|
memManager(memManager),
|
|
isMultiOsContextCapable(isMultiOsContextCapable) {}
|
|
GraphicsAllocation *getHeapAllocation(uint32_t heapType, size_t heapSize, size_t alignment, uint32_t rootDeviceIndex);
|
|
void storeHeapAllocation(GraphicsAllocation *heapAllocation);
|
|
|
|
protected:
|
|
InternalAllocationStorage *storageForReuse = nullptr;
|
|
MemoryManager *memManager = nullptr;
|
|
bool isMultiOsContextCapable = false;
|
|
};
|
|
} // namespace NEO
|