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 |