mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-03 14:26:09 +00:00
5b26f4f042
This relands commit 8f4f26ba5bd04f7b335836021e5e63b4236c0305, which was reverted in 91c96a806cae58539e40c9e443a08bde91ccc91e because of Buildbot failures. The previous model test is not compatible with tflite. e.g. https://lab.llvm.org/buildbot/#/builders/6/builds/14041 Differential Revision: https://reviews.llvm.org/D133616
115 lines
4.3 KiB
C++
115 lines
4.3 KiB
C++
//===- RegAllocPriorityAdvisor.cpp - live ranges priority advisor ---------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Implementation of the default priority advisor and of the Analysis pass.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "RegAllocPriorityAdvisor.h"
|
|
#include "RegAllocGreedy.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
#include "llvm/CodeGen/VirtRegMap.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/InitializePasses.h"
|
|
#include "llvm/Pass.h"
|
|
|
|
using namespace llvm;
|
|
|
|
static cl::opt<RegAllocPriorityAdvisorAnalysis::AdvisorMode> Mode(
|
|
"regalloc-enable-priority-advisor", cl::Hidden,
|
|
cl::init(RegAllocPriorityAdvisorAnalysis::AdvisorMode::Default),
|
|
cl::desc("Enable regalloc advisor mode"),
|
|
cl::values(
|
|
clEnumValN(RegAllocPriorityAdvisorAnalysis::AdvisorMode::Default,
|
|
"default", "Default"),
|
|
clEnumValN(RegAllocPriorityAdvisorAnalysis::AdvisorMode::Release,
|
|
"release", "precompiled"),
|
|
clEnumValN(RegAllocPriorityAdvisorAnalysis::AdvisorMode::Development,
|
|
"development", "for training")));
|
|
|
|
char RegAllocPriorityAdvisorAnalysis::ID = 0;
|
|
INITIALIZE_PASS(RegAllocPriorityAdvisorAnalysis, "regalloc-priority",
|
|
"Regalloc priority policy", false, true)
|
|
|
|
namespace {
|
|
class DefaultPriorityAdvisorAnalysis final
|
|
: public RegAllocPriorityAdvisorAnalysis {
|
|
public:
|
|
DefaultPriorityAdvisorAnalysis(bool NotAsRequested)
|
|
: RegAllocPriorityAdvisorAnalysis(AdvisorMode::Default),
|
|
NotAsRequested(NotAsRequested) {}
|
|
|
|
// support for isa<> and dyn_cast.
|
|
static bool classof(const RegAllocPriorityAdvisorAnalysis *R) {
|
|
return R->getAdvisorMode() == AdvisorMode::Default;
|
|
}
|
|
|
|
private:
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override {
|
|
AU.addRequired<SlotIndexes>();
|
|
RegAllocPriorityAdvisorAnalysis::getAnalysisUsage(AU);
|
|
}
|
|
std::unique_ptr<RegAllocPriorityAdvisor>
|
|
getAdvisor(const MachineFunction &MF, const RAGreedy &RA) override {
|
|
return std::make_unique<DefaultPriorityAdvisor>(
|
|
MF, RA, &getAnalysis<SlotIndexes>());
|
|
}
|
|
bool doInitialization(Module &M) override {
|
|
if (NotAsRequested)
|
|
M.getContext().emitError("Requested regalloc priority advisor analysis "
|
|
"could be created. Using default");
|
|
return RegAllocPriorityAdvisorAnalysis::doInitialization(M);
|
|
}
|
|
const bool NotAsRequested;
|
|
};
|
|
} // namespace
|
|
|
|
template <> Pass *llvm::callDefaultCtor<RegAllocPriorityAdvisorAnalysis>() {
|
|
Pass *Ret = nullptr;
|
|
switch (Mode) {
|
|
case RegAllocPriorityAdvisorAnalysis::AdvisorMode::Default:
|
|
Ret = new DefaultPriorityAdvisorAnalysis(/*NotAsRequested*/ false);
|
|
break;
|
|
case RegAllocPriorityAdvisorAnalysis::AdvisorMode::Development:
|
|
#if defined(LLVM_HAVE_TF_API)
|
|
Ret = createDevelopmentModePriorityAdvisor();
|
|
#endif
|
|
break;
|
|
case RegAllocPriorityAdvisorAnalysis::AdvisorMode::Release:
|
|
#if defined(LLVM_HAVE_TF_AOT_REGALLOCPRIORITYMODEL)
|
|
Ret = createReleaseModePriorityAdvisor();
|
|
#endif
|
|
break;
|
|
}
|
|
if (Ret)
|
|
return Ret;
|
|
return new DefaultPriorityAdvisorAnalysis(/*NotAsRequested*/ true);
|
|
}
|
|
|
|
StringRef RegAllocPriorityAdvisorAnalysis::getPassName() const {
|
|
switch (getAdvisorMode()) {
|
|
case AdvisorMode::Default:
|
|
return "Default Regalloc Priority Advisor";
|
|
case AdvisorMode::Release:
|
|
return "Release mode Regalloc Priority Advisor";
|
|
case AdvisorMode::Development:
|
|
return "Development mode Regalloc Priority Advisor";
|
|
}
|
|
llvm_unreachable("Unknown advisor kind");
|
|
}
|
|
|
|
RegAllocPriorityAdvisor::RegAllocPriorityAdvisor(const MachineFunction &MF,
|
|
const RAGreedy &RA,
|
|
SlotIndexes *const Indexes)
|
|
: RA(RA), LIS(RA.getLiveIntervals()), VRM(RA.getVirtRegMap()),
|
|
MRI(&VRM->getRegInfo()), TRI(MF.getSubtarget().getRegisterInfo()),
|
|
RegClassInfo(RA.getRegClassInfo()), Indexes(Indexes),
|
|
RegClassPriorityTrumpsGlobalness(
|
|
RA.getRegClassPriorityTrumpsGlobalness()),
|
|
ReverseLocalAssignment(RA.getReverseLocalAssignment()) {}
|