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