mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-10-19 11:30:15 +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.
84 lines
2.6 KiB
C++
84 lines
2.6 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.
|
|
|
|
#ifndef LIBSPIRV_OPT_FEATURE_MANAGER_H_
|
|
#define LIBSPIRV_OPT_FEATURE_MANAGER_H_
|
|
|
|
#include "assembly_grammar.h"
|
|
#include "extensions.h"
|
|
#include "module.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
|
|
// Tracks features enabled by a module. The IRContext has a FeatureManager.
|
|
class FeatureManager {
|
|
public:
|
|
explicit FeatureManager(const libspirv::AssemblyGrammar& grammar)
|
|
: grammar_(grammar) {}
|
|
|
|
// Returns true if |ext| is an enabled extension in the module.
|
|
bool HasExtension(libspirv::Extension ext) const {
|
|
return extensions_.Contains(ext);
|
|
}
|
|
|
|
// Returns true if |cap| is an enabled capability in the module.
|
|
bool HasCapability(SpvCapability cap) const {
|
|
return capabilities_.Contains(cap);
|
|
}
|
|
|
|
// Analyzes |module| and records enabled extensions and capabilities.
|
|
void Analyze(ir::Module* module);
|
|
|
|
libspirv::CapabilitySet* GetCapabilities() { return &capabilities_; }
|
|
const libspirv::CapabilitySet* GetCapabilities() const {
|
|
return &capabilities_;
|
|
}
|
|
|
|
uint32_t GetExtInstImportId_GLSLstd450() const {
|
|
return extinst_importid_GLSLstd450_;
|
|
}
|
|
|
|
private:
|
|
// Analyzes |module| and records enabled extensions.
|
|
void AddExtensions(ir::Module* module);
|
|
|
|
// Adds the given |capability| and all implied capabilities into the current
|
|
// FeatureManager.
|
|
void AddCapability(SpvCapability capability);
|
|
|
|
// Analyzes |module| and records enabled capabilities.
|
|
void AddCapabilities(ir::Module* module);
|
|
|
|
// Analyzes |module| and records imported external instruction sets.
|
|
void AddExtInstImportIds(ir::Module* module);
|
|
|
|
// Auxiliary object for querying SPIR-V grammar facts.
|
|
const libspirv::AssemblyGrammar& grammar_;
|
|
|
|
// The enabled extensions.
|
|
libspirv::ExtensionSet extensions_;
|
|
|
|
// The enabled capabilities.
|
|
libspirv::CapabilitySet capabilities_;
|
|
|
|
// Common external instruction import ids, cached for performance.
|
|
uint32_t extinst_importid_GLSLstd450_ = 0;
|
|
};
|
|
|
|
} // namespace opt
|
|
} // namespace spvtools
|
|
|
|
#endif // LIBSPIRV_OPT_FEATURE_MANAGER_H_
|