File: ModuleSplitter.h

package info (click to toggle)
intel-graphics-compiler2 2.28.4-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 792,744 kB
  • sloc: cpp: 5,761,745; ansic: 466,928; lisp: 312,143; python: 114,790; asm: 44,736; pascal: 10,930; sh: 8,033; perl: 7,914; ml: 3,625; awk: 3,523; yacc: 2,747; javascript: 2,667; lex: 1,898; f90: 1,028; cs: 573; xml: 474; makefile: 344; objc: 162
file content (32 lines) | stat: -rw-r--r-- 1,006 bytes parent folder | download 
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
 
/*========================== 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