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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2017-2021 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#pragma once
#include "Compiler/CISACodeGen/ComputeShaderBase.hpp"
namespace IGC
{
class KernelArg;
}
namespace IGC
{
class COpenCLKernel : public CComputeShaderBase
{
public:
friend class CShaderProgram;
COpenCLKernel(OpenCLProgramContext* ctx, llvm::Function*, CShaderProgram* pProgram);
~COpenCLKernel();
void PreCompile() override;
void AllocatePayload() override;
void ParseShaderSpecificOpcode(llvm::Instruction* inst) override;
void ExtractGlobalVariables() override {}
bool hasReadWriteImage(llvm::Function& F) override;
bool CompileSIMDSize(SIMDMode simdMode, EmitPass& EP, llvm::Function& F) override;
SIMDStatus checkSIMDCompileConds(SIMDMode simdMode, EmitPass& EP, llvm::Function& F, bool hasSyncRTCalls);
SIMDStatus checkSIMDCompileCondsPVC(SIMDMode simdMode, EmitPass& EP, llvm::Function& F, bool hasSyncRTCalls);
bool IsRegularGRFRequested() override;
bool IsLargeGRFRequested() override;
unsigned getAnnotatedNumThreads() override;
void FillKernel(SIMDMode simdMode);
// Recomputes the binding table layout according to the present kernel args
void RecomputeBTLayout();
// Set m_HasTID to true if TID functions were found
void SetHasTID();
// Set m_HasGlobalSize to true if TID functions were found
void SetHasGlobalSize();
bool HasFullDispatchMask() override;
// Returns the immediate value mapped to GlobalVariable c.
// (GlobalVariables represent the pointer to the global,
// which is a compile-time constant)
unsigned int GetGlobalMappingValue(llvm::Value* c) override;
CVariable* GetGlobalMapping(llvm::Value* c) override;
const SOpenCLKernelInfo& getKernelInfo() const { return m_kernelInfo; }
static bool IsValidShader(COpenCLKernel* shader);
public:
SOpenCLProgramInfo* m_programInfo;
SOpenCLKernelInfo m_kernelInfo;
unsigned int m_perWIStatelessPrivateMemSize;
bool GetDisableMidThreadPreemption() const { return m_disableMidThreadPreemption; }
void SetDisableMidthreadPreemption() { m_disableMidThreadPreemption = true; }
bool passNOSInlineData() override;
bool loadThreadPayload() override;
protected:
// Creates appropriate annotation based on the kernel arg
void CreateAnnotations(IGC::KernelArg* kernelArg, uint payloadPosition);
// Fill SOpenCLKernelInfo::m_zePayloadArgs
// Return true: if the argument is supported in ZEBinary and it's created successfully
// Return false: if the argument cannot be supported by ZEBinary
bool CreateZEPayloadArguments(IGC::KernelArg* kernelArg, uint payloadPosition);
// Fill SOpenCLKernelInfo::m_zeUserAttribute for ZEBinary
// (PT pass: CreateKernelAttributeInfo)
void FillZEUserAttributes(IGC::IGCMD::FunctionInfoMetaDataHandle& funcInfoMD);
// Fill SOpenCLKernelInfo::m_zeKernelArgInfo for ZEBinary
// (PT pass: CreateKernelArgInfo)
void FillZEKernelArgInfo();
// a helper function to get image type from kernelArg
iOpenCL::IMAGE_MEMORY_OBJECT_TYPE getImageTypeFromKernelArg(const KernelArg& kernelArg);
// a helper function to get sampler type from kernelArg
iOpenCL::SAMPLER_OBJECT_TYPE getSamplerTypeFromKernelArg(const KernelArg& kernelArg);
// Creates annotations for inline sampler_t objects
void CreateInlineSamplerAnnotations();
void CreateZEInlineSamplerAnnotations();
// Creates annotations for kernel argument information (kernel reflection)
void CreateKernelArgInfo();
// Creates annotations for kernel attribution information (kernel reflection)
void CreateKernelAttributeInfo();
std::string getVecTypeHintString(const IGC::IGCMD::VectorTypeHintMetaDataHandle& vecTypeHintInfo) const;
std::string getVecTypeHintTypeString(const IGC::IGCMD::VectorTypeHintMetaDataHandle& vecTypeHintInfo) const;
std::string getThreadGroupSizeString(IGC::IGCMD::ThreadGroupSizeMetaDataHandle& threadGroupSize, bool isHint);
std::string getSubGroupSizeString(IGC::IGCMD::SubGroupSizeMetaDataHandle& subGroupSize);
std::string getWorkgroupWalkOrderString(const IGC::WorkGroupWalkOrderMD& workgroupWalkOrder);
// Create annotation for printf strings.
void CreatePrintfStringAnnotations();
// Load from MD and return the resource information for argument number argNo
SOpenCLKernelInfo::SResourceInfo getResourceInfo(int argNo);
// Load from MD and return the resource extension information for argument number argNo
ResourceExtensionTypeEnum getExtensionInfo(int argNo);
// Resolve the binding table index for resource resInfo (using the BTL)
unsigned int getBTI(SOpenCLKernelInfo::SResourceInfo& resInfo);
// Find the sum of inline local sizes used by this kernel
unsigned int getSumFixedTGSMSizes(llvm::Function* F);
bool m_HasTID;
bool m_HasGlobalSize;
bool m_disableMidThreadPreemption;
bool m_largeGRFRequested;
bool m_regularGRFRequested;
unsigned m_annotatedNumThreads;
// Maps GlobalVariables representing local address-space pointers
// to their offsets in SLM.
std::map<llvm::Value*, unsigned int> m_localOffsetsMap;
OpenCLProgramContext* m_Context;
void ClearKernelInfo();
private:
WorkGroupWalkOrderMD getWorkGroupWalkOrder();
void tryHWGenerateLocalIDs();
// helper functions for collecting kernel argument info
// Format the strings the way the OpenCL runtime expects them
std::string getKernelArgTypeName(const FunctionMetaData& funcMD, uint argIndex) const;
std::string getKernelArgTypeQualifier(const FunctionMetaData& funcMD, uint argIndex) const;
std::string getKernelArgAddressQualifier(const FunctionMetaData& funcMD, uint argIndex) const;
std::string getKernelArgAccessQualifier(const FunctionMetaData& funcMD, uint argIndex) const;
};
}
|
