1//===-------------------- RISCVCustomBehaviour.h -----------------*-C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8/// \file
9///
10/// This file defines the RISCVCustomBehaviour class which inherits from
11/// CustomBehaviour. This class is used by the tool llvm-mca to enforce
12/// target specific behaviour that is not expressed well enough in the
13/// scheduling model for mca to enforce it automatically.
14///
15//===----------------------------------------------------------------------===//
16
17#ifndef LLVM_LIB_TARGET_RISCV_MCA_RISCVCUSTOMBEHAVIOUR_H
18#define LLVM_LIB_TARGET_RISCV_MCA_RISCVCUSTOMBEHAVIOUR_H
19
20#include "llvm/ADT/SmallVector.h"
21#include "llvm/MC/MCInst.h"
22#include "llvm/MC/MCInstrDesc.h"
23#include "llvm/MC/MCInstrInfo.h"
24#include "llvm/MCA/CustomBehaviour.h"
25
26namespace llvm {
27namespace mca {
28
29class RISCVLMULInstrument : public Instrument {
30public:
31 static const StringRef DESC_NAME;
32 static bool isDataValid(StringRef Data);
33
34 explicit RISCVLMULInstrument(StringRef Data) : Instrument(DESC_NAME, Data) {}
35
36 ~RISCVLMULInstrument() = default;
37
38 uint8_t getLMUL() const;
39};
40
41class RISCVSEWInstrument : public Instrument {
42public:
43 static const StringRef DESC_NAME;
44 static bool isDataValid(StringRef Data);
45
46 explicit RISCVSEWInstrument(StringRef Data) : Instrument(DESC_NAME, Data) {}
47
48 ~RISCVSEWInstrument() = default;
49
50 uint8_t getSEW() const;
51};
52
53class RISCVInstrumentManager : public InstrumentManager {
54public:
55 RISCVInstrumentManager(const MCSubtargetInfo &STI, const MCInstrInfo &MCII)
56 : InstrumentManager(STI, MCII) {}
57
58 bool shouldIgnoreInstruments() const override { return false; }
59 bool supportsInstrumentType(StringRef Type) const override;
60
61 /// Create a Instrument for RISC-V target
62 UniqueInstrument createInstrument(StringRef Desc, StringRef Data) override;
63
64 SmallVector<UniqueInstrument> createInstruments(const MCInst &Inst) override;
65
66 /// Using the Instrument, returns a SchedClassID to use instead of
67 /// the SchedClassID that belongs to the MCI or the original SchedClassID.
68 unsigned
69 getSchedClassID(const MCInstrInfo &MCII, const MCInst &MCI,
70 const SmallVector<Instrument *> &IVec) const override;
71};
72
73} // namespace mca
74} // namespace llvm
75
76#endif
77