fc62990657
This PR upstreams the GotoSolver pass. It works by walking the function and matching each label to a goto. If a label is not matched to a goto, it is removed and not lowered.
58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
//====- GotoSolver.cpp -----------------------------------===//
|
|
//
|
|
// 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 "PassDetail.h"
|
|
#include "clang/CIR/Dialect/IR/CIRDialect.h"
|
|
#include "clang/CIR/Dialect/Passes.h"
|
|
#include "llvm/Support/TimeProfiler.h"
|
|
#include <memory>
|
|
|
|
using namespace mlir;
|
|
using namespace cir;
|
|
|
|
namespace {
|
|
|
|
struct GotoSolverPass : public GotoSolverBase<GotoSolverPass> {
|
|
GotoSolverPass() = default;
|
|
void runOnOperation() override;
|
|
};
|
|
|
|
static void process(cir::FuncOp func) {
|
|
mlir::OpBuilder rewriter(func.getContext());
|
|
llvm::StringMap<Block *> labels;
|
|
llvm::SmallVector<cir::GotoOp, 4> gotos;
|
|
|
|
func.getBody().walk([&](mlir::Operation *op) {
|
|
if (auto lab = dyn_cast<cir::LabelOp>(op)) {
|
|
// Will construct a string copy inplace. Safely erase the label
|
|
labels.try_emplace(lab.getLabel(), lab->getBlock());
|
|
lab.erase();
|
|
} else if (auto goTo = dyn_cast<cir::GotoOp>(op)) {
|
|
gotos.push_back(goTo);
|
|
}
|
|
});
|
|
|
|
for (auto goTo : gotos) {
|
|
mlir::OpBuilder::InsertionGuard guard(rewriter);
|
|
rewriter.setInsertionPoint(goTo);
|
|
Block *dest = labels[goTo.getLabel()];
|
|
cir::BrOp::create(rewriter, goTo.getLoc(), dest);
|
|
goTo.erase();
|
|
}
|
|
}
|
|
|
|
void GotoSolverPass::runOnOperation() {
|
|
llvm::TimeTraceScope scope("Goto Solver");
|
|
getOperation()->walk(&process);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
std::unique_ptr<Pass> mlir::createGotoSolverPass() {
|
|
return std::make_unique<GotoSolverPass>();
|
|
}
|