mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-30 15:00:06 +00:00
86627f7b3f
Add extra iterators for ir::Module's sections Add extra getters to ir::Function Add a const version of BasicBlock::GetLabelInst() Use the max of all inputs' version as version Split debug in debug1 and debug2 - Debug1 instructions have to be placed before debug2 instructions. Error out if different addressing or memory models are found Exit early if no binaries were given Error out if entry points are redeclared Implement copy ctors for Function and BasicBlock - Visual Studio ends up generating copy constructors that call deleted functions while compiling the linker code, while GCC and clang do not. So explicitly write those functions to avoid Visual Studio messing up. Move removing duplicate capabilities to its own pass Add functions running on all IDs present in an instruction Remove duplicate SpvOpExtInstImport Give default options value for link functions Remove linkage capability if not making a library Check types before allowing to link Detect if two types/variables/functions have different decorations Remove decorations of imported variables/functions and their types Add a DecorationManager Add a method for removing all decorations of id Add methods for removing operands from instructions Error out if one of the modules has a non-zero schema Update README.md to talk about the linker Do not freak out if an imported built-in variable has no export
104 lines
3.4 KiB
C++
104 lines
3.4 KiB
C++
// Copyright (c) 2016 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_PASS_MANAGER_H_
|
|
#define LIBSPIRV_OPT_PASS_MANAGER_H_
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "log.h"
|
|
#include "module.h"
|
|
#include "pass.h"
|
|
|
|
#include "spirv-tools/libspirv.hpp"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
|
|
// The pass manager, responsible for tracking and running passes.
|
|
// Clients should first call AddPass() to add passes and then call Run()
|
|
// to run on a module. Passes are executed in the exact order of addition.
|
|
class PassManager {
|
|
public:
|
|
// Constructs a pass manager.
|
|
//
|
|
// The constructed instance will have an empty message consumer, which just
|
|
// ignores all messages from the library. Use SetMessageConsumer() to supply
|
|
// one if messages are of concern.
|
|
PassManager() : consumer_(nullptr) {}
|
|
|
|
// Sets the message consumer to the given |consumer|.
|
|
void SetMessageConsumer(MessageConsumer c) { consumer_ = std::move(c); }
|
|
|
|
// Adds an externally constructed pass.
|
|
void AddPass(std::unique_ptr<Pass> pass);
|
|
// Uses the argument |args| to construct a pass instance of type |T|, and adds
|
|
// the pass instance to this pass manger. The pass added will use this pass
|
|
// manager's message consumer.
|
|
template <typename T, typename... Args>
|
|
void AddPass(Args&&... args);
|
|
|
|
// Returns the number of passes added.
|
|
uint32_t NumPasses() const;
|
|
// Returns a pointer to the |index|th pass added.
|
|
inline Pass* GetPass(uint32_t index) const;
|
|
|
|
// Returns the message consumer.
|
|
inline const MessageConsumer& consumer() const;
|
|
|
|
// Runs all passes on the given |module|. Returns Status::Failure if errors
|
|
// occur when processing using one of the registered passes. All passes
|
|
// registered after the error-reporting pass will be skipped. Returns the
|
|
// corresponding Status::Success if processing is succesful to indicate
|
|
// whether changes are made to the module.
|
|
//
|
|
// After running all the passes, they are removed from the list.
|
|
Pass::Status Run(ir::Module* module);
|
|
|
|
private:
|
|
// Consumer for messages.
|
|
MessageConsumer consumer_;
|
|
// A vector of passes. Order matters.
|
|
std::vector<std::unique_ptr<Pass>> passes_;
|
|
};
|
|
|
|
inline void PassManager::AddPass(std::unique_ptr<Pass> pass) {
|
|
passes_.push_back(std::move(pass));
|
|
}
|
|
|
|
template <typename T, typename... Args>
|
|
inline void PassManager::AddPass(Args&&... args) {
|
|
passes_.emplace_back(new T(std::forward<Args>(args)...));
|
|
passes_.back()->SetMessageConsumer(consumer_);
|
|
}
|
|
|
|
inline uint32_t PassManager::NumPasses() const {
|
|
return static_cast<uint32_t>(passes_.size());
|
|
}
|
|
|
|
inline Pass* PassManager::GetPass(uint32_t index) const {
|
|
SPIRV_ASSERT(consumer_, index < passes_.size(), "index out of bound");
|
|
return passes_[index].get();
|
|
}
|
|
|
|
inline const MessageConsumer& PassManager::consumer() const {
|
|
return consumer_;
|
|
}
|
|
|
|
} // namespace opt
|
|
} // namespace spvtools
|
|
|
|
#endif // LIBSPIRV_OPT_PASS_MANAGER_H_
|