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
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2017-2021 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
//
// GenXAnalysisDumper is a pass that calls the print() method on a function
// pass to dump its state out to a file.
// GenXGroupAnalysisDumper is the same, but for a function group pass.
//
//===----------------------------------------------------------------------===//
#include "FunctionGroup.h"
#include "GenX.h"
#include "vc/Support/BackendConfig.h"
#include "vc/Utils/GenX/KernelInfo.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/raw_ostream.h"
using namespace llvm;
namespace {
// GenXAnalysisDumper : a pass to dump an analysis to a file
class GenXAnalysisDumper : public FunctionPass {
FunctionPass *P;
std::string DumpNamePrefix;
std::string DumpNameSuffix;
public:
static char ID;
explicit GenXAnalysisDumper(FunctionPass *P, StringRef DumpNamePrefixIn,
StringRef DumpNameSuffixIn)
: FunctionPass(ID), P(P), DumpNamePrefix(DumpNamePrefixIn.str()),
DumpNameSuffix(DumpNameSuffixIn) {}
StringRef getPassName() const override { return "GenX analysis dumper pass"; }
void getAnalysisUsage(AnalysisUsage &AU) const override {
FunctionPass::getAnalysisUsage(AU);
AU.setPreservesAll();
}
bool runOnFunction(Function &F) override;
};
// GenXGroupAnalysisDumper : a pass to dump an analysis to a file
class GenXModuleAnalysisDumper : public ModulePass {
ModulePass *P;
std::string DumpNamePrefix;
std::string DumpNameSuffix;
public:
static char ID;
GenXModuleAnalysisDumper(ModulePass *P, StringRef DumpNamePrefixIn,
StringRef DumpNameSuffixIn)
: ModulePass(ID), P(P), DumpNamePrefix(DumpNamePrefixIn.str()),
DumpNameSuffix(DumpNameSuffixIn.str()) {}
StringRef getPassName() const override {
return "GenX Module analysis dumper pass";
}
void getAnalysisUsage(AnalysisUsage &AU) const override {
AU.addRequired<GenXBackendConfig>();
AU.setPreservesAll();
}
bool runOnModule(Module &M) override;
};
} // end anonymous namespace
char GenXAnalysisDumper::ID = 0;
FunctionPass *llvm::createGenXAnalysisDumperPass(FunctionPass *P,
StringRef DumpNamePrefix,
StringRef DumpNameSuffix) {
return new GenXAnalysisDumper(P, DumpNamePrefix, DumpNameSuffix);
}
char GenXModuleAnalysisDumper::ID = 0;
ModulePass *llvm::createGenXModuleAnalysisDumperPass(ModulePass *P,
StringRef DumpNamePrefix,
StringRef DumpNameSuffix) {
return new GenXModuleAnalysisDumper(P, DumpNamePrefix, DumpNameSuffix);
}
static std::string makeOutputName(const Function &F, StringRef DumpNamePrefix,
StringRef DumpNameSuffix) {
vc::KernelMetadata KM{&F};
StringRef Name = KM.getName();
if (Name.empty())
Name = F.getName();
std::string Filename = (DumpNamePrefix + Name + DumpNameSuffix).str();
// Sanitize templated kernel names.
std::replace_if(Filename.begin(), Filename.end(),
[](const char x) { return x == '<' || x == '>'; }, '_');
return Filename;
}
/***********************************************************************
* GenXAnalysisDumper::runOnFunction : dump analysis to file
*/
bool GenXAnalysisDumper::runOnFunction(Function &F) {
std::string SerializedData;
llvm::raw_string_ostream OS(SerializedData);
P->print(OS, F.getParent());
auto DumpName = makeOutputName(F, "f_" + DumpNamePrefix, DumpNameSuffix);
const auto &BC = getAnalysis<GenXBackendConfig>();
vc::produceAuxiliaryShaderDumpFile(BC, DumpName, OS.str());
return false;
}
/***********************************************************************
* GenXAnalysisDumper::runOnFunction : dump analysis to file
*/
bool GenXModuleAnalysisDumper::runOnModule(Module &M) {
std::string SerializedData;
llvm::raw_string_ostream OS(SerializedData);
P->print(OS, &M);
auto DumpName = DumpNamePrefix + "M_" + DumpNameSuffix;
const auto &BC = getAnalysis<GenXBackendConfig>();
vc::produceAuxiliaryShaderDumpFile(BC, DumpName, OS.str());
return false;
}
|
