Files
compute-runtime/runtime/command_stream/scratch_space_controller_base.h
Mateusz Jablonski f7f420b066 Pass os context to ScratchSpaceController::setRequiredScratchSpace
Related-To: NEO-3691

Change-Id: I3c040b9dd8e31dc798d70c7d0b12e2a89d22fa80
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
2019-09-16 17:56:03 +02:00

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(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