1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/*
* Copyright (C) 2018-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/helpers/bindless_heaps_helper.h"
#include "shared/source/helpers/non_copyable_or_moveable.h"
#include "shared/source/indirect_heap/indirect_heap.h"
#include <cstddef>
#include <cstdint>
namespace NEO {
class Device;
class ExecutionEnvironment;
class GraphicsAllocation;
class InternalAllocationStorage;
class MemoryManager;
struct HardwareInfo;
class OsContext;
class CommandStreamReceiver;
namespace ScratchSpaceConstants {
inline constexpr size_t scratchSpaceOffsetFor64Bit = 4096u;
}
using ResidencyContainer = std::vector<GraphicsAllocation *>;
class ScratchSpaceController : NonCopyableAndNonMovableClass {
public:
ScratchSpaceController(uint32_t rootDeviceIndex, ExecutionEnvironment &environment, InternalAllocationStorage &allocationStorage);
virtual ~ScratchSpaceController();
MOCKABLE_VIRTUAL GraphicsAllocation *getScratchSpaceSlot0Allocation() {
return scratchSlot0Allocation;
}
GraphicsAllocation *getScratchSpaceSlot1Allocation() {
return scratchSlot1Allocation;
}
virtual void setRequiredScratchSpace(void *sshBaseAddress,
uint32_t scratchSlot,
uint32_t requiredPerThreadScratchSizeSlot0,
uint32_t requiredPerThreadScratchSizeSlot1,
OsContext &osContext,
bool &stateBaseAddressDirty,
bool &vfeStateDirty) = 0;
virtual uint64_t calculateNewGSH() = 0;
virtual uint64_t getScratchPatchAddress() = 0;
inline uint32_t getPerThreadScratchSpaceSizeSlot0() const {
return perThreadScratchSpaceSlot0Size;
}
inline uint32_t getPerThreadScratchSizeSlot1() const {
return perThreadScratchSpaceSlot1Size;
}
virtual void reserveHeap(IndirectHeap::Type heapType, IndirectHeap *&indirectHeap) = 0;
virtual void programHeaps(HeapContainer &heapContainer,
uint32_t scratchSlot,
uint32_t requiredPerThreadScratchSizeSlot0,
uint32_t requiredPerThreadScratchSizeSlot1,
OsContext &osContext,
bool &stateBaseAddressDirty,
bool &vfeStateDirty) = 0;
virtual void programBindlessSurfaceStateForScratch(BindlessHeapsHelper *heapsHelper,
uint32_t requiredPerThreadScratchSizeSlot0,
uint32_t requiredPerThreadScratchSizeSlot1,
OsContext &osContext,
bool &stateBaseAddressDirty,
bool &vfeStateDirty,
CommandStreamReceiver *csr) = 0;
protected:
MemoryManager *getMemoryManager() const;
const uint32_t rootDeviceIndex;
ExecutionEnvironment &executionEnvironment;
GraphicsAllocation *scratchSlot0Allocation = nullptr;
GraphicsAllocation *scratchSlot1Allocation = nullptr;
InternalAllocationStorage &csrAllocationStorage;
size_t scratchSlot0SizeInBytes = 0;
size_t scratchSlot1SizeInBytes = 0;
uint32_t perThreadScratchSpaceSlot0Size = 0;
uint32_t perThreadScratchSpaceSlot1Size = 0;
bool force32BitAllocation = false;
uint32_t computeUnitsUsedForScratch = 0;
};
} // namespace NEO
|
