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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2017-2021 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#pragma once
#include "Compiler/MetaDataUtilsWrapper.h"
#include "common/LLVMWarningsPush.hpp"
#include "llvm/Config/llvm-config.h"
#include <llvm/Pass.h>
#include <llvm/IR/Module.h>
#include <llvm/IR/Constants.h>
#include <llvm/IR/Instructions.h>
#include <llvm/IR/DerivedTypes.h>
#include <llvm/IR/DebugInfo.h>
#include <llvm/IR/DIBuilder.h>
#include <llvm/IR/IRBuilder.h>
#include <llvm/IR/DataLayout.h>
#include "common/LLVMWarningsPop.hpp"
#include <vector>
namespace IGC
{
class CodeGenContextWrapper;
/// @brief Inserts debugging information for OpenCL debugging.
class ImplicitGlobalId : public llvm::ModulePass
{
enum class GlobalOrLocal
{
Global = 0,
Local = 1,
WorkItem = 2
};
public:
static char ID;
/// @brief C'tor
ImplicitGlobalId();
/// @brief D'tor
~ImplicitGlobalId() {}
/// @brief Provides name of pass
virtual llvm::StringRef getPassName() const override
{
return "ImplicitGlobalId";
}
/// @brief execute pass on given module
/// @param M module to update
/// @returns True if module was modified
virtual bool runOnModule(llvm::Module& M) override;
/// @brief Inform about usage/modification/dependency of this pass
virtual void getAnalysisUsage(llvm::AnalysisUsage& AU) const override
{
AU.addRequired<MetaDataUtilsWrapper>();
AU.addRequired<CodeGenContextWrapper>();
}
private:
/// @brief execute pass on given function
/// @param F Function to modify
/// @returns True if function was modified
virtual bool runOnFunction(llvm::Function& F);
/// @brief execute pass on given basic block
/// @param insertBefore Instruction to insert other instructions before
/// @param wi chooses the particular pre-defined variable to emit
/// @param loc is DebugLoc to use for inserted instructions
/// @returns vector of values. Vector size is equal to 3, which is number of dimensions.
std::vector<llvm::Value*> runOnBasicBlock(llvm::Instruction* insertBefore, GlobalOrLocal wi, llvm::DebugLoc loc);
/// @brief Adds instructions to the beginning of the given function to compute the
/// global/local IDs for 3 dimensions. Fills in the FunctionContext.
/// @param pFunc Function to modify
void insertComputeIds(llvm::Function* pFunc);
/// @brief Gets or create int/long long debug info tyoe
/// returns DIType int/long long DIType
llvm::DIType* getOrCreateIntDIType();
/// @brief Iterates instructions in a basic block and tries to find the first
/// instruction with scope and loc information and return these.
/// @param BB Basic block to get the scope and loc from
/// @param scope scope information
/// @param loc debug location
/// @returns bool True if a scope and loc were found
bool getBBScope(const llvm::BasicBlock& BB, llvm::DIScope*& scope, llvm::DebugLoc& loc);
llvm::Value* CreateGetId(llvm::IRBuilder<>& B, GlobalOrLocal wi);
private:
/// This holds the processed module
llvm::Module* m_pModule;
/// This holds the debug info builder
std::unique_ptr<llvm::DIBuilder> m_pDIB;
/// This holds the processed module context
llvm::LLVMContext* m_pContext;
/// This holds debug information for a module which can be queried
llvm::DebugInfoFinder m_DbgInfoFinder;
unsigned int m_uiSizeT;
};
class CleanImplicitIds : public llvm::ModulePass
{
public:
static char ID;
/// @brief C'tor
CleanImplicitIds();
/// @brief D'tor
~CleanImplicitIds() {}
/// @brief Provides name of pass
virtual llvm::StringRef getPassName() const override
{
return "CleanImplicitIds";
}
/// @brief execute pass on given module
/// @param M module to update
/// @returns True if module was modified
virtual bool runOnModule(llvm::Module& M) override;
private:
bool processFunc(llvm::Function& F);
};
} // namespace IGC
|
