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
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2022 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#include "CShaderProgram.hpp"
#include "ShaderCodeGen.hpp"
#include "Compiler/CISACodeGen/OpenCLKernelCodeGen.hpp"
#include "Compiler/CISACodeGen/VectorProcess.hpp"
using namespace llvm;
using namespace IGC;
using namespace IGC::IGCMD;
CShader* CShaderProgram::GetShader(SIMDMode simd, ShaderDispatchMode mode)
{
return GetShaderPtr(simd, mode);
}
CShader*& CShaderProgram::GetShaderPtr(SIMDMode simd, ShaderDispatchMode mode)
{
switch (mode)
{
case ShaderDispatchMode::DUAL_PATCH:
return m_SIMDshaders[3];
default:
break;
}
switch (simd)
{
case SIMDMode::SIMD8:
return m_SIMDshaders[0];
case SIMDMode::SIMD16:
return m_SIMDshaders[1];
case SIMDMode::SIMD32:
return m_SIMDshaders[2];
default:
IGC_ASSERT_MESSAGE(0, "wrong SIMD size");
break;
}
return m_SIMDshaders[0];
}
void CShaderProgram::ClearShaderPtr(SIMDMode simd)
{
switch (simd)
{
case SIMDMode::SIMD8: m_SIMDshaders[0] = nullptr; break;
case SIMDMode::SIMD16: m_SIMDshaders[1] = nullptr; break;
case SIMDMode::SIMD32: m_SIMDshaders[2] = nullptr; break;
default:
IGC_ASSERT_MESSAGE(0, "wrong SIMD size");
break;
}
}
void CShaderProgram::clearBeforeRetry() {
m_kernel = nullptr;
for (auto S : m_SIMDshaders) {
if (S != nullptr) {
S->entry = nullptr;
}
}
}
bool CShaderProgram::hasShaderOutput(CShader* shader)
{
return (shader && shader->ProgramOutput()->m_programSize > 0);
}
void CShaderProgram::freeShaderOutput(CShader* shader)
{
if (hasShaderOutput(shader))
{
shader->ProgramOutput()->Destroy();
shader->ProgramOutput()->m_programSize = 0;
}
}
CShader* CShaderProgram::GetOrCreateShader(SIMDMode simd, ShaderDispatchMode mode)
{
CShader*& pShader = GetShaderPtr(simd, mode);
if (pShader == nullptr)
{
pShader = CreateNewShader(simd);
}
return pShader;
}
CShader* CShaderProgram::CreateNewShader(SIMDMode simd)
{
CShader* pShader = nullptr;
{
switch (m_context->type)
{
case ShaderType::OPENCL_SHADER:
pShader = new COpenCLKernel((OpenCLProgramContext*)m_context, m_kernel, this);
break;
default:
IGC_ASSERT_MESSAGE(0, "wrong shader type");
break;
}
}
IGC_ASSERT(nullptr != pShader);
pShader->m_shaderStats = m_shaderStats;
pShader->m_DriverInfo = &m_context->m_DriverInfo;
pShader->m_Platform = &m_context->platform;
pShader->m_pBtiLayout = &m_context->btiLayout;
pShader->m_ModuleMetadata = m_context->getModuleMetaData();
pShader->m_PrivateMemoryPerWI = pShader->m_ModuleMetadata->PrivateMemoryPerFG[m_kernel];
return pShader;
}
void CShaderProgram::DeleteShader(SIMDMode simd, ShaderDispatchMode mode)
{
CShader*& pShader = GetShaderPtr(simd, mode);
delete pShader;
pShader = nullptr;
}
|
