1 //
2 // Copyright (C) 2014-2016 LunarG, Inc.
3 // Copyright (C) 2018-2020 Google, Inc.
4 //
5 // All rights reserved.
6 //
7 // Redistribution and use in source and binary forms, with or without
8 // modification, are permitted provided that the following conditions
9 // are met:
10 //
11 // Redistributions of source code must retain the above copyright
12 // notice, this list of conditions and the following disclaimer.
13 //
14 // Redistributions in binary form must reproduce the above
15 // copyright notice, this list of conditions and the following
16 // disclaimer in the documentation and/or other materials provided
17 // with the distribution.
18 //
19 // Neither the name of 3Dlabs Inc. Ltd. nor the names of its
20 // contributors may be used to endorse or promote products derived
21 // from this software without specific prior written permission.
22 //
23 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
26 // FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
27 // COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
28 // INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
29 // BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
30 // LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
31 // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 // LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
33 // ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
34 // POSSIBILITY OF SUCH DAMAGE.
35
36 //
37 // Call into SPIRV-Tools to disassemble, validate, and optimize.
38 //
39
40 #if ENABLE_OPT
41
42 #include <cstdio>
43 #include <iostream>
44
45 #include "SpvTools.h"
46 #include "spirv-tools/optimizer.hpp"
47 #include "spirv-tools/libspirv.h"
48
49 namespace glslang {
50
51 // Translate glslang's view of target versioning to what SPIRV-Tools uses.
MapToSpirvToolsEnv(const SpvVersion & spvVersion,spv::SpvBuildLogger * logger)52 spv_target_env MapToSpirvToolsEnv(const SpvVersion& spvVersion, spv::SpvBuildLogger* logger)
53 {
54 switch (spvVersion.vulkan) {
55 case glslang::EShTargetVulkan_1_0:
56 return spv_target_env::SPV_ENV_VULKAN_1_0;
57 case glslang::EShTargetVulkan_1_1:
58 switch (spvVersion.spv) {
59 case EShTargetSpv_1_0:
60 case EShTargetSpv_1_1:
61 case EShTargetSpv_1_2:
62 case EShTargetSpv_1_3:
63 return spv_target_env::SPV_ENV_VULKAN_1_1;
64 case EShTargetSpv_1_4:
65 return spv_target_env::SPV_ENV_VULKAN_1_1_SPIRV_1_4;
66 default:
67 logger->missingFunctionality("Target version for SPIRV-Tools validator");
68 return spv_target_env::SPV_ENV_VULKAN_1_1;
69 }
70 case glslang::EShTargetVulkan_1_2:
71 return spv_target_env::SPV_ENV_VULKAN_1_2;
72 default:
73 break;
74 }
75
76 if (spvVersion.openGl > 0)
77 return spv_target_env::SPV_ENV_OPENGL_4_5;
78
79 logger->missingFunctionality("Target version for SPIRV-Tools validator");
80 return spv_target_env::SPV_ENV_UNIVERSAL_1_0;
81 }
82
83
84 // Use the SPIRV-Tools disassembler to print SPIR-V.
SpirvToolsDisassemble(std::ostream & out,const std::vector<unsigned int> & spirv)85 void SpirvToolsDisassemble(std::ostream& out, const std::vector<unsigned int>& spirv)
86 {
87 // disassemble
88 spv_context context = spvContextCreate(SPV_ENV_UNIVERSAL_1_3);
89 spv_text text;
90 spv_diagnostic diagnostic = nullptr;
91 spvBinaryToText(context, spirv.data(), spirv.size(),
92 SPV_BINARY_TO_TEXT_OPTION_FRIENDLY_NAMES | SPV_BINARY_TO_TEXT_OPTION_INDENT,
93 &text, &diagnostic);
94
95 // dump
96 if (diagnostic == nullptr)
97 out << text->str;
98 else
99 spvDiagnosticPrint(diagnostic);
100
101 // teardown
102 spvDiagnosticDestroy(diagnostic);
103 spvContextDestroy(context);
104 }
105
106 // Apply the SPIRV-Tools validator to generated SPIR-V.
SpirvToolsValidate(const glslang::TIntermediate & intermediate,std::vector<unsigned int> & spirv,spv::SpvBuildLogger * logger,bool prelegalization)107 void SpirvToolsValidate(const glslang::TIntermediate& intermediate, std::vector<unsigned int>& spirv,
108 spv::SpvBuildLogger* logger, bool prelegalization)
109 {
110 // validate
111 spv_context context = spvContextCreate(MapToSpirvToolsEnv(intermediate.getSpv(), logger));
112 spv_const_binary_t binary = { spirv.data(), spirv.size() };
113 spv_diagnostic diagnostic = nullptr;
114 spv_validator_options options = spvValidatorOptionsCreate();
115 spvValidatorOptionsSetRelaxBlockLayout(options, intermediate.usingHlslOffsets());
116 spvValidatorOptionsSetBeforeHlslLegalization(options, prelegalization);
117 spvValidateWithOptions(context, options, &binary, &diagnostic);
118
119 // report
120 if (diagnostic != nullptr) {
121 logger->error("SPIRV-Tools Validation Errors");
122 logger->error(diagnostic->error);
123 }
124
125 // tear down
126 spvValidatorOptionsDestroy(options);
127 spvDiagnosticDestroy(diagnostic);
128 spvContextDestroy(context);
129 }
130
131 // Apply the SPIRV-Tools optimizer to generated SPIR-V, for the purpose of
132 // legalizing HLSL SPIR-V.
SpirvToolsLegalize(const glslang::TIntermediate & intermediate,std::vector<unsigned int> & spirv,spv::SpvBuildLogger * logger,const SpvOptions * options)133 void SpirvToolsLegalize(const glslang::TIntermediate& intermediate, std::vector<unsigned int>& spirv,
134 spv::SpvBuildLogger* logger, const SpvOptions* options)
135 {
136 spv_target_env target_env = SPV_ENV_UNIVERSAL_1_2;
137
138 spvtools::Optimizer optimizer(target_env);
139 optimizer.SetMessageConsumer(
140 [](spv_message_level_t level, const char *source, const spv_position_t &position, const char *message) {
141 auto &out = std::cerr;
142 switch (level)
143 {
144 case SPV_MSG_FATAL:
145 case SPV_MSG_INTERNAL_ERROR:
146 case SPV_MSG_ERROR:
147 out << "error: ";
148 break;
149 case SPV_MSG_WARNING:
150 out << "warning: ";
151 break;
152 case SPV_MSG_INFO:
153 case SPV_MSG_DEBUG:
154 out << "info: ";
155 break;
156 default:
157 break;
158 }
159 if (source)
160 {
161 out << source << ":";
162 }
163 out << position.line << ":" << position.column << ":" << position.index << ":";
164 if (message)
165 {
166 out << " " << message;
167 }
168 out << std::endl;
169 });
170
171 // If debug (specifically source line info) is being generated, propagate
172 // line information into all SPIR-V instructions. This avoids loss of
173 // information when instructions are deleted or moved. Later, remove
174 // redundant information to minimize final SPRIR-V size.
175 if (options->generateDebugInfo) {
176 optimizer.RegisterPass(spvtools::CreatePropagateLineInfoPass());
177 }
178 optimizer.RegisterPass(spvtools::CreateWrapOpKillPass());
179 optimizer.RegisterPass(spvtools::CreateDeadBranchElimPass());
180 optimizer.RegisterPass(spvtools::CreateMergeReturnPass());
181 optimizer.RegisterPass(spvtools::CreateInlineExhaustivePass());
182 optimizer.RegisterPass(spvtools::CreateEliminateDeadFunctionsPass());
183 optimizer.RegisterPass(spvtools::CreateScalarReplacementPass());
184 optimizer.RegisterPass(spvtools::CreateLocalAccessChainConvertPass());
185 optimizer.RegisterPass(spvtools::CreateLocalSingleBlockLoadStoreElimPass());
186 optimizer.RegisterPass(spvtools::CreateLocalSingleStoreElimPass());
187 optimizer.RegisterPass(spvtools::CreateSimplificationPass());
188 optimizer.RegisterPass(spvtools::CreateAggressiveDCEPass());
189 optimizer.RegisterPass(spvtools::CreateVectorDCEPass());
190 optimizer.RegisterPass(spvtools::CreateDeadInsertElimPass());
191 optimizer.RegisterPass(spvtools::CreateAggressiveDCEPass());
192 optimizer.RegisterPass(spvtools::CreateDeadBranchElimPass());
193 optimizer.RegisterPass(spvtools::CreateBlockMergePass());
194 optimizer.RegisterPass(spvtools::CreateLocalMultiStoreElimPass());
195 optimizer.RegisterPass(spvtools::CreateIfConversionPass());
196 optimizer.RegisterPass(spvtools::CreateSimplificationPass());
197 optimizer.RegisterPass(spvtools::CreateAggressiveDCEPass());
198 optimizer.RegisterPass(spvtools::CreateVectorDCEPass());
199 optimizer.RegisterPass(spvtools::CreateDeadInsertElimPass());
200 if (options->optimizeSize) {
201 optimizer.RegisterPass(spvtools::CreateRedundancyEliminationPass());
202 }
203 optimizer.RegisterPass(spvtools::CreateAggressiveDCEPass());
204 optimizer.RegisterPass(spvtools::CreateCFGCleanupPass());
205 if (options->generateDebugInfo) {
206 optimizer.RegisterPass(spvtools::CreateRedundantLineInfoElimPass());
207 }
208
209 spvtools::OptimizerOptions spvOptOptions;
210 optimizer.SetTargetEnv(MapToSpirvToolsEnv(intermediate.getSpv(), logger));
211 spvOptOptions.set_run_validator(false); // The validator may run as a separate step later on
212 optimizer.Run(spirv.data(), spirv.size(), &spirv, spvOptOptions);
213 }
214
215 }; // end namespace glslang
216
217 #endif
218