blob: 097a9db9e14a41f6df8038f2200dc4964fd89edb (
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===---------------- IncrementalSourceMgr.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
/// This file contains IncrementalSourceMgr, an implementation of SourceMgr
/// that allows users to add new instructions incrementally / dynamically.
///
//===----------------------------------------------------------------------===//
#ifndef LLVM_MCA_INCREMENTALSOURCEMGR_H
#define LLVM_MCA_INCREMENTALSOURCEMGR_H
#include "llvm/MCA/SourceMgr.h"
#include <deque>
namespace llvm {
namespace mca {
/// An implementation of \a SourceMgr that allows users to add new instructions
/// incrementally / dynamically.
/// Note that this SourceMgr takes ownership of all \a mca::Instruction.
class IncrementalSourceMgr : public SourceMgr {
/// Owner of all mca::Instruction instances. Note that we use std::deque here
/// to have a better throughput, in comparison to std::vector or
/// llvm::SmallVector, as they usually pay a higher re-allocation cost when
/// there is a large number of instructions.
std::deque<UniqueInst> InstStorage;
/// Instructions that are ready to be used. Each of them is a pointer of an
/// \a UniqueInst inside InstStorage.
std::deque<Instruction *> Staging;
/// Current instruction index.
unsigned TotalCounter;
/// End-of-stream flag.
bool EOS;
/// Called when an instruction is no longer needed.
using InstFreedCallback = llvm::function_ref<void(Instruction *)>;
InstFreedCallback InstFreedCB;
public:
IncrementalSourceMgr() : TotalCounter(0U), EOS(false) {}
void clear();
/// Set a callback that is invoked when a mca::Instruction is
/// no longer needed. This is usually used for recycling the
/// instruction.
void setOnInstFreedCallback(InstFreedCallback CB) { InstFreedCB = CB; }
ArrayRef<UniqueInst> getInstructions() const override {
llvm_unreachable("Not applicable");
}
bool hasNext() const override { return !Staging.empty(); }
bool isEnd() const override { return EOS; }
SourceRef peekNext() const override {
assert(hasNext());
return SourceRef(TotalCounter, *Staging.front());
}
/// Add a new instruction.
void addInst(UniqueInst &&Inst) {
InstStorage.emplace_back(std::move(Inst));
Staging.push_back(InstStorage.back().get());
}
/// Add a recycled instruction.
void addRecycledInst(Instruction *Inst) { Staging.push_back(Inst); }
void updateNext() override;
/// Mark the end of instruction stream.
void endOfStream() { EOS = true; }
#ifndef NDEBUG
/// Print statistic about instruction recycling stats.
void printStatistic(raw_ostream &OS);
#endif
};
} // end namespace mca
} // end namespace llvm
#endif // LLVM_MCA_INCREMENTALSOURCEMGR_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
