1 //===--- CodeGenHwModes.cpp -----------------------------------------------===//
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 // Classes to parse and store HW mode information for instruction selection
9 //===----------------------------------------------------------------------===//
10 
11 #include "CodeGenHwModes.h"
12 #include "llvm/Support/Debug.h"
13 #include "llvm/Support/raw_ostream.h"
14 #include "llvm/TableGen/Error.h"
15 #include "llvm/TableGen/Record.h"
16 
17 using namespace llvm;
18 
19 StringRef CodeGenHwModes::DefaultModeName = "DefaultMode";
20 
21 HwMode::HwMode(Record *R) {
22   Name = R->getName();
23   Features = std::string(R->getValueAsString("Features"));
24 }
25 
26 LLVM_DUMP_METHOD
27 void HwMode::dump() const {
28   dbgs() << Name << ": " << Features << '\n';
29 }
30 
31 HwModeSelect::HwModeSelect(Record *R, CodeGenHwModes &CGH) {
32   std::vector<Record*> Modes = R->getValueAsListOfDefs("Modes");
33   std::vector<Record*> Objects = R->getValueAsListOfDefs("Objects");
34   if (Modes.size() != Objects.size()) {
35     PrintError(R->getLoc(), "in record " + R->getName() +
36         " derived from HwModeSelect: the lists Modes and Objects should "
37         "have the same size");
38     report_fatal_error("error in target description.");
39   }
40   for (unsigned i = 0, e = Modes.size(); i != e; ++i) {
41     unsigned ModeId = CGH.getHwModeId(Modes[i]->getName());
42     Items.push_back(std::make_pair(ModeId, Objects[i]));
43   }
44 }
45 
46 LLVM_DUMP_METHOD
47 void HwModeSelect::dump() const {
48   dbgs() << '{';
49   for (const PairType &P : Items)
50     dbgs() << " (" << P.first << ',' << P.second->getName() << ')';
51   dbgs() << " }\n";
52 }
53 
54 CodeGenHwModes::CodeGenHwModes(RecordKeeper &RK) : Records(RK) {
55   std::vector<Record*> MRs = Records.getAllDerivedDefinitions("HwMode");
56   // The default mode needs a definition in the .td sources for TableGen
57   // to accept references to it. We need to ignore the definition here.
58   for (auto I = MRs.begin(), E = MRs.end(); I != E; ++I) {
59     if ((*I)->getName() != DefaultModeName)
60       continue;
61     MRs.erase(I);
62     break;
63   }
64 
65   for (Record *R : MRs) {
66     Modes.emplace_back(R);
67     unsigned NewId = Modes.size();
68     ModeIds.insert(std::make_pair(Modes[NewId-1].Name, NewId));
69   }
70 
71   std::vector<Record*> MSs = Records.getAllDerivedDefinitions("HwModeSelect");
72   for (Record *R : MSs) {
73     auto P = ModeSelects.emplace(std::make_pair(R, HwModeSelect(R, *this)));
74     assert(P.second);
75     (void)P;
76   }
77 }
78 
79 unsigned CodeGenHwModes::getHwModeId(StringRef Name) const {
80   if (Name == DefaultModeName)
81     return DefaultMode;
82   auto F = ModeIds.find(Name);
83   assert(F != ModeIds.end() && "Unknown mode name");
84   return F->second;
85 }
86 
87 const HwModeSelect &CodeGenHwModes::getHwModeSelect(Record *R) const {
88   auto F = ModeSelects.find(R);
89   assert(F != ModeSelects.end() && "Record is not a \"mode select\"");
90   return F->second;
91 }
92 
93 LLVM_DUMP_METHOD
94 void CodeGenHwModes::dump() const {
95   dbgs() << "Modes: {\n";
96   for (const HwMode &M : Modes) {
97     dbgs() << "  ";
98     M.dump();
99   }
100   dbgs() << "}\n";
101 
102   dbgs() << "ModeIds: {\n";
103   for (const auto &P : ModeIds)
104     dbgs() << "  " << P.first() << " -> " << P.second << '\n';
105   dbgs() << "}\n";
106 
107   dbgs() << "ModeSelects: {\n";
108   for (const auto &P : ModeSelects) {
109     dbgs() << "  " << P.first->getName() << " -> ";
110     P.second.dump();
111   }
112   dbgs() << "}\n";
113 }
114