blob: 0b46e9121e6f9593eb3be139e6105f225e221442 (
plain) (
blame)
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
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===- ConvergenceUtils.h -----------------------*- C++ -*-----------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
/// \file
/// \brief Convergence info and convergence-aware uniform info for LLVM IR
///
/// This differs from traditional divergence analysis by taking convergence
/// intrinsics into account.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_ANALYSIS_UNIFORMITYANALYSIS_H
#define LLVM_ANALYSIS_UNIFORMITYANALYSIS_H
#include "llvm/ADT/GenericUniformityInfo.h"
#include "llvm/Analysis/CycleAnalysis.h"
namespace llvm {
extern template class GenericUniformityInfo<SSAContext>;
using UniformityInfo = GenericUniformityInfo<SSAContext>;
/// Analysis pass which computes \ref UniformityInfo.
class UniformityInfoAnalysis
: public AnalysisInfoMixin<UniformityInfoAnalysis> {
friend AnalysisInfoMixin<UniformityInfoAnalysis>;
static AnalysisKey Key;
public:
/// Provide the result typedef for this analysis pass.
using Result = UniformityInfo;
/// Run the analysis pass over a function and produce a dominator tree.
UniformityInfo run(Function &F, FunctionAnalysisManager &);
// TODO: verify analysis
};
/// Printer pass for the \c UniformityInfo.
class UniformityInfoPrinterPass
: public PassInfoMixin<UniformityInfoPrinterPass> {
raw_ostream &OS;
public:
explicit UniformityInfoPrinterPass(raw_ostream &OS);
PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM);
};
/// Legacy analysis pass which computes a \ref CycleInfo.
class UniformityInfoWrapperPass : public FunctionPass {
Function *m_function = nullptr;
UniformityInfo m_uniformityInfo;
public:
static char ID;
UniformityInfoWrapperPass();
UniformityInfo &getUniformityInfo() { return m_uniformityInfo; }
const UniformityInfo &getUniformityInfo() const { return m_uniformityInfo; }
bool runOnFunction(Function &F) override;
void getAnalysisUsage(AnalysisUsage &AU) const override;
void releaseMemory() override;
void print(raw_ostream &OS, const Module *M = nullptr) const override;
// TODO: verify analysis
};
} // namespace llvm
#endif // LLVM_ANALYSIS_UNIFORMITYANALYSIS_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
