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
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2022 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#pragma once
#include "common/LLVMWarningsPush.hpp"
#include "llvmWrapper/IR/Module.h"
#include <llvm/Pass.h>
#include <llvm/IR/InstVisitor.h>
#include <llvm/IR/IRBuilder.h>
#include "common/LLVMWarningsPop.hpp"
#include <string>
#include <queue>
#define VISIT_FUNCTION_IMPL(InstructionType) \
void visit##InstructionType(llvm::InstructionType &instruction) { \
changed |= !wasPromoted(instruction.getFunction()) && getOrCreatePromotedValue(&instruction) != &instruction; \
}
namespace IGC {
class PromoteBools : public llvm::ModulePass, public llvm::InstVisitor<PromoteBools> {
public:
static char ID;
PromoteBools();
~PromoteBools() {}
virtual llvm::StringRef getPassName() const override { return "PromoteBools"; }
virtual bool runOnModule(llvm::Module &module) override;
VISIT_FUNCTION_IMPL(AllocaInst)
VISIT_FUNCTION_IMPL(CallInst)
VISIT_FUNCTION_IMPL(GetElementPtrInst)
VISIT_FUNCTION_IMPL(LoadInst)
VISIT_FUNCTION_IMPL(StoreInst)
VISIT_FUNCTION_IMPL(ICmpInst)
private:
bool changed;
llvm::Value *convertI1ToI8(llvm::Value *argument, llvm::Instruction *insertBefore);
llvm::Value *convertI8ToI1(llvm::Value *argument, llvm::Instruction *insertBefore);
llvm::Value *castTo(llvm::Value *value, llvm::Type *desiredType, llvm::Instruction *insertBefore);
void cleanUp(llvm::Module &module);
// Checking if type needs promotion
bool typeNeedsPromotion(llvm::Type *type, llvm::DenseSet<llvm::Type *> visitedTypes = {});
// Promoting types
llvm::DenseMap<llvm::Type *, llvm::Type *> promotedTypesCache;
llvm::Type *getOrCreatePromotedType(llvm::Type *type);
// Promoting values
llvm::DenseMap<llvm::Value *, llvm::Value *> promotedValuesCache;
std::queue<llvm::Value *> promotionQueue;
bool wasPromoted(llvm::Value *value);
template <typename Range> bool wasPromotedAnyOf(const Range &range) {
return std::any_of(std::begin(range), std::end(range), [this](const auto &item) { return wasPromoted(item); });
}
template <typename T> void setPromotedAttributes(T *callOrFunc, const llvm::AttributeList &attributeList);
llvm::Value *getOrCreatePromotedValue(llvm::Value *value);
llvm::Function *promoteFunction(llvm::Function *function);
llvm::GlobalVariable *promoteGlobalVariable(llvm::GlobalVariable *globalVariable);
llvm::Constant *promoteConstant(llvm::Constant *constant);
llvm::AllocaInst *promoteAlloca(llvm::AllocaInst *alloca);
llvm::AddrSpaceCastInst *promoteAddrSpaceCast(llvm::AddrSpaceCastInst *addrSpaceCast);
llvm::Value *promoteBitCast(llvm::BitCastInst *bitcast);
llvm::CallInst *promoteCall(llvm::CallInst *call);
llvm::CallInst *promoteIndirectCallOrInlineAsm(llvm::CallInst *call);
llvm::ExtractValueInst *promoteExtractValue(llvm::ExtractValueInst *extractValue);
llvm::GetElementPtrInst *promoteGetElementPtr(llvm::GetElementPtrInst *getElementPtr);
llvm::Value *promoteICmp(llvm::ICmpInst *icmp);
llvm::InlineAsm *promoteInlineAsm(llvm::InlineAsm *inlineAsm);
llvm::InsertValueInst *promoteInsertValue(llvm::InsertValueInst *insertValue);
llvm::LoadInst *promoteLoad(llvm::LoadInst *load);
llvm::PHINode *promotePHI(llvm::PHINode *phi);
llvm::StoreInst *promoteStore(llvm::StoreInst *store);
llvm::IntToPtrInst *promoteIntToPtr(llvm::IntToPtrInst *inttoptr);
llvm::ExtractElementInst *promoteExtractElement(llvm::ExtractElementInst *extractElement);
llvm::InsertElementInst *promoteInsertElement(llvm::InsertElementInst *insertElement);
// Promoting values - helping vars
llvm::DenseSet<llvm::PHINode *> visitedPHINodes;
};
} // namespace IGC
|
