46 lines
994 B
C++
46 lines
994 B
C++
/*
|
|
* Copyright (C) 2019-2024 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "shared/source/helpers/constants.h"
|
|
#include "shared/source/helpers/non_copyable_or_moveable.h"
|
|
|
|
#include <mutex>
|
|
|
|
namespace NEO {
|
|
|
|
class CommandStreamReceiver;
|
|
class Device;
|
|
class GraphicsAllocation;
|
|
class MemoryManager;
|
|
|
|
class SyncBufferHandler : NonCopyableOrMovableClass {
|
|
public:
|
|
~SyncBufferHandler();
|
|
|
|
SyncBufferHandler(Device &device);
|
|
|
|
template <typename KernelT>
|
|
void prepareForEnqueue(size_t workGroupsCount, KernelT &kernel);
|
|
void makeResident(CommandStreamReceiver &csr);
|
|
|
|
std::pair<GraphicsAllocation *, size_t> obtainAllocationAndOffset(size_t requiredSize);
|
|
|
|
protected:
|
|
void allocateNewBuffer();
|
|
|
|
Device &device;
|
|
MemoryManager &memoryManager;
|
|
GraphicsAllocation *graphicsAllocation;
|
|
const size_t bufferSize = 64 * MemoryConstants::kiloByte;
|
|
size_t usedBufferSize = 0;
|
|
std::mutex mutex;
|
|
};
|
|
|
|
} // namespace NEO
|