unittests/Passes/TestPlugin.cpp

   1 //===- unittests/Passes/Plugins/Plugin.cpp --------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/Passes/PassBuilder.h"
  10 #include "llvm/Passes/PassPlugin.h"
  11
  12 #include "TestPlugin.h"
  13
  14 using namespace llvm;
  15
  16 struct TestModulePass : public PassInfoMixin<TestModulePass> {
  17   PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM) {
  18     return PreservedAnalyses::all();
  19   }
  20 };
  21
  22 void registerCallbacks(PassBuilder &PB) {
  23   PB.registerPipelineParsingCallback(
  24       [](StringRef Name, ModulePassManager &PM,
  25          ArrayRef<PassBuilder::PipelineElement> InnerPipeline) {
  26         if (Name == "plugin-pass") {
  27           PM.addPass(TestModulePass());
  28           return true;
  29         }
  30         return false;
  31       });
  32 }
  33
  34 extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK
  35 llvmGetPassPluginInfo() {
  36   return {LLVM_PLUGIN_API_VERSION, TEST_PLUGIN_NAME, TEST_PLUGIN_VERSION,
  37           registerCallbacks};
  38 }