361acbb362
We currently generate reproducer configurations using a comment placed at the top of the generated .mlir file. This is kind of hacky given that comments have no semantic context in the source file and can easily be dropped. This strategy also wouldn't work if/when we have a bitcode format. This commit switches to using an external assembly resource, which is verifiable/can work with a hypothetical bitcode naturally/and removes the awkward processing from mlir-opt for splicing comments and re-applying command line options. With the removal of command line munging, this opens up new possibilities for executing reproducers in memory. Differential Revision: https://reviews.llvm.org/D126447
59 lines
1.9 KiB
C++
59 lines
1.9 KiB
C++
//===- TestModuleCombiner.cpp - Pass to test SPIR-V module combiner lib ---===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
|
|
#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
|
|
#include "mlir/Dialect/SPIRV/Linking/ModuleCombiner.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
class TestModuleCombinerPass
|
|
: public PassWrapper<TestModuleCombinerPass,
|
|
OperationPass<mlir::ModuleOp>> {
|
|
public:
|
|
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestModuleCombinerPass)
|
|
|
|
StringRef getArgument() const final { return "test-spirv-module-combiner"; }
|
|
StringRef getDescription() const final {
|
|
return "Tests SPIR-V module combiner library";
|
|
}
|
|
TestModuleCombinerPass() = default;
|
|
TestModuleCombinerPass(const TestModuleCombinerPass &) {}
|
|
void runOnOperation() override;
|
|
};
|
|
} // namespace
|
|
|
|
void TestModuleCombinerPass::runOnOperation() {
|
|
auto modules = llvm::to_vector<4>(getOperation().getOps<spirv::ModuleOp>());
|
|
|
|
OpBuilder combinedModuleBuilder(modules[0]);
|
|
|
|
auto listener = [](spirv::ModuleOp originalModule, StringRef oldSymbol,
|
|
StringRef newSymbol) {
|
|
llvm::outs() << "[" << originalModule.getName() << "] " << oldSymbol
|
|
<< " -> " << newSymbol << "\n";
|
|
};
|
|
|
|
OwningOpRef<spirv::ModuleOp> combinedModule =
|
|
spirv::combine(modules, combinedModuleBuilder, listener);
|
|
|
|
for (spirv::ModuleOp module : modules)
|
|
module.erase();
|
|
combinedModule.release();
|
|
}
|
|
|
|
namespace mlir {
|
|
void registerTestSpirvModuleCombinerPass() {
|
|
PassRegistration<TestModuleCombinerPass>();
|
|
}
|
|
} // namespace mlir
|