aboutsummaryrefslogtreecommitdiff
path: root/backend/utils/TableGen/GlobalISelEmitter.cpp
blob: 784d418b68e68f7d0f076e5ffe418eb9dbb56de5 (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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
// SPDX-License-Identifier: GPL-3.0-or-later

#include "TableGenBackends.h"

#include <cstddef>
#include <cstdint>
#include <ostream>
#include <sstream>
#include <string>
#include <string_view>

namespace rcc::targetgen {
namespace {

[[nodiscard]] std::string_view typeKindName(NormalizedSelectionTypeKind Kind) noexcept {
  switch (Kind) {
  case NormalizedSelectionTypeKind::Any:
    return "Any";
  case NormalizedSelectionTypeKind::Scalar:
    return "Scalar";
  case NormalizedSelectionTypeKind::Pointer:
    return "Pointer";
  }
  return "Any";
}

void emitInstruction(std::ostream &Output, std::string_view Opcode, std::uint32_t OperandIndex,
                     std::uint32_t Value = 0, std::string_view Auxiliary = "0",
                     std::int64_t Immediate = 0) {
  Output << "    {GIMatchTableOpcode::" << Opcode << ", " << OperandIndex << ", " << Value << ", "
         << Auxiliary << ", " << Immediate << "},\n";
}

void emitTypePredicate(std::ostream &Output, std::uint32_t OperandIndex,
                       const NormalizedSelectionTypePredicate &Predicate) {
  emitInstruction(Output, "CheckOperandType", OperandIndex, Predicate.Bits,
                  "static_cast<std::uint32_t>(GIMatchTypeKind::" +
                      std::string(typeKindName(Predicate.Kind)) + ")");
}

[[nodiscard]] std::size_t instructionCount(const NormalizedSelectionRule &Rule) noexcept {
  std::size_t Count = 2U + (Rule.ResultType ? 1U : 0U) +
                      (Rule.IntegerPredicate ? 1U : 0U) +
         Rule.OperandTypes.size() + Rule.OperandBanks.size() + Rule.OperandClasses.size() +
                      Rule.OperandDefConstants.size() + Rule.Temporaries.size() +
                      Rule.EraseOperandDefinitions.size();
  for (const NormalizedSelectionAction &Action : Rule.Actions)
    Count += 1U + Action.OperandRenderers.size();
  return Count;
}

} // namespace

std::string emitGlobalISel(const NormalizedTargetModel &Model, std::string_view SourceName) {
  std::size_t InstructionCount = 0;
  std::size_t ConstantRangeCount = 0;
  for (const NormalizedSelectionRule &Rule : Model.SelectionRules)
    InstructionCount += instructionCount(Rule);
  for (const NormalizedSelectionRule &Rule : Model.SelectionRules) {
    for (const NormalizedSelectionOperandDefConstantPredicate &Predicate :
         Rule.OperandDefConstants) {
      ConstantRangeCount += Predicate.Ranges.size();
    }
  }

  std::ostringstream Output;
  Output << "// SPDX-License-Identifier: GPL-3.0-or-later\n\n";
  Output << "// This file is generated by rcc-tblgen from ";
  appendQuoted(Output, SourceName);
  Output << ". Do not edit it manually.\n\n";
  Output << "#ifdef GET_GLOBALISEL_MATCH_TABLE\n\n";
  Output << "constexpr std::array<GIMatchTableConstantRange, " << ConstantRangeCount
         << "> GeneratedGlobalISelConstantRanges = {{\n";
  for (const NormalizedSelectionRule &Rule : Model.SelectionRules) {
    for (const NormalizedSelectionOperandDefConstantPredicate &Predicate :
         Rule.OperandDefConstants) {
      for (const NormalizedSelectionOperandDefConstantPredicate::Range &Range : Predicate.Ranges)
        Output << "    {" << Range.Minimum << ", " << Range.Maximum << "},\n";
    }
  }
  Output << "}};\n\n";

  Output << "constexpr std::array<GIMatchTableInstruction, " << InstructionCount
         << "> GeneratedGlobalISelInstructions = {{\n";
  std::size_t FirstConstantRange = 0;
  for (const NormalizedSelectionRule &Rule : Model.SelectionRules) {
    emitInstruction(Output, "CheckOpcode", 0, Rule.MatchOpcode);
    if (Rule.ResultType)
      emitTypePredicate(Output, 0, *Rule.ResultType);
    if (Rule.IntegerPredicate)
      emitInstruction(Output, "CheckIntegerPredicate", Rule.IntegerPredicate->OperandIndex,
                      Rule.IntegerPredicate->Predicate);
    for (const NormalizedSelectionOperandTypePredicate &Predicate : Rule.OperandTypes)
      emitTypePredicate(Output, Predicate.OperandIndex, Predicate.Type);
    for (const NormalizedSelectionOperandBankPredicate &Predicate : Rule.OperandBanks)
      emitInstruction(Output, "CheckOperandBank", Predicate.OperandIndex, Predicate.Bank);
    for (const NormalizedSelectionOperandClassPredicate &Predicate : Rule.OperandClasses)
      emitInstruction(Output, "CheckOperandClass", Predicate.OperandIndex, Predicate.RegClass);
    for (const NormalizedSelectionOperandDefConstantPredicate &Predicate :
         Rule.OperandDefConstants) {
      emitInstruction(Output, "CheckOperandDefConstantRanges", Predicate.OperandIndex,
                      static_cast<std::uint32_t>(FirstConstantRange),
                      std::to_string(Predicate.Ranges.size()));
      FirstConstantRange += Predicate.Ranges.size();
    }
    for (const NormalizedSelectionTemporary &Temporary : Rule.Temporaries)
      emitInstruction(Output, "CreateTemporary", Temporary.TypeFromOperand, Temporary.ID);
    for (const NormalizedSelectionAction &Action : Rule.Actions) {
      emitInstruction(Output, Action.ReplacesRoot ? "MutateOpcode" : "BuildInstruction", 0,
                      Action.ResultOpcode);
      for (const NormalizedSelectionOperandRenderer &Renderer : Action.OperandRenderers) {
        switch (Renderer.SourceKind) {
        case NormalizedSelectionOperandRenderer::Kind::Operand:
          emitInstruction(Output, "RenderOperand", Renderer.Value);
          break;
        case NormalizedSelectionOperandRenderer::Kind::PhysicalRegister:
          emitInstruction(Output, "RenderPhysicalRegister", 0, Renderer.Value,
                          std::to_string(Renderer.Flags));
          break;
        case NormalizedSelectionOperandRenderer::Kind::Temporary:
          emitInstruction(Output, "RenderTemporary", 0, Renderer.Value,
                          std::to_string(Renderer.Flags));
          break;
        case NormalizedSelectionOperandRenderer::Kind::CapturedImmediate:
          emitInstruction(Output, "RenderCapturedImmediate", Renderer.Value,
                          static_cast<std::uint32_t>(Renderer.Transform));
          break;
        case NormalizedSelectionOperandRenderer::Kind::Immediate:
          emitInstruction(Output, "RenderImmediate", 0, 0, "0", Renderer.Immediate);
          break;
        }
      }
    }
    for (const std::uint32_t OperandIndex : Rule.EraseOperandDefinitions)
      emitInstruction(Output, "EraseOperandDefinition", OperandIndex);
    emitInstruction(Output, "Complete", 0);
  }
  Output << "}};\n\n";

  Output << "constexpr std::array<GIMatchTableRule, " << Model.SelectionRules.size()
         << "> GeneratedGlobalISelRules = {{\n";
  std::size_t FirstInstruction = 0;
  for (const NormalizedSelectionRule &Rule : Model.SelectionRules) {
    const std::size_t Count = instructionCount(Rule);
    Output << "    {";
    appendQuoted(Output, Rule.Symbol);
    Output << ", " << Rule.Priority << ", " << FirstInstruction << ", " << Count << "},\n";
    FirstInstruction += Count;
  }
  Output << "}};\n\n";
  Output << "[[nodiscard]] constexpr GIMatchTable generatedGlobalISelMatchTable() noexcept {\n"
            "  return {GeneratedGlobalISelInstructions, GeneratedGlobalISelRules,\n"
            "          GeneratedGlobalISelConstantRanges};\n"
            "}\n\n";
  Output << "#endif // GET_GLOBALISEL_MATCH_TABLE\n";
  return Output.str();
}

} // namespace rcc::targetgen