mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 11:13:28 +01:00
fbc0eeef58
FailedISel MachineFunction property is part of the CodeGen pipeline state as much as every other property, notably, Legalized, RegBankSelected, and Selected. Let's make that part of the state also serializable / de-serializable, so if GlobalISel aborts on some of the functions of a large module, but not the others, it could be easily seen and the state of the pipeline could be maintained through llc's invocations with -stop-after / -start-after. To make MIR printable and generally to not to break it too much too soon, this patch also defers cleaning up the vreg -> LLT map until ResetMachineFunctionPass. To make MIR with FailedISel: true also machine verifiable, machine verifier is changed so it treats a MIR-module as non-regbankselected and non-selected if there is FailedISel property set. Reviewers: qcolombet, ab Reviewed By: dsanders Subscribers: javed.absar, rovka, kristof.beyls, llvm-commits Differential Revision: https://reviews.llvm.org/D42877 llvm-svn: 326343
83 lines
3.1 KiB
C++
83 lines
3.1 KiB
C++
//===-- ResetMachineFunctionPass.cpp - Reset Machine Function ----*- C++ -*-==//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
/// \file
|
|
/// This file implements a pass that will conditionally reset a machine
|
|
/// function as if it was just created. This is used to provide a fallback
|
|
/// mechanism when GlobalISel fails, thus the condition for the reset to
|
|
/// happen is that the MachineFunction has the FailedISel property.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/ScopeExit.h"
|
|
#include "llvm/ADT/Statistic.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
#include "llvm/CodeGen/MachineFunctionPass.h"
|
|
#include "llvm/CodeGen/MachineRegisterInfo.h"
|
|
#include "llvm/CodeGen/Passes.h"
|
|
#include "llvm/IR/DiagnosticInfo.h"
|
|
#include "llvm/Support/Debug.h"
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "reset-machine-function"
|
|
|
|
STATISTIC(NumFunctionsReset, "Number of functions reset");
|
|
|
|
namespace {
|
|
class ResetMachineFunction : public MachineFunctionPass {
|
|
/// Tells whether or not this pass should emit a fallback
|
|
/// diagnostic when it resets a function.
|
|
bool EmitFallbackDiag;
|
|
/// Whether we should abort immediately instead of resetting the function.
|
|
bool AbortOnFailedISel;
|
|
|
|
public:
|
|
static char ID; // Pass identification, replacement for typeid
|
|
ResetMachineFunction(bool EmitFallbackDiag = false,
|
|
bool AbortOnFailedISel = false)
|
|
: MachineFunctionPass(ID), EmitFallbackDiag(EmitFallbackDiag),
|
|
AbortOnFailedISel(AbortOnFailedISel) {}
|
|
|
|
StringRef getPassName() const override { return "ResetMachineFunction"; }
|
|
|
|
bool runOnMachineFunction(MachineFunction &MF) override {
|
|
// No matter what happened, whether we successfully selected the function
|
|
// or not, nothing is going to use the vreg types after us. Make sure they
|
|
// disappear.
|
|
auto ClearVRegTypesOnReturn =
|
|
make_scope_exit([&MF]() { MF.getRegInfo().getVRegToType().clear(); });
|
|
|
|
if (MF.getProperties().hasProperty(
|
|
MachineFunctionProperties::Property::FailedISel)) {
|
|
if (AbortOnFailedISel)
|
|
report_fatal_error("Instruction selection failed");
|
|
DEBUG(dbgs() << "Resetting: " << MF.getName() << '\n');
|
|
++NumFunctionsReset;
|
|
MF.reset();
|
|
if (EmitFallbackDiag) {
|
|
const Function &F = MF.getFunction();
|
|
DiagnosticInfoISelFallback DiagFallback(F);
|
|
F.getContext().diagnose(DiagFallback);
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
char ResetMachineFunction::ID = 0;
|
|
INITIALIZE_PASS(ResetMachineFunction, DEBUG_TYPE,
|
|
"Reset machine function if ISel failed", false, false)
|
|
|
|
MachineFunctionPass *
|
|
llvm::createResetMachineFunctionPass(bool EmitFallbackDiag = false,
|
|
bool AbortOnFailedISel = false) {
|
|
return new ResetMachineFunction(EmitFallbackDiag, AbortOnFailedISel);
|
|
}
|