1 //===- unittests/Passes/Plugins/Plugin.cpp --------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "llvm/Passes/PassBuilder.h" 11 #include "llvm/Passes/PassPlugin.h" 12 13 #include "TestPlugin.h" 14 15 using namespace llvm; 16 17 struct TestModulePass : public PassInfoMixin<TestModulePass> { 18 PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM) { 19 return PreservedAnalyses::all(); 20 } 21 }; 22 23 void registerCallbacks(PassBuilder &PB) { 24 PB.registerPipelineParsingCallback( 25 [](StringRef Name, ModulePassManager &PM, 26 ArrayRef<PassBuilder::PipelineElement> InnerPipeline) { 27 if (Name == "plugin-pass") { 28 PM.addPass(TestModulePass()); 29 return true; 30 } 31 return false; 32 }); 33 } 34 35 extern "C" ::llvm::PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK 36 llvmGetPassPluginInfo() { 37 return {LLVM_PLUGIN_API_VERSION, TEST_PLUGIN_NAME, TEST_PLUGIN_VERSION, 38 registerCallbacks}; 39 } 40
