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
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2017-2021 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#include "Compiler/Optimizer/OpenCLPasses/ClampLoopUnroll/ClampLoopUnroll.hpp"
#include "Compiler/IGCPassSupport.h"
#include "common/LLVMWarningsPush.hpp"
#include <llvm/Transforms/Utils/UnrollLoop.h>
#include <llvm/Pass.h>
#include "common/LLVMWarningsPop.hpp"
using namespace llvm;
using namespace IGC;
#define PASS_FLAG "igc-clamp-loop-unroll"
#define PASS_DESCRIPTION "Sets a limit on unrolling metadata to control compile time"
#define PASS_CFG_ONLY false
#define PASS_ANALYSIS false
IGC_INITIALIZE_PASS_BEGIN(ClampLoopUnroll, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
IGC_INITIALIZE_PASS_END(ClampLoopUnroll, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
static const char* const LoopMDName = "llvm.loop";
char ClampLoopUnroll::ID = 0;
ClampLoopUnroll::ClampLoopUnroll() :
FunctionPass(ID), m_MaxUnrollFactor(0), m_Changed(false)
{
initializeClampLoopUnrollPass(*PassRegistry::getPassRegistry());
}
ClampLoopUnroll::ClampLoopUnroll(unsigned maxUnrollFactor) :
FunctionPass(ID), m_MaxUnrollFactor(maxUnrollFactor), m_Changed(false)
{
initializeClampLoopUnrollPass(*PassRegistry::getPassRegistry());
}
static MDNode* GetUnrollCountMD(const IGCLLVM::TerminatorInst& I, unsigned unroll)
{
LLVMContext& Context = I.getContext();
SmallVector<Metadata*, 2> UnrollMD;
UnrollMD.push_back(MDString::get(Context, "llvm.loop.unroll.count"));
UnrollMD.push_back(ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(Context), unroll)));
SmallVector<Metadata*, 2> MDs;
MDs.push_back(nullptr);
MDs.push_back(MDNode::get(Context, UnrollMD));
MDNode* NewUnrollMD = MDNode::get(Context, MDs);
// Set operand 0 to refer to the loop id itself.
NewUnrollMD->replaceOperandWith(0, NewUnrollMD);
return NewUnrollMD;
}
void ClampLoopUnroll::visitTerminatorInst(IGCLLVM::TerminatorInst& I)
{
MDNode* MD = I.getMetadata(LoopMDName);
if (!MD)
return;
if (!GetUnrollMetadata(MD, "llvm.loop.unroll.enable") &&
!GetUnrollMetadata(MD, "llvm.loop.unroll.full"))
return;
MDNode* NewMD = GetUnrollCountMD(I, m_MaxUnrollFactor);
I.setMetadata(LoopMDName, NewMD);
m_Changed = true;
}
bool ClampLoopUnroll::runOnFunction(Function& F)
{
#if 0
visit(F);
#endif
return m_Changed;
}
|
