1 //===- MachineSSAContext.cpp ------------------------------------*- 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 a specialization of the GenericSSAContext<X>
11 /// template class for Machine IR.
12 ///
13 //===----------------------------------------------------------------------===//
14
15 #include "llvm/CodeGen/MachineSSAContext.h"
16 #include "llvm/CodeGen/GlobalISel/GenericMachineInstrs.h"
17 #include "llvm/CodeGen/MachineBasicBlock.h"
18 #include "llvm/CodeGen/MachineFunction.h"
19 #include "llvm/CodeGen/MachineInstr.h"
20 #include "llvm/CodeGen/MachineRegisterInfo.h"
21 #include "llvm/Support/raw_ostream.h"
22
23 using namespace llvm;
24
25 template <>
appendBlockDefs(SmallVectorImpl<Register> & defs,const MachineBasicBlock & block)26 void MachineSSAContext::appendBlockDefs(SmallVectorImpl<Register> &defs,
27 const MachineBasicBlock &block) {
28 for (auto &instr : block.instrs()) {
29 for (auto &op : instr.all_defs())
30 defs.push_back(op.getReg());
31 }
32 }
33
34 template <>
appendBlockTerms(SmallVectorImpl<MachineInstr * > & terms,MachineBasicBlock & block)35 void MachineSSAContext::appendBlockTerms(SmallVectorImpl<MachineInstr *> &terms,
36 MachineBasicBlock &block) {
37 for (auto &T : block.terminators())
38 terms.push_back(&T);
39 }
40
41 template <>
appendBlockTerms(SmallVectorImpl<const MachineInstr * > & terms,const MachineBasicBlock & block)42 void MachineSSAContext::appendBlockTerms(
43 SmallVectorImpl<const MachineInstr *> &terms,
44 const MachineBasicBlock &block) {
45 for (auto &T : block.terminators())
46 terms.push_back(&T);
47 }
48
49 /// Get the defining block of a value.
50 template <>
getDefBlock(Register value) const51 const MachineBasicBlock *MachineSSAContext::getDefBlock(Register value) const {
52 if (!value)
53 return nullptr;
54 return F->getRegInfo().getVRegDef(value)->getParent();
55 }
56
57 template <>
isConstantOrUndefValuePhi(const MachineInstr & Phi)58 bool MachineSSAContext::isConstantOrUndefValuePhi(const MachineInstr &Phi) {
59 return Phi.isConstantValuePHI();
60 }
61
62 template <>
getIntrinsicID(const MachineInstr & MI)63 Intrinsic::ID MachineSSAContext::getIntrinsicID(const MachineInstr &MI) {
64 if (auto *GI = dyn_cast<GIntrinsic>(&MI))
65 return GI->getIntrinsicID();
66 return Intrinsic::not_intrinsic;
67 }
68
69 template <>
print(const MachineBasicBlock * Block) const70 Printable MachineSSAContext::print(const MachineBasicBlock *Block) const {
71 if (!Block)
72 return Printable([](raw_ostream &Out) { Out << "<nullptr>"; });
73 return Printable([Block](raw_ostream &Out) { Block->printName(Out); });
74 }
75
print(const MachineInstr * I) const76 template <> Printable MachineSSAContext::print(const MachineInstr *I) const {
77 return Printable([I](raw_ostream &Out) { I->print(Out); });
78 }
79
print(Register Value) const80 template <> Printable MachineSSAContext::print(Register Value) const {
81 auto *MRI = &F->getRegInfo();
82 return Printable([MRI, Value](raw_ostream &Out) {
83 Out << printReg(Value, MRI->getTargetRegisterInfo(), 0, MRI);
84
85 if (Value) {
86 // Try to print the definition.
87 if (auto *Instr = MRI->getUniqueVRegDef(Value)) {
88 Out << ": ";
89 Instr->print(Out);
90 }
91 }
92 });
93 }
94
95 template <>
printAsOperand(const MachineBasicBlock * BB) const96 Printable MachineSSAContext::printAsOperand(const MachineBasicBlock *BB) const {
97 return Printable([BB](raw_ostream &Out) { BB->printAsOperand(Out); });
98 }
99