xref: /freebsd-src/contrib/llvm-project/llvm/lib/Target/AMDGPU/AMDGPUSplitModule.h (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
1*0fca6ea1SDimitry Andric //===- AMDGPUSplitModule.h -------------------------------------*- C++ -*-===//
2*0fca6ea1SDimitry Andric //
3*0fca6ea1SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0fca6ea1SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0fca6ea1SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0fca6ea1SDimitry Andric //
7*0fca6ea1SDimitry Andric //===----------------------------------------------------------------------===//
8*0fca6ea1SDimitry Andric //
9*0fca6ea1SDimitry Andric //===----------------------------------------------------------------------===//
10*0fca6ea1SDimitry Andric 
11*0fca6ea1SDimitry Andric #ifndef LLVM_TARGET_AMDGPUSPLITMODULE_H
12*0fca6ea1SDimitry Andric #define LLVM_TARGET_AMDGPUSPLITMODULE_H
13*0fca6ea1SDimitry Andric 
14*0fca6ea1SDimitry Andric #include "llvm/ADT/STLFunctionalExtras.h"
15*0fca6ea1SDimitry Andric #include "llvm/IR/PassManager.h"
16*0fca6ea1SDimitry Andric #include <memory>
17*0fca6ea1SDimitry Andric 
18*0fca6ea1SDimitry Andric namespace llvm {
19*0fca6ea1SDimitry Andric 
20*0fca6ea1SDimitry Andric /// Splits the module M into N linkable partitions. The function ModuleCallback
21*0fca6ea1SDimitry Andric /// is called N times passing each individual partition as the MPart argument.
22*0fca6ea1SDimitry Andric class AMDGPUSplitModulePass : public PassInfoMixin<AMDGPUSplitModulePass> {
23*0fca6ea1SDimitry Andric public:
24*0fca6ea1SDimitry Andric   using ModuleCreationCallback =
25*0fca6ea1SDimitry Andric       function_ref<void(std::unique_ptr<Module> MPart)>;
26*0fca6ea1SDimitry Andric 
27*0fca6ea1SDimitry Andric   AMDGPUSplitModulePass(unsigned N, ModuleCreationCallback ModuleCallback)
28*0fca6ea1SDimitry Andric       : N(N), ModuleCallback(ModuleCallback) {}
29*0fca6ea1SDimitry Andric 
30*0fca6ea1SDimitry Andric   PreservedAnalyses run(Module &M, ModuleAnalysisManager &MAM);
31*0fca6ea1SDimitry Andric 
32*0fca6ea1SDimitry Andric private:
33*0fca6ea1SDimitry Andric   unsigned N;
34*0fca6ea1SDimitry Andric   ModuleCreationCallback ModuleCallback;
35*0fca6ea1SDimitry Andric };
36*0fca6ea1SDimitry Andric 
37*0fca6ea1SDimitry Andric } // end namespace llvm
38*0fca6ea1SDimitry Andric 
39*0fca6ea1SDimitry Andric #endif // LLVM_TARGET_AMDGPUSPLITMODULE_H
40