1 //===- IPO/OpenMPOpt.h - Collection of OpenMP optimizations -----*- C++ -*-===// 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 #ifndef LLVM_TRANSFORMS_IPO_OPENMPOPT_H 10 #define LLVM_TRANSFORMS_IPO_OPENMPOPT_H 11 12 #include "llvm/Analysis/CGSCCPassManager.h" 13 #include "llvm/Analysis/LazyCallGraph.h" 14 #include "llvm/IR/PassManager.h" 15 16 namespace llvm { 17 18 namespace omp { 19 20 /// Summary of a kernel (=entry point for target offloading). 21 using Kernel = Function *; 22 23 /// Set of kernels in the module 24 using KernelSet = SetVector<Kernel>; 25 26 /// Helper to determine if \p M contains OpenMP. 27 bool containsOpenMP(Module &M); 28 29 /// Helper to determine if \p M is a OpenMP target offloading device module. 30 bool isOpenMPDevice(Module &M); 31 32 /// Return true iff \p Fn is an OpenMP GPU kernel; \p Fn has the "kernel" 33 /// attribute. 34 bool isOpenMPKernel(Function &Fn); 35 36 /// Get OpenMP device kernels in \p M. 37 KernelSet getDeviceKernels(Module &M); 38 39 } // namespace omp 40 41 /// OpenMP optimizations pass. 42 class OpenMPOptPass : public PassInfoMixin<OpenMPOptPass> { 43 public: 44 OpenMPOptPass() = default; 45 OpenMPOptPass(ThinOrFullLTOPhase LTOPhase) : LTOPhase(LTOPhase) {} 46 47 PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM); 48 49 private: 50 const ThinOrFullLTOPhase LTOPhase = ThinOrFullLTOPhase::None; 51 }; 52 53 class OpenMPOptCGSCCPass : public PassInfoMixin<OpenMPOptCGSCCPass> { 54 public: 55 OpenMPOptCGSCCPass() = default; 56 OpenMPOptCGSCCPass(ThinOrFullLTOPhase LTOPhase) : LTOPhase(LTOPhase) {} 57 58 PreservedAnalyses run(LazyCallGraph::SCC &C, CGSCCAnalysisManager &AM, 59 LazyCallGraph &CG, CGSCCUpdateResult &UR); 60 61 private: 62 const ThinOrFullLTOPhase LTOPhase = ThinOrFullLTOPhase::None; 63 }; 64 65 } // end namespace llvm 66 67 #endif // LLVM_TRANSFORMS_IPO_OPENMPOPT_H 68