xref: /freebsd/contrib/llvm-project/llvm/lib/CodeGen/MachineOperand.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
1 //===- lib/CodeGen/MachineOperand.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 //
9 /// \file Methods common to all machine operands.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "llvm/CodeGen/MachineOperand.h"
14 #include "llvm/ADT/StableHashing.h"
15 #include "llvm/ADT/StringExtras.h"
16 #include "llvm/Analysis/Loads.h"
17 #include "llvm/CodeGen/MIRFormatter.h"
18 #include "llvm/CodeGen/MachineFrameInfo.h"
19 #include "llvm/CodeGen/MachineJumpTableInfo.h"
20 #include "llvm/CodeGen/MachineRegisterInfo.h"
21 #include "llvm/CodeGen/PseudoSourceValueManager.h"
22 #include "llvm/CodeGen/TargetInstrInfo.h"
23 #include "llvm/CodeGen/TargetRegisterInfo.h"
24 #include "llvm/Config/llvm-config.h"
25 #include "llvm/IR/Constants.h"
26 #include "llvm/IR/IRPrintingPasses.h"
27 #include "llvm/IR/Instructions.h"
28 #include "llvm/IR/ModuleSlotTracker.h"
29 #include "llvm/MC/MCDwarf.h"
30 #include "llvm/Target/TargetIntrinsicInfo.h"
31 #include "llvm/Target/TargetMachine.h"
32 #include <optional>
33 
34 using namespace llvm;
35 
36 static cl::opt<int>
37     PrintRegMaskNumRegs("print-regmask-num-regs",
38                         cl::desc("Number of registers to limit to when "
39                                  "printing regmask operands in IR dumps. "
40                                  "unlimited = -1"),
41                         cl::init(32), cl::Hidden);
42 
getMFIfAvailable(const MachineOperand & MO)43 static const MachineFunction *getMFIfAvailable(const MachineOperand &MO) {
44   if (const MachineInstr *MI = MO.getParent())
45     if (const MachineBasicBlock *MBB = MI->getParent())
46       if (const MachineFunction *MF = MBB->getParent())
47         return MF;
48   return nullptr;
49 }
50 
getMFIfAvailable(MachineOperand & MO)51 static MachineFunction *getMFIfAvailable(MachineOperand &MO) {
52   return const_cast<MachineFunction *>(
53       getMFIfAvailable(const_cast<const MachineOperand &>(MO)));
54 }
55 
getOperandNo() const56 unsigned MachineOperand::getOperandNo() const {
57   assert(getParent() && "Operand does not belong to any instruction!");
58   return getParent()->getOperandNo(this);
59 }
60 
setReg(Register Reg)61 void MachineOperand::setReg(Register Reg) {
62   if (getReg() == Reg)
63     return; // No change.
64 
65   // Clear the IsRenamable bit to keep it conservatively correct.
66   IsRenamable = false;
67 
68   // Otherwise, we have to change the register.  If this operand is embedded
69   // into a machine function, we need to update the old and new register's
70   // use/def lists.
71   if (MachineFunction *MF = getMFIfAvailable(*this)) {
72     MachineRegisterInfo &MRI = MF->getRegInfo();
73     MRI.removeRegOperandFromUseList(this);
74     SmallContents.RegNo = Reg;
75     MRI.addRegOperandToUseList(this);
76     return;
77   }
78 
79   // Otherwise, just change the register, no problem.  :)
80   SmallContents.RegNo = Reg;
81 }
82 
substVirtReg(Register Reg,unsigned SubIdx,const TargetRegisterInfo & TRI)83 void MachineOperand::substVirtReg(Register Reg, unsigned SubIdx,
84                                   const TargetRegisterInfo &TRI) {
85   assert(Reg.isVirtual());
86   if (SubIdx && getSubReg())
87     SubIdx = TRI.composeSubRegIndices(SubIdx, getSubReg());
88   setReg(Reg);
89   if (SubIdx)
90     setSubReg(SubIdx);
91 }
92 
substPhysReg(MCRegister Reg,const TargetRegisterInfo & TRI)93 void MachineOperand::substPhysReg(MCRegister Reg, const TargetRegisterInfo &TRI) {
94   assert(Register::isPhysicalRegister(Reg));
95   if (getSubReg()) {
96     Reg = TRI.getSubReg(Reg, getSubReg());
97     // Note that getSubReg() may return 0 if the sub-register doesn't exist.
98     // That won't happen in legal code.
99     setSubReg(0);
100     if (isDef())
101       setIsUndef(false);
102   }
103   setReg(Reg);
104 }
105 
106 /// Change a def to a use, or a use to a def.
setIsDef(bool Val)107 void MachineOperand::setIsDef(bool Val) {
108   assert(isReg() && "Wrong MachineOperand accessor");
109   assert((!Val || !isDebug()) && "Marking a debug operation as def");
110   if (IsDef == Val)
111     return;
112   assert(!IsDeadOrKill && "Changing def/use with dead/kill set not supported");
113   // MRI may keep uses and defs in different list positions.
114   if (MachineFunction *MF = getMFIfAvailable(*this)) {
115     MachineRegisterInfo &MRI = MF->getRegInfo();
116     MRI.removeRegOperandFromUseList(this);
117     IsDef = Val;
118     MRI.addRegOperandToUseList(this);
119     return;
120   }
121   IsDef = Val;
122 }
123 
isRenamable() const124 bool MachineOperand::isRenamable() const {
125   assert(isReg() && "Wrong MachineOperand accessor");
126   assert(getReg().isPhysical() &&
127          "isRenamable should only be checked on physical registers");
128   if (!IsRenamable)
129     return false;
130 
131   const MachineInstr *MI = getParent();
132   if (!MI)
133     return true;
134 
135   if (isDef())
136     return !MI->hasExtraDefRegAllocReq(MachineInstr::IgnoreBundle);
137 
138   assert(isUse() && "Reg is not def or use");
139   return !MI->hasExtraSrcRegAllocReq(MachineInstr::IgnoreBundle);
140 }
141 
setIsRenamable(bool Val)142 void MachineOperand::setIsRenamable(bool Val) {
143   assert(isReg() && "Wrong MachineOperand accessor");
144   assert(getReg().isPhysical() &&
145          "setIsRenamable should only be called on physical registers");
146   IsRenamable = Val;
147 }
148 
149 // If this operand is currently a register operand, and if this is in a
150 // function, deregister the operand from the register's use/def list.
removeRegFromUses()151 void MachineOperand::removeRegFromUses() {
152   if (!isReg() || !isOnRegUseList())
153     return;
154 
155   if (MachineFunction *MF = getMFIfAvailable(*this))
156     MF->getRegInfo().removeRegOperandFromUseList(this);
157 }
158 
159 /// ChangeToImmediate - Replace this operand with a new immediate operand of
160 /// the specified value.  If an operand is known to be an immediate already,
161 /// the setImm method should be used.
ChangeToImmediate(int64_t ImmVal,unsigned TargetFlags)162 void MachineOperand::ChangeToImmediate(int64_t ImmVal, unsigned TargetFlags) {
163   assert((!isReg() || !isTied()) && "Cannot change a tied operand into an imm");
164 
165   removeRegFromUses();
166 
167   OpKind = MO_Immediate;
168   Contents.ImmVal = ImmVal;
169   setTargetFlags(TargetFlags);
170 }
171 
ChangeToFPImmediate(const ConstantFP * FPImm,unsigned TargetFlags)172 void MachineOperand::ChangeToFPImmediate(const ConstantFP *FPImm,
173                                          unsigned TargetFlags) {
174   assert((!isReg() || !isTied()) && "Cannot change a tied operand into an imm");
175 
176   removeRegFromUses();
177 
178   OpKind = MO_FPImmediate;
179   Contents.CFP = FPImm;
180   setTargetFlags(TargetFlags);
181 }
182 
ChangeToES(const char * SymName,unsigned TargetFlags)183 void MachineOperand::ChangeToES(const char *SymName,
184                                 unsigned TargetFlags) {
185   assert((!isReg() || !isTied()) &&
186          "Cannot change a tied operand into an external symbol");
187 
188   removeRegFromUses();
189 
190   OpKind = MO_ExternalSymbol;
191   Contents.OffsetedInfo.Val.SymbolName = SymName;
192   setOffset(0); // Offset is always 0.
193   setTargetFlags(TargetFlags);
194 }
195 
ChangeToGA(const GlobalValue * GV,int64_t Offset,unsigned TargetFlags)196 void MachineOperand::ChangeToGA(const GlobalValue *GV, int64_t Offset,
197                                 unsigned TargetFlags) {
198   assert((!isReg() || !isTied()) &&
199          "Cannot change a tied operand into a global address");
200 
201   removeRegFromUses();
202 
203   OpKind = MO_GlobalAddress;
204   Contents.OffsetedInfo.Val.GV = GV;
205   setOffset(Offset);
206   setTargetFlags(TargetFlags);
207 }
208 
ChangeToBA(const BlockAddress * BA,int64_t Offset,unsigned TargetFlags)209 void MachineOperand::ChangeToBA(const BlockAddress *BA, int64_t Offset,
210                                 unsigned TargetFlags) {
211   assert((!isReg() || !isTied()) &&
212          "Cannot change a tied operand into a block address");
213 
214   removeRegFromUses();
215 
216   OpKind = MO_BlockAddress;
217   Contents.OffsetedInfo.Val.BA = BA;
218   setOffset(Offset);
219   setTargetFlags(TargetFlags);
220 }
221 
ChangeToMCSymbol(MCSymbol * Sym,unsigned TargetFlags)222 void MachineOperand::ChangeToMCSymbol(MCSymbol *Sym, unsigned TargetFlags) {
223   assert((!isReg() || !isTied()) &&
224          "Cannot change a tied operand into an MCSymbol");
225 
226   removeRegFromUses();
227 
228   OpKind = MO_MCSymbol;
229   Contents.Sym = Sym;
230   setTargetFlags(TargetFlags);
231 }
232 
ChangeToFrameIndex(int Idx,unsigned TargetFlags)233 void MachineOperand::ChangeToFrameIndex(int Idx, unsigned TargetFlags) {
234   assert((!isReg() || !isTied()) &&
235          "Cannot change a tied operand into a FrameIndex");
236 
237   removeRegFromUses();
238 
239   OpKind = MO_FrameIndex;
240   setIndex(Idx);
241   setTargetFlags(TargetFlags);
242 }
243 
ChangeToTargetIndex(unsigned Idx,int64_t Offset,unsigned TargetFlags)244 void MachineOperand::ChangeToTargetIndex(unsigned Idx, int64_t Offset,
245                                          unsigned TargetFlags) {
246   assert((!isReg() || !isTied()) &&
247          "Cannot change a tied operand into a FrameIndex");
248 
249   removeRegFromUses();
250 
251   OpKind = MO_TargetIndex;
252   setIndex(Idx);
253   setOffset(Offset);
254   setTargetFlags(TargetFlags);
255 }
256 
ChangeToDbgInstrRef(unsigned InstrIdx,unsigned OpIdx,unsigned TargetFlags)257 void MachineOperand::ChangeToDbgInstrRef(unsigned InstrIdx, unsigned OpIdx,
258                                          unsigned TargetFlags) {
259   assert((!isReg() || !isTied()) &&
260          "Cannot change a tied operand into a DbgInstrRef");
261 
262   removeRegFromUses();
263 
264   OpKind = MO_DbgInstrRef;
265   setInstrRefInstrIndex(InstrIdx);
266   setInstrRefOpIndex(OpIdx);
267   setTargetFlags(TargetFlags);
268 }
269 
270 /// ChangeToRegister - Replace this operand with a new register operand of
271 /// the specified value.  If an operand is known to be an register already,
272 /// the setReg method should be used.
ChangeToRegister(Register Reg,bool isDef,bool isImp,bool isKill,bool isDead,bool isUndef,bool isDebug)273 void MachineOperand::ChangeToRegister(Register Reg, bool isDef, bool isImp,
274                                       bool isKill, bool isDead, bool isUndef,
275                                       bool isDebug) {
276   MachineRegisterInfo *RegInfo = nullptr;
277   if (MachineFunction *MF = getMFIfAvailable(*this))
278     RegInfo = &MF->getRegInfo();
279   // If this operand is already a register operand, remove it from the
280   // register's use/def lists.
281   bool WasReg = isReg();
282   if (RegInfo && WasReg)
283     RegInfo->removeRegOperandFromUseList(this);
284 
285   // Ensure debug instructions set debug flag on register uses.
286   const MachineInstr *MI = getParent();
287   if (!isDef && MI && MI->isDebugInstr())
288     isDebug = true;
289 
290   // Change this to a register and set the reg#.
291   assert(!(isDead && !isDef) && "Dead flag on non-def");
292   assert(!(isKill && isDef) && "Kill flag on def");
293   OpKind = MO_Register;
294   SmallContents.RegNo = Reg;
295   SubReg_TargetFlags = 0;
296   IsDef = isDef;
297   IsImp = isImp;
298   IsDeadOrKill = isKill | isDead;
299   IsRenamable = false;
300   IsUndef = isUndef;
301   IsInternalRead = false;
302   IsEarlyClobber = false;
303   IsDebug = isDebug;
304   // Ensure isOnRegUseList() returns false.
305   Contents.Reg.Prev = nullptr;
306   // Preserve the tie when the operand was already a register.
307   if (!WasReg)
308     TiedTo = 0;
309 
310   // If this operand is embedded in a function, add the operand to the
311   // register's use/def list.
312   if (RegInfo)
313     RegInfo->addRegOperandToUseList(this);
314 }
315 
316 /// isIdenticalTo - Return true if this operand is identical to the specified
317 /// operand. Note that this should stay in sync with the hash_value overload
318 /// below.
isIdenticalTo(const MachineOperand & Other) const319 bool MachineOperand::isIdenticalTo(const MachineOperand &Other) const {
320   if (getType() != Other.getType() ||
321       getTargetFlags() != Other.getTargetFlags())
322     return false;
323 
324   switch (getType()) {
325   case MachineOperand::MO_Register:
326     return getReg() == Other.getReg() && isDef() == Other.isDef() &&
327            getSubReg() == Other.getSubReg();
328   case MachineOperand::MO_Immediate:
329     return getImm() == Other.getImm();
330   case MachineOperand::MO_CImmediate:
331     return getCImm() == Other.getCImm();
332   case MachineOperand::MO_FPImmediate:
333     return getFPImm() == Other.getFPImm();
334   case MachineOperand::MO_MachineBasicBlock:
335     return getMBB() == Other.getMBB();
336   case MachineOperand::MO_FrameIndex:
337     return getIndex() == Other.getIndex();
338   case MachineOperand::MO_ConstantPoolIndex:
339   case MachineOperand::MO_TargetIndex:
340     return getIndex() == Other.getIndex() && getOffset() == Other.getOffset();
341   case MachineOperand::MO_JumpTableIndex:
342     return getIndex() == Other.getIndex();
343   case MachineOperand::MO_GlobalAddress:
344     return getGlobal() == Other.getGlobal() && getOffset() == Other.getOffset();
345   case MachineOperand::MO_ExternalSymbol:
346     return strcmp(getSymbolName(), Other.getSymbolName()) == 0 &&
347            getOffset() == Other.getOffset();
348   case MachineOperand::MO_BlockAddress:
349     return getBlockAddress() == Other.getBlockAddress() &&
350            getOffset() == Other.getOffset();
351   case MachineOperand::MO_RegisterMask:
352   case MachineOperand::MO_RegisterLiveOut: {
353     // Shallow compare of the two RegMasks
354     const uint32_t *RegMask = getRegMask();
355     const uint32_t *OtherRegMask = Other.getRegMask();
356     if (RegMask == OtherRegMask)
357       return true;
358 
359     if (const MachineFunction *MF = getMFIfAvailable(*this)) {
360       const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
361       unsigned RegMaskSize = MachineOperand::getRegMaskSize(TRI->getNumRegs());
362       // Deep compare of the two RegMasks
363       return std::equal(RegMask, RegMask + RegMaskSize, OtherRegMask);
364     }
365     // We don't know the size of the RegMask, so we can't deep compare the two
366     // reg masks.
367     return false;
368   }
369   case MachineOperand::MO_MCSymbol:
370     return getMCSymbol() == Other.getMCSymbol();
371   case MachineOperand::MO_DbgInstrRef:
372     return getInstrRefInstrIndex() == Other.getInstrRefInstrIndex() &&
373            getInstrRefOpIndex() == Other.getInstrRefOpIndex();
374   case MachineOperand::MO_CFIIndex:
375     return getCFIIndex() == Other.getCFIIndex();
376   case MachineOperand::MO_Metadata:
377     return getMetadata() == Other.getMetadata();
378   case MachineOperand::MO_IntrinsicID:
379     return getIntrinsicID() == Other.getIntrinsicID();
380   case MachineOperand::MO_Predicate:
381     return getPredicate() == Other.getPredicate();
382   case MachineOperand::MO_ShuffleMask:
383     return getShuffleMask() == Other.getShuffleMask();
384   }
385   llvm_unreachable("Invalid machine operand type");
386 }
387 
388 // Note: this must stay exactly in sync with isIdenticalTo above.
hash_value(const MachineOperand & MO)389 hash_code llvm::hash_value(const MachineOperand &MO) {
390   switch (MO.getType()) {
391   case MachineOperand::MO_Register:
392     // Register operands don't have target flags.
393     return hash_combine(MO.getType(), (unsigned)MO.getReg(), MO.getSubReg(), MO.isDef());
394   case MachineOperand::MO_Immediate:
395     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getImm());
396   case MachineOperand::MO_CImmediate:
397     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getCImm());
398   case MachineOperand::MO_FPImmediate:
399     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getFPImm());
400   case MachineOperand::MO_MachineBasicBlock:
401     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getMBB());
402   case MachineOperand::MO_FrameIndex:
403     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getIndex());
404   case MachineOperand::MO_ConstantPoolIndex:
405   case MachineOperand::MO_TargetIndex:
406     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getIndex(),
407                         MO.getOffset());
408   case MachineOperand::MO_JumpTableIndex:
409     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getIndex());
410   case MachineOperand::MO_ExternalSymbol:
411     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getOffset(),
412                         StringRef(MO.getSymbolName()));
413   case MachineOperand::MO_GlobalAddress:
414     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getGlobal(),
415                         MO.getOffset());
416   case MachineOperand::MO_BlockAddress:
417     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getBlockAddress(),
418                         MO.getOffset());
419   case MachineOperand::MO_RegisterMask:
420   case MachineOperand::MO_RegisterLiveOut: {
421     if (const MachineFunction *MF = getMFIfAvailable(MO)) {
422       const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
423       unsigned RegMaskSize = MachineOperand::getRegMaskSize(TRI->getNumRegs());
424       const uint32_t *RegMask = MO.getRegMask();
425       std::vector<stable_hash> RegMaskHashes(RegMask, RegMask + RegMaskSize);
426       return hash_combine(MO.getType(), MO.getTargetFlags(),
427                           stable_hash_combine_array(RegMaskHashes.data(),
428                                                     RegMaskHashes.size()));
429     }
430 
431     assert(0 && "MachineOperand not associated with any MachineFunction");
432     return hash_combine(MO.getType(), MO.getTargetFlags());
433   }
434   case MachineOperand::MO_Metadata:
435     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getMetadata());
436   case MachineOperand::MO_MCSymbol:
437     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getMCSymbol());
438   case MachineOperand::MO_DbgInstrRef:
439     return hash_combine(MO.getType(), MO.getTargetFlags(),
440                         MO.getInstrRefInstrIndex(), MO.getInstrRefOpIndex());
441   case MachineOperand::MO_CFIIndex:
442     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getCFIIndex());
443   case MachineOperand::MO_IntrinsicID:
444     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getIntrinsicID());
445   case MachineOperand::MO_Predicate:
446     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getPredicate());
447   case MachineOperand::MO_ShuffleMask:
448     return hash_combine(MO.getType(), MO.getTargetFlags(), MO.getShuffleMask());
449   }
450   llvm_unreachable("Invalid machine operand type");
451 }
452 
453 // Try to crawl up to the machine function and get TRI and IntrinsicInfo from
454 // it.
tryToGetTargetInfo(const MachineOperand & MO,const TargetRegisterInfo * & TRI,const TargetIntrinsicInfo * & IntrinsicInfo)455 static void tryToGetTargetInfo(const MachineOperand &MO,
456                                const TargetRegisterInfo *&TRI,
457                                const TargetIntrinsicInfo *&IntrinsicInfo) {
458   if (const MachineFunction *MF = getMFIfAvailable(MO)) {
459     TRI = MF->getSubtarget().getRegisterInfo();
460     IntrinsicInfo = MF->getTarget().getIntrinsicInfo();
461   }
462 }
463 
getTargetIndexName(const MachineFunction & MF,int Index)464 static const char *getTargetIndexName(const MachineFunction &MF, int Index) {
465   const auto *TII = MF.getSubtarget().getInstrInfo();
466   assert(TII && "expected instruction info");
467   auto Indices = TII->getSerializableTargetIndices();
468   auto Found = find_if(Indices, [&](const std::pair<int, const char *> &I) {
469     return I.first == Index;
470   });
471   if (Found != Indices.end())
472     return Found->second;
473   return nullptr;
474 }
475 
getTargetIndexName() const476 const char *MachineOperand::getTargetIndexName() const {
477   const MachineFunction *MF = getMFIfAvailable(*this);
478   return MF ? ::getTargetIndexName(*MF, this->getIndex()) : nullptr;
479 }
480 
getTargetFlagName(const TargetInstrInfo * TII,unsigned TF)481 static const char *getTargetFlagName(const TargetInstrInfo *TII, unsigned TF) {
482   auto Flags = TII->getSerializableDirectMachineOperandTargetFlags();
483   for (const auto &I : Flags) {
484     if (I.first == TF) {
485       return I.second;
486     }
487   }
488   return nullptr;
489 }
490 
printCFIRegister(unsigned DwarfReg,raw_ostream & OS,const TargetRegisterInfo * TRI)491 static void printCFIRegister(unsigned DwarfReg, raw_ostream &OS,
492                              const TargetRegisterInfo *TRI) {
493   if (!TRI) {
494     OS << "%dwarfreg." << DwarfReg;
495     return;
496   }
497 
498   if (std::optional<unsigned> Reg = TRI->getLLVMRegNum(DwarfReg, true))
499     OS << printReg(*Reg, TRI);
500   else
501     OS << "<badreg>";
502 }
503 
printIRBlockReference(raw_ostream & OS,const BasicBlock & BB,ModuleSlotTracker & MST)504 static void printIRBlockReference(raw_ostream &OS, const BasicBlock &BB,
505                                   ModuleSlotTracker &MST) {
506   OS << "%ir-block.";
507   if (BB.hasName()) {
508     printLLVMNameWithoutPrefix(OS, BB.getName());
509     return;
510   }
511   std::optional<int> Slot;
512   if (const Function *F = BB.getParent()) {
513     if (F == MST.getCurrentFunction()) {
514       Slot = MST.getLocalSlot(&BB);
515     } else if (const Module *M = F->getParent()) {
516       ModuleSlotTracker CustomMST(M, /*ShouldInitializeAllMetadata=*/false);
517       CustomMST.incorporateFunction(*F);
518       Slot = CustomMST.getLocalSlot(&BB);
519     }
520   }
521   if (Slot)
522     MachineOperand::printIRSlotNumber(OS, *Slot);
523   else
524     OS << "<unknown>";
525 }
526 
printSyncScope(raw_ostream & OS,const LLVMContext & Context,SyncScope::ID SSID,SmallVectorImpl<StringRef> & SSNs)527 static void printSyncScope(raw_ostream &OS, const LLVMContext &Context,
528                            SyncScope::ID SSID,
529                            SmallVectorImpl<StringRef> &SSNs) {
530   switch (SSID) {
531   case SyncScope::System:
532     break;
533   default:
534     if (SSNs.empty())
535       Context.getSyncScopeNames(SSNs);
536 
537     OS << "syncscope(\"";
538     printEscapedString(SSNs[SSID], OS);
539     OS << "\") ";
540     break;
541   }
542 }
543 
getTargetMMOFlagName(const TargetInstrInfo & TII,unsigned TMMOFlag)544 static const char *getTargetMMOFlagName(const TargetInstrInfo &TII,
545                                         unsigned TMMOFlag) {
546   auto Flags = TII.getSerializableMachineMemOperandTargetFlags();
547   for (const auto &I : Flags) {
548     if (I.first == TMMOFlag) {
549       return I.second;
550     }
551   }
552   return nullptr;
553 }
554 
printFrameIndex(raw_ostream & OS,int FrameIndex,bool IsFixed,const MachineFrameInfo * MFI)555 static void printFrameIndex(raw_ostream& OS, int FrameIndex, bool IsFixed,
556                             const MachineFrameInfo *MFI) {
557   StringRef Name;
558   if (MFI) {
559     IsFixed = MFI->isFixedObjectIndex(FrameIndex);
560     if (const AllocaInst *Alloca = MFI->getObjectAllocation(FrameIndex))
561       if (Alloca->hasName())
562         Name = Alloca->getName();
563     if (IsFixed)
564       FrameIndex -= MFI->getObjectIndexBegin();
565   }
566   MachineOperand::printStackObjectReference(OS, FrameIndex, IsFixed, Name);
567 }
568 
printSubRegIdx(raw_ostream & OS,uint64_t Index,const TargetRegisterInfo * TRI)569 void MachineOperand::printSubRegIdx(raw_ostream &OS, uint64_t Index,
570                                     const TargetRegisterInfo *TRI) {
571   OS << "%subreg.";
572   if (TRI && Index != 0 && Index < TRI->getNumSubRegIndices())
573     OS << TRI->getSubRegIndexName(Index);
574   else
575     OS << Index;
576 }
577 
printTargetFlags(raw_ostream & OS,const MachineOperand & Op)578 void MachineOperand::printTargetFlags(raw_ostream &OS,
579                                       const MachineOperand &Op) {
580   if (!Op.getTargetFlags())
581     return;
582   const MachineFunction *MF = getMFIfAvailable(Op);
583   if (!MF)
584     return;
585 
586   const auto *TII = MF->getSubtarget().getInstrInfo();
587   assert(TII && "expected instruction info");
588   auto Flags = TII->decomposeMachineOperandsTargetFlags(Op.getTargetFlags());
589   OS << "target-flags(";
590   const bool HasDirectFlags = Flags.first;
591   const bool HasBitmaskFlags = Flags.second;
592   if (!HasDirectFlags && !HasBitmaskFlags) {
593     OS << "<unknown>) ";
594     return;
595   }
596   if (HasDirectFlags) {
597     if (const auto *Name = getTargetFlagName(TII, Flags.first))
598       OS << Name;
599     else
600       OS << "<unknown target flag>";
601   }
602   if (!HasBitmaskFlags) {
603     OS << ") ";
604     return;
605   }
606   bool IsCommaNeeded = HasDirectFlags;
607   unsigned BitMask = Flags.second;
608   auto BitMasks = TII->getSerializableBitmaskMachineOperandTargetFlags();
609   for (const auto &Mask : BitMasks) {
610     // Check if the flag's bitmask has the bits of the current mask set.
611     if ((BitMask & Mask.first) == Mask.first) {
612       if (IsCommaNeeded)
613         OS << ", ";
614       IsCommaNeeded = true;
615       OS << Mask.second;
616       // Clear the bits which were serialized from the flag's bitmask.
617       BitMask &= ~(Mask.first);
618     }
619   }
620   if (BitMask) {
621     // When the resulting flag's bitmask isn't zero, we know that we didn't
622     // serialize all of the bit flags.
623     if (IsCommaNeeded)
624       OS << ", ";
625     OS << "<unknown bitmask target flag>";
626   }
627   OS << ") ";
628 }
629 
printSymbol(raw_ostream & OS,MCSymbol & Sym)630 void MachineOperand::printSymbol(raw_ostream &OS, MCSymbol &Sym) {
631   OS << "<mcsymbol " << Sym << ">";
632 }
633 
printStackObjectReference(raw_ostream & OS,unsigned FrameIndex,bool IsFixed,StringRef Name)634 void MachineOperand::printStackObjectReference(raw_ostream &OS,
635                                                unsigned FrameIndex,
636                                                bool IsFixed, StringRef Name) {
637   if (IsFixed) {
638     OS << "%fixed-stack." << FrameIndex;
639     return;
640   }
641 
642   OS << "%stack." << FrameIndex;
643   if (!Name.empty())
644     OS << '.' << Name;
645 }
646 
printOperandOffset(raw_ostream & OS,int64_t Offset)647 void MachineOperand::printOperandOffset(raw_ostream &OS, int64_t Offset) {
648   if (Offset == 0)
649     return;
650   if (Offset < 0) {
651     OS << " - " << -Offset;
652     return;
653   }
654   OS << " + " << Offset;
655 }
656 
printIRSlotNumber(raw_ostream & OS,int Slot)657 void MachineOperand::printIRSlotNumber(raw_ostream &OS, int Slot) {
658   if (Slot == -1)
659     OS << "<badref>";
660   else
661     OS << Slot;
662 }
663 
printCFI(raw_ostream & OS,const MCCFIInstruction & CFI,const TargetRegisterInfo * TRI)664 static void printCFI(raw_ostream &OS, const MCCFIInstruction &CFI,
665                      const TargetRegisterInfo *TRI) {
666   switch (CFI.getOperation()) {
667   case MCCFIInstruction::OpSameValue:
668     OS << "same_value ";
669     if (MCSymbol *Label = CFI.getLabel())
670       MachineOperand::printSymbol(OS, *Label);
671     printCFIRegister(CFI.getRegister(), OS, TRI);
672     break;
673   case MCCFIInstruction::OpRememberState:
674     OS << "remember_state ";
675     if (MCSymbol *Label = CFI.getLabel())
676       MachineOperand::printSymbol(OS, *Label);
677     break;
678   case MCCFIInstruction::OpRestoreState:
679     OS << "restore_state ";
680     if (MCSymbol *Label = CFI.getLabel())
681       MachineOperand::printSymbol(OS, *Label);
682     break;
683   case MCCFIInstruction::OpOffset:
684     OS << "offset ";
685     if (MCSymbol *Label = CFI.getLabel())
686       MachineOperand::printSymbol(OS, *Label);
687     printCFIRegister(CFI.getRegister(), OS, TRI);
688     OS << ", " << CFI.getOffset();
689     break;
690   case MCCFIInstruction::OpDefCfaRegister:
691     OS << "def_cfa_register ";
692     if (MCSymbol *Label = CFI.getLabel())
693       MachineOperand::printSymbol(OS, *Label);
694     printCFIRegister(CFI.getRegister(), OS, TRI);
695     break;
696   case MCCFIInstruction::OpDefCfaOffset:
697     OS << "def_cfa_offset ";
698     if (MCSymbol *Label = CFI.getLabel())
699       MachineOperand::printSymbol(OS, *Label);
700     OS << CFI.getOffset();
701     break;
702   case MCCFIInstruction::OpDefCfa:
703     OS << "def_cfa ";
704     if (MCSymbol *Label = CFI.getLabel())
705       MachineOperand::printSymbol(OS, *Label);
706     printCFIRegister(CFI.getRegister(), OS, TRI);
707     OS << ", " << CFI.getOffset();
708     break;
709   case MCCFIInstruction::OpLLVMDefAspaceCfa:
710     OS << "llvm_def_aspace_cfa ";
711     if (MCSymbol *Label = CFI.getLabel())
712       MachineOperand::printSymbol(OS, *Label);
713     printCFIRegister(CFI.getRegister(), OS, TRI);
714     OS << ", " << CFI.getOffset();
715     OS << ", " << CFI.getAddressSpace();
716     break;
717   case MCCFIInstruction::OpRelOffset:
718     OS << "rel_offset ";
719     if (MCSymbol *Label = CFI.getLabel())
720       MachineOperand::printSymbol(OS, *Label);
721     printCFIRegister(CFI.getRegister(), OS, TRI);
722     OS << ", " << CFI.getOffset();
723     break;
724   case MCCFIInstruction::OpAdjustCfaOffset:
725     OS << "adjust_cfa_offset ";
726     if (MCSymbol *Label = CFI.getLabel())
727       MachineOperand::printSymbol(OS, *Label);
728     OS << CFI.getOffset();
729     break;
730   case MCCFIInstruction::OpRestore:
731     OS << "restore ";
732     if (MCSymbol *Label = CFI.getLabel())
733       MachineOperand::printSymbol(OS, *Label);
734     printCFIRegister(CFI.getRegister(), OS, TRI);
735     break;
736   case MCCFIInstruction::OpEscape: {
737     OS << "escape ";
738     if (MCSymbol *Label = CFI.getLabel())
739       MachineOperand::printSymbol(OS, *Label);
740     if (!CFI.getValues().empty()) {
741       size_t e = CFI.getValues().size() - 1;
742       for (size_t i = 0; i < e; ++i)
743         OS << format("0x%02x", uint8_t(CFI.getValues()[i])) << ", ";
744       OS << format("0x%02x", uint8_t(CFI.getValues()[e]));
745     }
746     break;
747   }
748   case MCCFIInstruction::OpUndefined:
749     OS << "undefined ";
750     if (MCSymbol *Label = CFI.getLabel())
751       MachineOperand::printSymbol(OS, *Label);
752     printCFIRegister(CFI.getRegister(), OS, TRI);
753     break;
754   case MCCFIInstruction::OpRegister:
755     OS << "register ";
756     if (MCSymbol *Label = CFI.getLabel())
757       MachineOperand::printSymbol(OS, *Label);
758     printCFIRegister(CFI.getRegister(), OS, TRI);
759     OS << ", ";
760     printCFIRegister(CFI.getRegister2(), OS, TRI);
761     break;
762   case MCCFIInstruction::OpWindowSave:
763     OS << "window_save ";
764     if (MCSymbol *Label = CFI.getLabel())
765       MachineOperand::printSymbol(OS, *Label);
766     break;
767   case MCCFIInstruction::OpNegateRAState:
768     OS << "negate_ra_sign_state ";
769     if (MCSymbol *Label = CFI.getLabel())
770       MachineOperand::printSymbol(OS, *Label);
771     break;
772   default:
773     // TODO: Print the other CFI Operations.
774     OS << "<unserializable cfi directive>";
775     break;
776   }
777 }
778 
print(raw_ostream & OS,const TargetRegisterInfo * TRI,const TargetIntrinsicInfo * IntrinsicInfo) const779 void MachineOperand::print(raw_ostream &OS, const TargetRegisterInfo *TRI,
780                            const TargetIntrinsicInfo *IntrinsicInfo) const {
781   print(OS, LLT{}, TRI, IntrinsicInfo);
782 }
783 
print(raw_ostream & OS,LLT TypeToPrint,const TargetRegisterInfo * TRI,const TargetIntrinsicInfo * IntrinsicInfo) const784 void MachineOperand::print(raw_ostream &OS, LLT TypeToPrint,
785                            const TargetRegisterInfo *TRI,
786                            const TargetIntrinsicInfo *IntrinsicInfo) const {
787   tryToGetTargetInfo(*this, TRI, IntrinsicInfo);
788   ModuleSlotTracker DummyMST(nullptr);
789   print(OS, DummyMST, TypeToPrint, std::nullopt, /*PrintDef=*/false,
790         /*IsStandalone=*/true,
791         /*ShouldPrintRegisterTies=*/true,
792         /*TiedOperandIdx=*/0, TRI, IntrinsicInfo);
793 }
794 
print(raw_ostream & OS,ModuleSlotTracker & MST,LLT TypeToPrint,std::optional<unsigned> OpIdx,bool PrintDef,bool IsStandalone,bool ShouldPrintRegisterTies,unsigned TiedOperandIdx,const TargetRegisterInfo * TRI,const TargetIntrinsicInfo * IntrinsicInfo) const795 void MachineOperand::print(raw_ostream &OS, ModuleSlotTracker &MST,
796                            LLT TypeToPrint, std::optional<unsigned> OpIdx,
797                            bool PrintDef, bool IsStandalone,
798                            bool ShouldPrintRegisterTies,
799                            unsigned TiedOperandIdx,
800                            const TargetRegisterInfo *TRI,
801                            const TargetIntrinsicInfo *IntrinsicInfo) const {
802   printTargetFlags(OS, *this);
803   switch (getType()) {
804   case MachineOperand::MO_Register: {
805     Register Reg = getReg();
806     if (isImplicit())
807       OS << (isDef() ? "implicit-def " : "implicit ");
808     else if (PrintDef && isDef())
809       // Print the 'def' flag only when the operand is defined after '='.
810       OS << "def ";
811     if (isInternalRead())
812       OS << "internal ";
813     if (isDead())
814       OS << "dead ";
815     if (isKill())
816       OS << "killed ";
817     if (isUndef())
818       OS << "undef ";
819     if (isEarlyClobber())
820       OS << "early-clobber ";
821     if (getReg().isPhysical() && isRenamable())
822       OS << "renamable ";
823     // isDebug() is exactly true for register operands of a DBG_VALUE. So we
824     // simply infer it when parsing and do not need to print it.
825 
826     const MachineRegisterInfo *MRI = nullptr;
827     if (Reg.isVirtual()) {
828       if (const MachineFunction *MF = getMFIfAvailable(*this)) {
829         MRI = &MF->getRegInfo();
830       }
831     }
832 
833     OS << printReg(Reg, TRI, 0, MRI);
834     // Print the sub register.
835     if (unsigned SubReg = getSubReg()) {
836       if (TRI)
837         OS << '.' << TRI->getSubRegIndexName(SubReg);
838       else
839         OS << ".subreg" << SubReg;
840     }
841     // Print the register class / bank.
842     if (Reg.isVirtual()) {
843       if (const MachineFunction *MF = getMFIfAvailable(*this)) {
844         const MachineRegisterInfo &MRI = MF->getRegInfo();
845         if (IsStandalone || !PrintDef || MRI.def_empty(Reg)) {
846           OS << ':';
847           OS << printRegClassOrBank(Reg, MRI, TRI);
848         }
849       }
850     }
851     // Print ties.
852     if (ShouldPrintRegisterTies && isTied() && !isDef())
853       OS << "(tied-def " << TiedOperandIdx << ")";
854     // Print types.
855     if (TypeToPrint.isValid())
856       OS << '(' << TypeToPrint << ')';
857     break;
858   }
859   case MachineOperand::MO_Immediate: {
860     const MIRFormatter *Formatter = nullptr;
861     if (const MachineFunction *MF = getMFIfAvailable(*this)) {
862       const auto *TII = MF->getSubtarget().getInstrInfo();
863       assert(TII && "expected instruction info");
864       Formatter = TII->getMIRFormatter();
865     }
866     if (Formatter)
867       Formatter->printImm(OS, *getParent(), OpIdx, getImm());
868     else
869       OS << getImm();
870     break;
871   }
872   case MachineOperand::MO_CImmediate:
873     getCImm()->printAsOperand(OS, /*PrintType=*/true, MST);
874     break;
875   case MachineOperand::MO_FPImmediate:
876     getFPImm()->printAsOperand(OS, /*PrintType=*/true, MST);
877     break;
878   case MachineOperand::MO_MachineBasicBlock:
879     OS << printMBBReference(*getMBB());
880     break;
881   case MachineOperand::MO_FrameIndex: {
882     int FrameIndex = getIndex();
883     bool IsFixed = false;
884     const MachineFrameInfo *MFI = nullptr;
885     if (const MachineFunction *MF = getMFIfAvailable(*this))
886       MFI = &MF->getFrameInfo();
887     printFrameIndex(OS, FrameIndex, IsFixed, MFI);
888     break;
889   }
890   case MachineOperand::MO_ConstantPoolIndex:
891     OS << "%const." << getIndex();
892     printOperandOffset(OS, getOffset());
893     break;
894   case MachineOperand::MO_TargetIndex: {
895     OS << "target-index(";
896     const char *Name = "<unknown>";
897     if (const MachineFunction *MF = getMFIfAvailable(*this))
898       if (const auto *TargetIndexName = ::getTargetIndexName(*MF, getIndex()))
899         Name = TargetIndexName;
900     OS << Name << ')';
901     printOperandOffset(OS, getOffset());
902     break;
903   }
904   case MachineOperand::MO_JumpTableIndex:
905     OS << printJumpTableEntryReference(getIndex());
906     break;
907   case MachineOperand::MO_GlobalAddress:
908     getGlobal()->printAsOperand(OS, /*PrintType=*/false, MST);
909     printOperandOffset(OS, getOffset());
910     break;
911   case MachineOperand::MO_ExternalSymbol: {
912     StringRef Name = getSymbolName();
913     OS << '&';
914     if (Name.empty()) {
915       OS << "\"\"";
916     } else {
917       printLLVMNameWithoutPrefix(OS, Name);
918     }
919     printOperandOffset(OS, getOffset());
920     break;
921   }
922   case MachineOperand::MO_BlockAddress: {
923     OS << "blockaddress(";
924     getBlockAddress()->getFunction()->printAsOperand(OS, /*PrintType=*/false,
925                                                      MST);
926     OS << ", ";
927     printIRBlockReference(OS, *getBlockAddress()->getBasicBlock(), MST);
928     OS << ')';
929     MachineOperand::printOperandOffset(OS, getOffset());
930     break;
931   }
932   case MachineOperand::MO_RegisterMask: {
933     OS << "<regmask";
934     if (TRI) {
935       unsigned NumRegsInMask = 0;
936       unsigned NumRegsEmitted = 0;
937       for (unsigned i = 0; i < TRI->getNumRegs(); ++i) {
938         unsigned MaskWord = i / 32;
939         unsigned MaskBit = i % 32;
940         if (getRegMask()[MaskWord] & (1 << MaskBit)) {
941           if (PrintRegMaskNumRegs < 0 ||
942               NumRegsEmitted <= static_cast<unsigned>(PrintRegMaskNumRegs)) {
943             OS << " " << printReg(i, TRI);
944             NumRegsEmitted++;
945           }
946           NumRegsInMask++;
947         }
948       }
949       if (NumRegsEmitted != NumRegsInMask)
950         OS << " and " << (NumRegsInMask - NumRegsEmitted) << " more...";
951     } else {
952       OS << " ...";
953     }
954     OS << ">";
955     break;
956   }
957   case MachineOperand::MO_RegisterLiveOut: {
958     const uint32_t *RegMask = getRegLiveOut();
959     OS << "liveout(";
960     if (!TRI) {
961       OS << "<unknown>";
962     } else {
963       bool IsCommaNeeded = false;
964       for (unsigned Reg = 0, E = TRI->getNumRegs(); Reg < E; ++Reg) {
965         if (RegMask[Reg / 32] & (1U << (Reg % 32))) {
966           if (IsCommaNeeded)
967             OS << ", ";
968           OS << printReg(Reg, TRI);
969           IsCommaNeeded = true;
970         }
971       }
972     }
973     OS << ")";
974     break;
975   }
976   case MachineOperand::MO_Metadata:
977     getMetadata()->printAsOperand(OS, MST);
978     break;
979   case MachineOperand::MO_MCSymbol:
980     printSymbol(OS, *getMCSymbol());
981     break;
982   case MachineOperand::MO_DbgInstrRef: {
983     OS << "dbg-instr-ref(" << getInstrRefInstrIndex() << ", "
984        << getInstrRefOpIndex() << ')';
985     break;
986   }
987   case MachineOperand::MO_CFIIndex: {
988     if (const MachineFunction *MF = getMFIfAvailable(*this))
989       printCFI(OS, MF->getFrameInstructions()[getCFIIndex()], TRI);
990     else
991       OS << "<cfi directive>";
992     break;
993   }
994   case MachineOperand::MO_IntrinsicID: {
995     Intrinsic::ID ID = getIntrinsicID();
996     if (ID < Intrinsic::num_intrinsics)
997       OS << "intrinsic(@" << Intrinsic::getBaseName(ID) << ')';
998     else if (IntrinsicInfo)
999       OS << "intrinsic(@" << IntrinsicInfo->getName(ID) << ')';
1000     else
1001       OS << "intrinsic(" << ID << ')';
1002     break;
1003   }
1004   case MachineOperand::MO_Predicate: {
1005     auto Pred = static_cast<CmpInst::Predicate>(getPredicate());
1006     OS << (CmpInst::isIntPredicate(Pred) ? "int" : "float") << "pred("
1007        << Pred << ')';
1008     break;
1009   }
1010   case MachineOperand::MO_ShuffleMask:
1011     OS << "shufflemask(";
1012     ArrayRef<int> Mask = getShuffleMask();
1013     StringRef Separator;
1014     for (int Elt : Mask) {
1015       if (Elt == -1)
1016         OS << Separator << "undef";
1017       else
1018         OS << Separator << Elt;
1019       Separator = ", ";
1020     }
1021 
1022     OS << ')';
1023     break;
1024   }
1025 }
1026 
1027 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
dump() const1028 LLVM_DUMP_METHOD void MachineOperand::dump() const { dbgs() << *this << '\n'; }
1029 #endif
1030 
1031 //===----------------------------------------------------------------------===//
1032 // MachineMemOperand Implementation
1033 //===----------------------------------------------------------------------===//
1034 
1035 /// getAddrSpace - Return the LLVM IR address space number that this pointer
1036 /// points into.
getAddrSpace() const1037 unsigned MachinePointerInfo::getAddrSpace() const { return AddrSpace; }
1038 
1039 /// isDereferenceable - Return true if V is always dereferenceable for
1040 /// Offset + Size byte.
isDereferenceable(unsigned Size,LLVMContext & C,const DataLayout & DL) const1041 bool MachinePointerInfo::isDereferenceable(unsigned Size, LLVMContext &C,
1042                                            const DataLayout &DL) const {
1043   if (!isa<const Value *>(V))
1044     return false;
1045 
1046   const Value *BasePtr = cast<const Value *>(V);
1047   if (BasePtr == nullptr)
1048     return false;
1049 
1050   return isDereferenceableAndAlignedPointer(
1051       BasePtr, Align(1), APInt(DL.getPointerSizeInBits(), Offset + Size), DL);
1052 }
1053 
1054 /// getConstantPool - Return a MachinePointerInfo record that refers to the
1055 /// constant pool.
getConstantPool(MachineFunction & MF)1056 MachinePointerInfo MachinePointerInfo::getConstantPool(MachineFunction &MF) {
1057   return MachinePointerInfo(MF.getPSVManager().getConstantPool());
1058 }
1059 
1060 /// getFixedStack - Return a MachinePointerInfo record that refers to the
1061 /// the specified FrameIndex.
getFixedStack(MachineFunction & MF,int FI,int64_t Offset)1062 MachinePointerInfo MachinePointerInfo::getFixedStack(MachineFunction &MF,
1063                                                      int FI, int64_t Offset) {
1064   return MachinePointerInfo(MF.getPSVManager().getFixedStack(FI), Offset);
1065 }
1066 
getJumpTable(MachineFunction & MF)1067 MachinePointerInfo MachinePointerInfo::getJumpTable(MachineFunction &MF) {
1068   return MachinePointerInfo(MF.getPSVManager().getJumpTable());
1069 }
1070 
getGOT(MachineFunction & MF)1071 MachinePointerInfo MachinePointerInfo::getGOT(MachineFunction &MF) {
1072   return MachinePointerInfo(MF.getPSVManager().getGOT());
1073 }
1074 
getStack(MachineFunction & MF,int64_t Offset,uint8_t ID)1075 MachinePointerInfo MachinePointerInfo::getStack(MachineFunction &MF,
1076                                                 int64_t Offset, uint8_t ID) {
1077   return MachinePointerInfo(MF.getPSVManager().getStack(), Offset, ID);
1078 }
1079 
getUnknownStack(MachineFunction & MF)1080 MachinePointerInfo MachinePointerInfo::getUnknownStack(MachineFunction &MF) {
1081   return MachinePointerInfo(MF.getDataLayout().getAllocaAddrSpace());
1082 }
1083 
MachineMemOperand(MachinePointerInfo ptrinfo,Flags f,LLT type,Align a,const AAMDNodes & AAInfo,const MDNode * Ranges,SyncScope::ID SSID,AtomicOrdering Ordering,AtomicOrdering FailureOrdering)1084 MachineMemOperand::MachineMemOperand(MachinePointerInfo ptrinfo, Flags f,
1085                                      LLT type, Align a, const AAMDNodes &AAInfo,
1086                                      const MDNode *Ranges, SyncScope::ID SSID,
1087                                      AtomicOrdering Ordering,
1088                                      AtomicOrdering FailureOrdering)
1089     : PtrInfo(ptrinfo), MemoryType(type), FlagVals(f), BaseAlign(a),
1090       AAInfo(AAInfo), Ranges(Ranges) {
1091   assert((PtrInfo.V.isNull() || isa<const PseudoSourceValue *>(PtrInfo.V) ||
1092           isa<PointerType>(cast<const Value *>(PtrInfo.V)->getType())) &&
1093          "invalid pointer value");
1094   assert((isLoad() || isStore()) && "Not a load/store!");
1095 
1096   AtomicInfo.SSID = static_cast<unsigned>(SSID);
1097   assert(getSyncScopeID() == SSID && "Value truncated");
1098   AtomicInfo.Ordering = static_cast<unsigned>(Ordering);
1099   assert(getSuccessOrdering() == Ordering && "Value truncated");
1100   AtomicInfo.FailureOrdering = static_cast<unsigned>(FailureOrdering);
1101   assert(getFailureOrdering() == FailureOrdering && "Value truncated");
1102 }
1103 
MachineMemOperand(MachinePointerInfo ptrinfo,Flags F,LocationSize TS,Align BaseAlignment,const AAMDNodes & AAInfo,const MDNode * Ranges,SyncScope::ID SSID,AtomicOrdering Ordering,AtomicOrdering FailureOrdering)1104 MachineMemOperand::MachineMemOperand(MachinePointerInfo ptrinfo, Flags F,
1105                                      LocationSize TS, Align BaseAlignment,
1106                                      const AAMDNodes &AAInfo,
1107                                      const MDNode *Ranges, SyncScope::ID SSID,
1108                                      AtomicOrdering Ordering,
1109                                      AtomicOrdering FailureOrdering)
1110     : MachineMemOperand(
1111           ptrinfo, F,
1112           !TS.hasValue() ? LLT()
1113           : TS.isScalable()
1114               ? LLT::scalable_vector(1, 8 * TS.getValue().getKnownMinValue())
1115               : LLT::scalar(8 * TS.getValue().getKnownMinValue()),
1116           BaseAlignment, AAInfo, Ranges, SSID, Ordering, FailureOrdering) {}
1117 
refineAlignment(const MachineMemOperand * MMO)1118 void MachineMemOperand::refineAlignment(const MachineMemOperand *MMO) {
1119   // The Value and Offset may differ due to CSE. But the flags and size
1120   // should be the same.
1121   assert(MMO->getFlags() == getFlags() && "Flags mismatch!");
1122   assert((!MMO->getSize().hasValue() || !getSize().hasValue() ||
1123           MMO->getSize() == getSize()) &&
1124          "Size mismatch!");
1125   if (MMO->getBaseAlign() >= getBaseAlign()) {
1126     // Update the alignment value.
1127     BaseAlign = MMO->getBaseAlign();
1128     // Also update the base and offset, because the new alignment may
1129     // not be applicable with the old ones.
1130     PtrInfo = MMO->PtrInfo;
1131   }
1132 }
1133 
1134 /// getAlign - Return the minimum known alignment in bytes of the
1135 /// actual memory reference.
getAlign() const1136 Align MachineMemOperand::getAlign() const {
1137   return commonAlignment(getBaseAlign(), getOffset());
1138 }
1139 
print(raw_ostream & OS,ModuleSlotTracker & MST,SmallVectorImpl<StringRef> & SSNs,const LLVMContext & Context,const MachineFrameInfo * MFI,const TargetInstrInfo * TII) const1140 void MachineMemOperand::print(raw_ostream &OS, ModuleSlotTracker &MST,
1141                               SmallVectorImpl<StringRef> &SSNs,
1142                               const LLVMContext &Context,
1143                               const MachineFrameInfo *MFI,
1144                               const TargetInstrInfo *TII) const {
1145   OS << '(';
1146   if (isVolatile())
1147     OS << "volatile ";
1148   if (isNonTemporal())
1149     OS << "non-temporal ";
1150   if (isDereferenceable())
1151     OS << "dereferenceable ";
1152   if (isInvariant())
1153     OS << "invariant ";
1154   if (TII) {
1155     if (getFlags() & MachineMemOperand::MOTargetFlag1)
1156       OS << '"' << getTargetMMOFlagName(*TII, MachineMemOperand::MOTargetFlag1)
1157          << "\" ";
1158     if (getFlags() & MachineMemOperand::MOTargetFlag2)
1159       OS << '"' << getTargetMMOFlagName(*TII, MachineMemOperand::MOTargetFlag2)
1160          << "\" ";
1161     if (getFlags() & MachineMemOperand::MOTargetFlag3)
1162       OS << '"' << getTargetMMOFlagName(*TII, MachineMemOperand::MOTargetFlag3)
1163          << "\" ";
1164   } else {
1165     if (getFlags() & MachineMemOperand::MOTargetFlag1)
1166       OS << "\"MOTargetFlag1\" ";
1167     if (getFlags() & MachineMemOperand::MOTargetFlag2)
1168       OS << "\"MOTargetFlag2\" ";
1169     if (getFlags() & MachineMemOperand::MOTargetFlag3)
1170       OS << "\"MOTargetFlag3\" ";
1171   }
1172 
1173   assert((isLoad() || isStore()) &&
1174          "machine memory operand must be a load or store (or both)");
1175   if (isLoad())
1176     OS << "load ";
1177   if (isStore())
1178     OS << "store ";
1179 
1180   printSyncScope(OS, Context, getSyncScopeID(), SSNs);
1181 
1182   if (getSuccessOrdering() != AtomicOrdering::NotAtomic)
1183     OS << toIRString(getSuccessOrdering()) << ' ';
1184   if (getFailureOrdering() != AtomicOrdering::NotAtomic)
1185     OS << toIRString(getFailureOrdering()) << ' ';
1186 
1187   if (getMemoryType().isValid())
1188     OS << '(' << getMemoryType() << ')';
1189   else
1190     OS << "unknown-size";
1191 
1192   if (const Value *Val = getValue()) {
1193     OS << ((isLoad() && isStore()) ? " on " : isLoad() ? " from " : " into ");
1194     MIRFormatter::printIRValue(OS, *Val, MST);
1195   } else if (const PseudoSourceValue *PVal = getPseudoValue()) {
1196     OS << ((isLoad() && isStore()) ? " on " : isLoad() ? " from " : " into ");
1197     assert(PVal && "Expected a pseudo source value");
1198     switch (PVal->kind()) {
1199     case PseudoSourceValue::Stack:
1200       OS << "stack";
1201       break;
1202     case PseudoSourceValue::GOT:
1203       OS << "got";
1204       break;
1205     case PseudoSourceValue::JumpTable:
1206       OS << "jump-table";
1207       break;
1208     case PseudoSourceValue::ConstantPool:
1209       OS << "constant-pool";
1210       break;
1211     case PseudoSourceValue::FixedStack: {
1212       int FrameIndex = cast<FixedStackPseudoSourceValue>(PVal)->getFrameIndex();
1213       bool IsFixed = true;
1214       printFrameIndex(OS, FrameIndex, IsFixed, MFI);
1215       break;
1216     }
1217     case PseudoSourceValue::GlobalValueCallEntry:
1218       OS << "call-entry ";
1219       cast<GlobalValuePseudoSourceValue>(PVal)->getValue()->printAsOperand(
1220           OS, /*PrintType=*/false, MST);
1221       break;
1222     case PseudoSourceValue::ExternalSymbolCallEntry:
1223       OS << "call-entry &";
1224       printLLVMNameWithoutPrefix(
1225           OS, cast<ExternalSymbolPseudoSourceValue>(PVal)->getSymbol());
1226       break;
1227     default: {
1228       const MIRFormatter *Formatter = TII->getMIRFormatter();
1229       // FIXME: This is not necessarily the correct MIR serialization format for
1230       // a custom pseudo source value, but at least it allows
1231       // MIR printing to work on a target with custom pseudo source
1232       // values.
1233       OS << "custom \"";
1234       Formatter->printCustomPseudoSourceValue(OS, MST, *PVal);
1235       OS << '\"';
1236       break;
1237     }
1238     }
1239   } else if (getOpaqueValue() == nullptr && getOffset() != 0) {
1240     OS << ((isLoad() && isStore()) ? " on "
1241            : isLoad()              ? " from "
1242                                    : " into ")
1243        << "unknown-address";
1244   }
1245   MachineOperand::printOperandOffset(OS, getOffset());
1246   if (!getSize().hasValue() ||
1247       getAlign() != getSize().getValue().getKnownMinValue())
1248     OS << ", align " << getAlign().value();
1249   if (getAlign() != getBaseAlign())
1250     OS << ", basealign " << getBaseAlign().value();
1251   auto AAInfo = getAAInfo();
1252   if (AAInfo.TBAA) {
1253     OS << ", !tbaa ";
1254     AAInfo.TBAA->printAsOperand(OS, MST);
1255   }
1256   if (AAInfo.Scope) {
1257     OS << ", !alias.scope ";
1258     AAInfo.Scope->printAsOperand(OS, MST);
1259   }
1260   if (AAInfo.NoAlias) {
1261     OS << ", !noalias ";
1262     AAInfo.NoAlias->printAsOperand(OS, MST);
1263   }
1264   if (getRanges()) {
1265     OS << ", !range ";
1266     getRanges()->printAsOperand(OS, MST);
1267   }
1268   // FIXME: Implement addrspace printing/parsing in MIR.
1269   // For now, print this even though parsing it is not available in MIR.
1270   if (unsigned AS = getAddrSpace())
1271     OS << ", addrspace " << AS;
1272 
1273   OS << ')';
1274 }
1275