mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-30 15:00:06 +00:00
309be423cc
This change implements instruction folding for arithmetic operations that are redundant, specifically: x + 0 = 0 + x = x x - 0 = x 0 - x = -x x * 0 = 0 * x = 0 x * 1 = 1 * x = x 0 / x = 0 x / 1 = x mix(a, b, 0) = a mix(a, b, 1) = b Cache ExtInst import id in feature manager This allows us to avoid string lookups during optimization; for now we just cache GLSL std450 import id but I can imagine caching more sets as they become utilized by the optimizer. Add tests for add/sub/mul/div/mix folding The tests cover scalar float/double cases, and some vector cases. Since most of the code for floating point folding is shared, the tests for vector folding are not as exhaustive as scalar. To test sub->negate folding I had to implement a custom fixture.
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
// Copyright (c) 2017 Google Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "feature_manager.h"
|
|
#include <queue>
|
|
#include <stack>
|
|
|
|
#include "enum_string_mapping.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
|
|
void FeatureManager::Analyze(ir::Module* module) {
|
|
AddExtensions(module);
|
|
AddCapabilities(module);
|
|
AddExtInstImportIds(module);
|
|
}
|
|
|
|
void FeatureManager::AddExtensions(ir::Module* module) {
|
|
for (auto ext : module->extensions()) {
|
|
const std::string name =
|
|
reinterpret_cast<const char*>(ext.GetInOperand(0u).words.data());
|
|
libspirv::Extension extension;
|
|
if (libspirv::GetExtensionFromString(name.c_str(), &extension)) {
|
|
extensions_.Add(extension);
|
|
}
|
|
}
|
|
}
|
|
|
|
void FeatureManager::AddCapability(SpvCapability cap) {
|
|
if (capabilities_.Contains(cap)) return;
|
|
|
|
capabilities_.Add(cap);
|
|
|
|
spv_operand_desc desc = {};
|
|
if (SPV_SUCCESS ==
|
|
grammar_.lookupOperand(SPV_OPERAND_TYPE_CAPABILITY, cap, &desc)) {
|
|
libspirv::CapabilitySet(desc->numCapabilities, desc->capabilities)
|
|
.ForEach([this](SpvCapability c) { AddCapability(c); });
|
|
}
|
|
}
|
|
|
|
void FeatureManager::AddCapabilities(ir::Module* module) {
|
|
for (ir::Instruction& inst : module->capabilities()) {
|
|
AddCapability(static_cast<SpvCapability>(inst.GetSingleWordInOperand(0)));
|
|
}
|
|
}
|
|
|
|
void FeatureManager::AddExtInstImportIds(ir::Module* module) {
|
|
extinst_importid_GLSLstd450_ = module->GetExtInstImportId("GLSL.std.450");
|
|
}
|
|
|
|
} // namespace opt
|
|
} // namespace spvtools
|