mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-21 09:14:47 +08:00
63fd26f6d6
Related-To: NEO-2941 Change-Id: Ia2362fd6b4e72ede02919152475f40b3edbc3658 Signed-off-by: Igor Venevtsev <igor.venevtsev@intel.com>
33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
/*
|
|
* Copyright (C) 2018-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "runtime/command_stream/scratch_space_controller.h"
|
|
|
|
namespace NEO {
|
|
|
|
class ScratchSpaceControllerBase : public ScratchSpaceController {
|
|
public:
|
|
ScratchSpaceControllerBase(uint32_t rootDeviceIndex, ExecutionEnvironment &environment, InternalAllocationStorage &allocationStorage);
|
|
|
|
void setRequiredScratchSpace(void *sshBaseAddress,
|
|
uint32_t requiredPerThreadScratchSize,
|
|
uint32_t requiredPerThreadPrivateScratchSize,
|
|
uint32_t currentTaskCount,
|
|
OsContext &osContext,
|
|
bool &stateBaseAddressDirty,
|
|
bool &vfeStateDirty) override;
|
|
uint64_t calculateNewGSH() override;
|
|
uint64_t getScratchPatchAddress() override;
|
|
|
|
void reserveHeap(IndirectHeap::Type heapType, IndirectHeap *&indirectHeap) override;
|
|
|
|
protected:
|
|
void createScratchSpaceAllocation();
|
|
};
|
|
} // namespace NEO
|