56a8655f4a
This PR exposes the backend pass config to plugins via a callback. Plugin authors can register a callback that is being triggered before the target backend adds their passes to the pipeline. In the callback they then get access to the `TargetMachine`, the `PassManager`, and the `TargetPassConfig`. This allows plugins to call `TargetPassConfig::insertPass`, which is honored in the subsequent `addPass` of the main backend. We implemented this using the legacy pass manager since backends still use it as the default.
40 lines
1.5 KiB
C++
40 lines
1.5 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// This file contains the registry for PassConfigCallbacks that enable changes
|
|
/// to the TargetPassConfig during the initialization of TargetMachine.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Target/RegisterTargetPassConfigCallback.h"
|
|
|
|
namespace llvm {
|
|
// TargetPassConfig callbacks
|
|
static SmallVector<RegisterTargetPassConfigCallback *, 1>
|
|
TargetPassConfigCallbacks{};
|
|
|
|
void invokeGlobalTargetPassConfigCallbacks(TargetMachine &TM,
|
|
PassManagerBase &PM,
|
|
TargetPassConfig *PassConfig) {
|
|
for (const RegisterTargetPassConfigCallback *Reg : TargetPassConfigCallbacks)
|
|
Reg->Callback(TM, PM, PassConfig);
|
|
}
|
|
|
|
RegisterTargetPassConfigCallback::RegisterTargetPassConfigCallback(
|
|
PassConfigCallback &&C)
|
|
: Callback(std::move(C)) {
|
|
TargetPassConfigCallbacks.push_back(this);
|
|
}
|
|
|
|
RegisterTargetPassConfigCallback::~RegisterTargetPassConfigCallback() {
|
|
const auto &It = find(TargetPassConfigCallbacks, this);
|
|
if (It != TargetPassConfigCallbacks.end())
|
|
TargetPassConfigCallbacks.erase(It);
|
|
}
|
|
} // namespace llvm
|