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