1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2021-2023 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#pragma once
#include <Compiler/CodeGenPublic.h>
namespace IGC {
// Class for splitting llvm::Module per kernel
// handles splitting, owns split module, handles proper cleanup
class KernelModuleSplitter
{
public:
KernelModuleSplitter(IGC::OpenCLProgramContext& oclContext, llvm::Module& module);
~KernelModuleSplitter();
KernelModuleSplitter(const KernelModuleSplitter&) = delete;
KernelModuleSplitter& operator=(const KernelModuleSplitter&) = delete;
void restoreOclContextModule();
void setSplittedModuleInOCLContext();
void retry();
void splitModuleForKernel(const llvm::Function *kernelF);
private:
IGC::OpenCLProgramContext& _oclContext;
llvm::Module& _originalModule;
std::unique_ptr<llvm::Module> _splittedModule;
};
} // namespace IGC
|
