blob: c9ca1b295b3b64a56e9d1c1b17a34808b22818e2 (
plain)
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
|
//===---------------------- Stage.h -----------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
/// \file
///
/// This file defines a stage.
/// A chain of stages compose an instruction pipeline.
///
//===----------------------------------------------------------------------===//
#ifndef LLVM_TOOLS_LLVM_MCA_STAGE_H
#define LLVM_TOOLS_LLVM_MCA_STAGE_H
#include "HWEventListener.h"
#include "llvm/Support/Error.h"
#include <set>
namespace mca {
class InstRef;
class Stage {
std::set<HWEventListener *> Listeners;
Stage(const Stage &Other) = delete;
Stage &operator=(const Stage &Other) = delete;
public:
/// A Stage's execute() returns Continue, Stop, or an error. Returning
/// Continue means that the stage successfully completed its 'execute'
/// action, and that the instruction being processed can be moved to the next
/// pipeline stage during this cycle. Continue allows the pipeline to
/// continue calling 'execute' on subsequent stages. Returning Stop
/// signifies that the stage ran into an error, and tells the pipeline to stop
/// passing the instruction to subsequent stages during this cycle. Any
/// failures that occur during 'execute' are represented by the error variant
/// that is provided by the Expected template.
enum State { Stop, Continue };
using Status = llvm::Expected<State>;
protected:
const std::set<HWEventListener *> &getListeners() const { return Listeners; }
public:
Stage() {}
virtual ~Stage();
/// Called prior to preExecute to ensure that the stage has items that it
/// is to process. For example, a FetchStage might have more instructions
/// that need to be processed, or a RCU might have items that have yet to
/// retire.
virtual bool hasWorkToComplete() const = 0;
/// Called once at the start of each cycle. This can be used as a setup
/// phase to prepare for the executions during the cycle.
virtual llvm::Error cycleStart() { return llvm::ErrorSuccess(); }
/// Called once at the end of each cycle.
virtual llvm::Error cycleEnd() { return llvm::ErrorSuccess(); }
/// Called prior to executing the list of stages.
/// This can be called multiple times per cycle.
virtual void preExecute() {}
/// Called as a cleanup and finalization phase after each execution.
/// This will only be called if all stages return a success from their
/// execute callback. This can be called multiple times per cycle.
virtual void postExecute() {}
/// The primary action that this stage performs.
/// Returning false prevents successor stages from having their 'execute'
/// routine called. This can be called multiple times during a single cycle.
virtual Status execute(InstRef &IR) = 0;
/// Add a listener to receive callbacks during the execution of this stage.
void addListener(HWEventListener *Listener);
/// Notify listeners of a particular hardware event.
template <typename EventT> void notifyEvent(const EventT &Event) const {
for (HWEventListener *Listener : Listeners)
Listener->onEvent(Event);
}
};
} // namespace mca
#endif // LLVM_TOOLS_LLVM_MCA_STAGE_H
|