Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 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 | /// Get OpenMP device kernels in \p M. | ||
| 33 | KernelSet getDeviceKernels(Module &M); | ||
| 34 | |||
| 35 | } // namespace omp | ||
| 36 | |||
| 37 | /// OpenMP optimizations pass. | ||
| 38 | class OpenMPOptPass : public PassInfoMixin<OpenMPOptPass> { | ||
| 39 | public: | ||
| 40 | OpenMPOptPass() : LTOPhase(ThinOrFullLTOPhase::None) {} | ||
| 41 | OpenMPOptPass(ThinOrFullLTOPhase LTOPhase) : LTOPhase(LTOPhase) {} | ||
| 42 | |||
| 43 | PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM); | ||
| 44 | |||
| 45 | private: | ||
| 46 | const ThinOrFullLTOPhase LTOPhase = ThinOrFullLTOPhase::None; | ||
| 47 | }; | ||
| 48 | |||
| 49 | class OpenMPOptCGSCCPass : public PassInfoMixin<OpenMPOptCGSCCPass> { | ||
| 50 | public: | ||
| 51 | OpenMPOptCGSCCPass() : LTOPhase(ThinOrFullLTOPhase::None) {} | ||
| 52 | OpenMPOptCGSCCPass(ThinOrFullLTOPhase LTOPhase) : LTOPhase(LTOPhase) {} | ||
| 53 | |||
| 54 | PreservedAnalyses run(LazyCallGraph::SCC &C, CGSCCAnalysisManager &AM, | ||
| 55 | LazyCallGraph &CG, CGSCCUpdateResult &UR); | ||
| 56 | |||
| 57 | private: | ||
| 58 | const ThinOrFullLTOPhase LTOPhase = ThinOrFullLTOPhase::None; | ||
| 59 | }; | ||
| 60 | |||
| 61 | } // end namespace llvm | ||
| 62 | |||
| 63 | #endif // LLVM_TRANSFORMS_IPO_OPENMPOPT_H |