This repository was archived by the owner on Jan 8, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 14
Expand file tree
/
Copy pathOptimizer.cpp
More file actions
74 lines (55 loc) · 2.67 KB
/
Optimizer.cpp
File metadata and controls
74 lines (55 loc) · 2.67 KB
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
#include "Optimization/Optimizer.h"
Optimizer::Optimizer(LLVMModule &module, unsigned int optLevel, unsigned int sizeLevel) : module(module),
optLevel(optLevel),
sizeLevel(sizeLevel) {}
void Optimizer::optimize() {
preoptimize();
llvm::Triple moduleTriple(getTargetTriple());
llvm::TargetLibraryInfoImpl tlii(moduleTriple);
llvm::codegen::setFunctionAttributes(llvm::codegen::getCPUStr(), llvm::codegen::getFeaturesStr(), module);
llvm::legacy::PassManager passes;
llvm::TargetMachine *targetMachine = getTargetMachine(optLevel);
passes.add(new llvm::TargetLibraryInfoWrapperPass(tlii));
passes.add(llvm::createTargetTransformInfoWrapperPass(targetMachine->getTargetIRAnalysis()));
llvm::legacy::FunctionPassManager funcPasses(&module);
funcPasses.add(llvm::createTargetTransformInfoWrapperPass(targetMachine->getTargetIRAnalysis()));
addStandardLinkPasses(passes);
addOptimizationPasses(passes, funcPasses, targetMachine);
funcPasses.doInitialization();
for (LLVMFunction &function : module) {
funcPasses.run(function);
}
funcPasses.doFinalization();
passes.add(llvm::createVerifierPass());
passes.run(module);
}
void Optimizer::preoptimize() {
for (LLVMFunction &function : module) {
if (function.getName() != "main" && !function.isDeclaration()) {
function.setLinkage(LLVMFunction::LinkageTypes::InternalLinkage);
}
}
}
void Optimizer::addOptimizationPasses(llvm::legacy::PassManager &passManager, llvm::legacy::FunctionPassManager &functionPassManager,
llvm::TargetMachine *targetMachine) const {
llvm::PassManagerBuilder builder;
builder.OptLevel = optLevel;
builder.SizeLevel = sizeLevel;
if (optLevel > 1) {
builder.Inliner = llvm::createFunctionInliningPass(optLevel, sizeLevel, false);
} else {
builder.Inliner = llvm::createAlwaysInlinerLegacyPass();
}
builder.DisableUnrollLoops = optLevel == 0;
builder.LoopVectorize = optLevel > 1 && sizeLevel < 2;
builder.SLPVectorize = optLevel > 1 && sizeLevel < 2;
targetMachine->adjustPassManager(builder);
builder.populateFunctionPassManager(functionPassManager);
builder.populateModulePassManager(passManager);
}
void Optimizer::addStandardLinkPasses(llvm::PassManagerBase &passManager) {
llvm::PassManagerBuilder Builder;
Builder.VerifyInput = true;
Builder.Inliner = llvm::createFunctionInliningPass();
Builder.populateLTOPassManager(passManager);
}