summaryrefslogtreecommitdiffstats
path: root/llvm/tools/llvm-mca/InstrBuilder.h
blob: 69a53b6fec210ade31d5fbae0176bf03e4b96e95 (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
//===--------------------- InstrBuilder.h -----------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
/// \file
///
/// A builder class for instructions that are statically analyzed by llvm-mca.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_TOOLS_LLVM_MCA_INSTRBUILDER_H
#define LLVM_TOOLS_LLVM_MCA_INSTRBUILDER_H

#include "Instruction.h"
#include "Support.h"
#include "llvm/MC/MCInstPrinter.h"
#include "llvm/MC/MCInstrAnalysis.h"
#include "llvm/MC/MCInstrInfo.h"
#include "llvm/MC/MCRegisterInfo.h"
#include "llvm/MC/MCSubtargetInfo.h"

namespace mca {

class DispatchUnit;

/// A builder class that knows how to construct Instruction objects.
///
/// Every llvm-mca Instruction is described by an object of class InstrDesc.
/// An InstrDesc describes which registers are read/written by the instruction,
/// as well as the instruction latency and hardware resources consumed.
///
/// This class is used by the tool to construct Instructions and instruction
/// descriptors (i.e. InstrDesc objects).
/// Information from the machine scheduling model is used to identify processor
/// resources that are consumed by an instruction.
class InstrBuilder {
  const llvm::MCSubtargetInfo &STI;
  const llvm::MCInstrInfo &MCII;
  const llvm::MCRegisterInfo &MRI;
  const llvm::MCInstrAnalysis &MCIA;
  llvm::MCInstPrinter &MCIP;
  llvm::SmallVector<uint64_t, 8> ProcResourceMasks;

  llvm::DenseMap<unsigned short, std::unique_ptr<const InstrDesc>> Descriptors;
  llvm::DenseMap<const llvm::MCInst *, std::unique_ptr<const InstrDesc>>
      VariantDescriptors;

  const InstrDesc &createInstrDescImpl(const llvm::MCInst &MCI);
  InstrBuilder(const InstrBuilder &) = delete;
  InstrBuilder &operator=(const InstrBuilder &) = delete;

  void populateWrites(InstrDesc &ID, const llvm::MCInst &MCI,
                      unsigned SchedClassID);
  void populateReads(InstrDesc &ID, const llvm::MCInst &MCI,
                     unsigned SchedClassID);

public:
  InstrBuilder(const llvm::MCSubtargetInfo &sti, const llvm::MCInstrInfo &mcii,
               const llvm::MCRegisterInfo &mri,
               const llvm::MCInstrAnalysis &mcia, llvm::MCInstPrinter &mcip)
      : STI(sti), MCII(mcii), MRI(mri), MCIA(mcia), MCIP(mcip),
        ProcResourceMasks(STI.getSchedModel().getNumProcResourceKinds()) {
    computeProcResourceMasks(STI.getSchedModel(), ProcResourceMasks);
  }

  const InstrDesc &getOrCreateInstrDesc(const llvm::MCInst &MCI);
  // Returns an array of processor resource masks.
  // Masks are computed by function mca::computeProcResourceMasks. see
  // Support.h for a description of how masks are computed and how masks can be
  // used to solve set membership problems.
  llvm::ArrayRef<uint64_t> getProcResourceMasks() const {
    return ProcResourceMasks;
  }

  void clear() { VariantDescriptors.shrink_and_clear(); }

  std::unique_ptr<Instruction> createInstruction(const llvm::MCInst &MCI);
};
} // namespace mca

#endif
OpenPOWER on IntegriCloud