LLVM 24.0.0git
RISCVInstrInfo.cpp
Go to the documentation of this file.
1//===-- RISCVInstrInfo.cpp - RISC-V Instruction Information -----*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file contains the RISC-V implementation of the TargetInstrInfo class.
10//
11//===----------------------------------------------------------------------===//
12
13#include "RISCVInstrInfo.h"
16#include "RISCV.h"
18#include "RISCVSubtarget.h"
19#include "llvm/ADT/STLExtras.h"
21#include "llvm/ADT/Statistic.h"
33#include "llvm/IR/Module.h"
34#include "llvm/MC/MCDwarf.h"
38
39using namespace llvm;
40
41#define GEN_CHECK_COMPRESS_INSTR
42#include "RISCVGenCompressInstEmitter.inc"
43
44#define GET_INSTRINFO_CTOR_DTOR
45#include "RISCVGenInstrInfo.inc"
46
47#define DEBUG_TYPE "riscv-instr-info"
48STATISTIC(NumVRegSpilled,
49 "Number of registers within vector register groups spilled");
50STATISTIC(NumVRegReloaded,
51 "Number of registers within vector register groups reloaded");
52
54 "riscv-prefer-whole-register-move", cl::init(false), cl::Hidden,
55 cl::desc("Prefer whole register move for vector registers."));
56
58 "riscv-force-machine-combiner-strategy", cl::Hidden,
59 cl::desc("Force machine combiner to use a specific strategy for machine "
60 "trace metrics evaluation."),
63 "Local strategy."),
65 "MinInstrCount strategy.")));
66
68 "riscv-outliner-regsave", cl::init(true), cl::Hidden,
69 cl::desc("Enable RegSave strategy in machine outliner (save X5 to a "
70 "temporary register when X5 is live across outlined calls)."));
71
73
74using namespace RISCV;
75
76#define GET_RISCVVPseudosTable_IMPL
77#include "RISCVGenSearchableTables.inc"
78
79} // namespace llvm::RISCVVPseudosTable
80
81namespace llvm::RISCV {
82
83#define GET_RISCVMaskedPseudosTable_IMPL
84#include "RISCVGenSearchableTables.inc"
85
86} // end namespace llvm::RISCV
87
89 : RISCVGenInstrInfo(STI, RegInfo, RISCV::ADJCALLSTACKDOWN,
90 RISCV::ADJCALLSTACKUP),
91 RegInfo(STI.getHwMode()), STI(STI) {}
92
93#define GET_INSTRINFO_HELPERS
94#include "RISCVGenInstrInfo.inc"
95
97 if (STI.hasStdExtZca())
98 return MCInstBuilder(RISCV::C_NOP);
99 return MCInstBuilder(RISCV::ADDI)
100 .addReg(RISCV::X0)
101 .addReg(RISCV::X0)
102 .addImm(0);
103}
104
106 int &FrameIndex) const {
107 TypeSize Dummy = TypeSize::getZero();
108 return isLoadFromStackSlot(MI, FrameIndex, Dummy);
109}
110
111static std::optional<unsigned> getLMULForRVVWholeLoadStore(unsigned Opcode) {
112 switch (Opcode) {
113 default:
114 return std::nullopt;
115 case RISCV::VS1R_V:
116 case RISCV::VL1RE8_V:
117 case RISCV::VL1RE16_V:
118 case RISCV::VL1RE32_V:
119 case RISCV::VL1RE64_V:
120 return 1;
121 case RISCV::VS2R_V:
122 case RISCV::VL2RE8_V:
123 case RISCV::VL2RE16_V:
124 case RISCV::VL2RE32_V:
125 case RISCV::VL2RE64_V:
126 return 2;
127 case RISCV::VS4R_V:
128 case RISCV::VL4RE8_V:
129 case RISCV::VL4RE16_V:
130 case RISCV::VL4RE32_V:
131 case RISCV::VL4RE64_V:
132 return 4;
133 case RISCV::VS8R_V:
134 case RISCV::VL8RE8_V:
135 case RISCV::VL8RE16_V:
136 case RISCV::VL8RE32_V:
137 case RISCV::VL8RE64_V:
138 return 8;
139 }
140}
141
143 int &FrameIndex,
144 TypeSize &MemBytes) const {
145 switch (MI.getOpcode()) {
146 default:
147 return 0;
148 case RISCV::LB:
149 case RISCV::LBU:
150 MemBytes = TypeSize::getFixed(1);
151 break;
152 case RISCV::LH:
153 case RISCV::LH_INX:
154 case RISCV::LHU:
155 case RISCV::FLH:
156 MemBytes = TypeSize::getFixed(2);
157 break;
158 case RISCV::LW:
159 case RISCV::LW_INX:
160 case RISCV::FLW:
161 case RISCV::LWU:
162 MemBytes = TypeSize::getFixed(4);
163 break;
164 case RISCV::LD:
165 case RISCV::LD_RV32:
166 case RISCV::FLD:
167 MemBytes = TypeSize::getFixed(8);
168 break;
169 case RISCV::VL1RE8_V:
170 case RISCV::VL2RE8_V:
171 case RISCV::VL4RE8_V:
172 case RISCV::VL8RE8_V:
173 if (!MI.getOperand(1).isFI())
174 return Register();
175 FrameIndex = MI.getOperand(1).getIndex();
176 unsigned LMUL = *getLMULForRVVWholeLoadStore(MI.getOpcode());
178 return MI.getOperand(0).getReg();
179 }
180
181 if (MI.getOperand(1).isFI() && MI.getOperand(2).isImm() &&
182 MI.getOperand(2).getImm() == 0) {
183 FrameIndex = MI.getOperand(1).getIndex();
184 return MI.getOperand(0).getReg();
185 }
186
187 return 0;
188}
189
191 int &FrameIndex) const {
192 TypeSize Dummy = TypeSize::getZero();
193 return isStoreToStackSlot(MI, FrameIndex, Dummy);
194}
195
197 int &FrameIndex,
198 TypeSize &MemBytes) const {
199 switch (MI.getOpcode()) {
200 default:
201 return 0;
202 case RISCV::SB:
203 MemBytes = TypeSize::getFixed(1);
204 break;
205 case RISCV::SH:
206 case RISCV::SH_INX:
207 case RISCV::FSH:
208 MemBytes = TypeSize::getFixed(2);
209 break;
210 case RISCV::SW:
211 case RISCV::SW_INX:
212 case RISCV::FSW:
213 MemBytes = TypeSize::getFixed(4);
214 break;
215 case RISCV::SD:
216 case RISCV::SD_RV32:
217 case RISCV::FSD:
218 MemBytes = TypeSize::getFixed(8);
219 break;
220 case RISCV::VS1R_V:
221 case RISCV::VS2R_V:
222 case RISCV::VS4R_V:
223 case RISCV::VS8R_V:
224 if (!MI.getOperand(1).isFI())
225 return Register();
226 FrameIndex = MI.getOperand(1).getIndex();
227 unsigned LMUL = *getLMULForRVVWholeLoadStore(MI.getOpcode());
229 return MI.getOperand(0).getReg();
230 }
231
232 if (MI.getOperand(1).isFI() && MI.getOperand(2).isImm() &&
233 MI.getOperand(2).getImm() == 0) {
234 FrameIndex = MI.getOperand(1).getIndex();
235 return MI.getOperand(0).getReg();
236 }
237
238 return 0;
239}
240
242 const MachineInstr &MI) const {
243 switch (RISCV::getRVVMCOpcode(MI.getOpcode())) {
244 case RISCV::VMV_V_X:
245 case RISCV::VFMV_V_F:
246 case RISCV::VMV_V_I:
247 case RISCV::VMV_S_X:
248 case RISCV::VFMV_S_F:
249 case RISCV::VID_V:
250 return MI.getOperand(1).isUndef();
251 default:
253 }
254}
255
256static bool forwardCopyWillClobberTuple(unsigned DstReg, unsigned SrcReg,
257 unsigned NumRegs) {
258 return DstReg > SrcReg && (DstReg - SrcReg) < NumRegs;
259}
260
262 const MachineBasicBlock &MBB,
265 RISCVVType::VLMUL LMul) {
267 return false;
268
269 assert(MBBI->getOpcode() == TargetOpcode::COPY &&
270 "Unexpected COPY instruction.");
271 Register SrcReg = MBBI->getOperand(1).getReg();
273
274 bool FoundDef = false;
275 bool FirstVSetVLI = false;
276 unsigned FirstSEW = 0;
277 while (MBBI != MBB.begin()) {
278 --MBBI;
279 if (MBBI->isMetaInstruction())
280 continue;
281
282 if (RISCVInstrInfo::isVectorConfigInstr(*MBBI)) {
283 // There is a vsetvli between COPY and source define instruction.
284 // vy = def_vop ... (producing instruction)
285 // ...
286 // vsetvli
287 // ...
288 // vx = COPY vy
289 if (!FoundDef) {
290 if (!FirstVSetVLI) {
291 FirstVSetVLI = true;
292 unsigned FirstVType = MBBI->getOperand(2).getImm();
293 RISCVVType::VLMUL FirstLMul = RISCVVType::getVLMUL(FirstVType);
294 FirstSEW = RISCVVType::getSEW(FirstVType);
295 // The first encountered vsetvli must have the same lmul as the
296 // register class of COPY.
297 if (FirstLMul != LMul)
298 return false;
299 }
300 // Only permit `vsetvli x0, x0, vtype` between COPY and the source
301 // define instruction.
302 if (!RISCVInstrInfo::isVLPreservingConfig(*MBBI))
303 return false;
304 continue;
305 }
306
307 // MBBI is the first vsetvli before the producing instruction.
308 unsigned VType = MBBI->getOperand(2).getImm();
309 // If there is a vsetvli between COPY and the producing instruction.
310 if (FirstVSetVLI) {
311 // If SEW is different, return false.
312 if (RISCVVType::getSEW(VType) != FirstSEW)
313 return false;
314 }
315
316 // If the vsetvli is tail undisturbed, keep the whole register move.
317 if (!RISCVVType::isTailAgnostic(VType))
318 return false;
319
320 // The checking is conservative. We only have register classes for
321 // LMUL = 1/2/4/8. We should be able to convert vmv1r.v to vmv.v.v
322 // for fractional LMUL operations. However, we could not use the vsetvli
323 // lmul for widening operations. The result of widening operation is
324 // 2 x LMUL.
325 return LMul == RISCVVType::getVLMUL(VType);
326 } else if (MBBI->isInlineAsm() || MBBI->isCall()) {
327 return false;
328 } else if (MBBI->getNumDefs()) {
329 // Check all the instructions which will change VL.
330 // For example, vleff has implicit def VL.
331 if (MBBI->modifiesRegister(RISCV::VL, /*TRI=*/nullptr))
332 return false;
333
334 // Only converting whole register copies to vmv.v.v when the defining
335 // value appears in the explicit operands.
336 for (const MachineOperand &MO : MBBI->explicit_operands()) {
337 if (!MO.isReg() || !MO.isDef())
338 continue;
339 if (!FoundDef && TRI->regsOverlap(MO.getReg(), SrcReg)) {
340 // We only permit the source of COPY has the same LMUL as the defined
341 // operand.
342 // There are cases we need to keep the whole register copy if the LMUL
343 // is different.
344 // For example,
345 // $x0 = PseudoVSETIVLI 4, 73 // vsetivli zero, 4, e16,m2,ta,m
346 // $v28m4 = PseudoVWADD_VV_M2 $v26m2, $v8m2
347 // # The COPY may be created by vlmul_trunc intrinsic.
348 // $v26m2 = COPY renamable $v28m2, implicit killed $v28m4
349 //
350 // After widening, the valid value will be 4 x e32 elements. If we
351 // convert the COPY to vmv.v.v, it will only copy 4 x e16 elements.
352 // FIXME: The COPY of subregister of Zvlsseg register will not be able
353 // to convert to vmv.v.[v|i] under the constraint.
354 if (MO.getReg() != SrcReg)
355 return false;
356
357 // In widening reduction instructions with LMUL_1 input vector case,
358 // only checking the LMUL is insufficient due to reduction result is
359 // always LMUL_1.
360 // For example,
361 // $x11 = PseudoVSETIVLI 1, 64 // vsetivli a1, 1, e8, m1, ta, mu
362 // $v8m1 = PseudoVWREDSUM_VS_M1 $v26, $v27
363 // $v26 = COPY killed renamable $v8
364 // After widening, The valid value will be 1 x e16 elements. If we
365 // convert the COPY to vmv.v.v, it will only copy 1 x e8 elements.
366 uint64_t TSFlags = MBBI->getDesc().TSFlags;
368 return false;
369
370 // If the producing instruction does not depend on vsetvli, do not
371 // convert COPY to vmv.v.v. For example, VL1R_V or PseudoVRELOAD.
372 if (!RISCVII::hasSEWOp(TSFlags) || !RISCVII::hasVLOp(TSFlags))
373 return false;
374
375 // Found the definition.
376 FoundDef = true;
377 DefMBBI = MBBI;
378 break;
379 }
380 }
381 }
382 }
383
384 return false;
385}
386
389 const DebugLoc &DL, MCRegister DstReg, MCRegister SrcReg, bool KillSrc,
390 const TargetRegisterClass *RegClass) const {
391 const RISCVRegisterInfo *TRI = STI.getRegisterInfo();
393 unsigned NF = RISCVRI::getNF(RegClass->TSFlags);
394
395 uint16_t SrcEncoding = TRI->getEncodingValue(SrcReg);
396 uint16_t DstEncoding = TRI->getEncodingValue(DstReg);
397 auto [LMulVal, Fractional] = RISCVVType::decodeVLMUL(LMul);
398 assert(!Fractional && "It is impossible be fractional lmul here.");
399 unsigned NumRegs = NF * LMulVal;
400 bool ReversedCopy =
401 forwardCopyWillClobberTuple(DstEncoding, SrcEncoding, NumRegs);
402 if (ReversedCopy) {
403 // If the src and dest overlap when copying a tuple, we need to copy the
404 // registers in reverse.
405 SrcEncoding += NumRegs - 1;
406 DstEncoding += NumRegs - 1;
407 }
408
409 unsigned I = 0;
410 auto GetCopyInfo = [&](uint16_t SrcEncoding, uint16_t DstEncoding)
411 -> std::tuple<RISCVVType::VLMUL, const TargetRegisterClass &, unsigned,
412 unsigned, unsigned> {
413 if (ReversedCopy) {
414 // For reversed copying, if there are enough aligned registers(8/4/2), we
415 // can do a larger copy(LMUL8/4/2).
416 // Besides, we have already known that DstEncoding is larger than
417 // SrcEncoding in forwardCopyWillClobberTuple, so the difference between
418 // DstEncoding and SrcEncoding should be >= LMUL value we try to use to
419 // avoid clobbering.
420 uint16_t Diff = DstEncoding - SrcEncoding;
421 if (I + 8 <= NumRegs && Diff >= 8 && SrcEncoding % 8 == 7 &&
422 DstEncoding % 8 == 7)
423 return {RISCVVType::LMUL_8, RISCV::VRM8RegClass, RISCV::VMV8R_V,
424 RISCV::PseudoVMV_V_V_M8, RISCV::PseudoVMV_V_I_M8};
425 if (I + 4 <= NumRegs && Diff >= 4 && SrcEncoding % 4 == 3 &&
426 DstEncoding % 4 == 3)
427 return {RISCVVType::LMUL_4, RISCV::VRM4RegClass, RISCV::VMV4R_V,
428 RISCV::PseudoVMV_V_V_M4, RISCV::PseudoVMV_V_I_M4};
429 if (I + 2 <= NumRegs && Diff >= 2 && SrcEncoding % 2 == 1 &&
430 DstEncoding % 2 == 1)
431 return {RISCVVType::LMUL_2, RISCV::VRM2RegClass, RISCV::VMV2R_V,
432 RISCV::PseudoVMV_V_V_M2, RISCV::PseudoVMV_V_I_M2};
433 // Or we should do LMUL1 copying.
434 return {RISCVVType::LMUL_1, RISCV::VRRegClass, RISCV::VMV1R_V,
435 RISCV::PseudoVMV_V_V_M1, RISCV::PseudoVMV_V_I_M1};
436 }
437
438 // For forward copying, if source register encoding and destination register
439 // encoding are aligned to 8/4/2, we can do a LMUL8/4/2 copying.
440 if (I + 8 <= NumRegs && SrcEncoding % 8 == 0 && DstEncoding % 8 == 0)
441 return {RISCVVType::LMUL_8, RISCV::VRM8RegClass, RISCV::VMV8R_V,
442 RISCV::PseudoVMV_V_V_M8, RISCV::PseudoVMV_V_I_M8};
443 if (I + 4 <= NumRegs && SrcEncoding % 4 == 0 && DstEncoding % 4 == 0)
444 return {RISCVVType::LMUL_4, RISCV::VRM4RegClass, RISCV::VMV4R_V,
445 RISCV::PseudoVMV_V_V_M4, RISCV::PseudoVMV_V_I_M4};
446 if (I + 2 <= NumRegs && SrcEncoding % 2 == 0 && DstEncoding % 2 == 0)
447 return {RISCVVType::LMUL_2, RISCV::VRM2RegClass, RISCV::VMV2R_V,
448 RISCV::PseudoVMV_V_V_M2, RISCV::PseudoVMV_V_I_M2};
449 // Or we should do LMUL1 copying.
450 return {RISCVVType::LMUL_1, RISCV::VRRegClass, RISCV::VMV1R_V,
451 RISCV::PseudoVMV_V_V_M1, RISCV::PseudoVMV_V_I_M1};
452 };
453
454 while (I != NumRegs) {
455 // For non-segment copying, we only do this once as the registers are always
456 // aligned.
457 // For segment copying, we may do this several times. If the registers are
458 // aligned to larger LMUL, we can eliminate some copyings.
459 auto [LMulCopied, RegClass, Opc, VVOpc, VIOpc] =
460 GetCopyInfo(SrcEncoding, DstEncoding);
461 auto [NumCopied, _] = RISCVVType::decodeVLMUL(LMulCopied);
462
464 if (LMul == LMulCopied &&
465 isConvertibleToVMV_V_V(STI, MBB, MBBI, DefMBBI, LMul)) {
466 Opc = VVOpc;
467 if (DefMBBI->getOpcode() == VIOpc)
468 Opc = VIOpc;
469 }
470
471 // Emit actual copying.
472 // For reversed copying, the encoding should be decreased.
473 MCRegister ActualSrcReg = TRI->findVRegWithEncoding(
474 RegClass, ReversedCopy ? (SrcEncoding - NumCopied + 1) : SrcEncoding);
475 MCRegister ActualDstReg = TRI->findVRegWithEncoding(
476 RegClass, ReversedCopy ? (DstEncoding - NumCopied + 1) : DstEncoding);
477
478 auto MIB = BuildMI(MBB, MBBI, DL, get(Opc), ActualDstReg);
479 bool UseVMV_V_I = RISCV::getRVVMCOpcode(Opc) == RISCV::VMV_V_I;
480 bool UseVMV = UseVMV_V_I || RISCV::getRVVMCOpcode(Opc) == RISCV::VMV_V_V;
481 if (UseVMV)
482 MIB.addReg(ActualDstReg, RegState::Undef);
483 if (UseVMV_V_I)
484 MIB = MIB.add(DefMBBI->getOperand(2));
485 else
486 MIB = MIB.addReg(ActualSrcReg, getKillRegState(KillSrc));
487 if (UseVMV) {
488 const MCInstrDesc &Desc = DefMBBI->getDesc();
489 MIB.add(DefMBBI->getOperand(RISCVII::getVLOpNum(Desc))); // AVL
490 unsigned Log2SEW =
491 DefMBBI->getOperand(RISCVII::getSEWOpNum(Desc)).getImm();
492 MIB.addImm(Log2SEW ? Log2SEW : 3); // SEW
493 MIB.addImm(0); // tu, mu
494 MIB.addReg(RISCV::VL, RegState::Implicit);
495 MIB.addReg(RISCV::VTYPE, RegState::Implicit);
496 }
497 // Add an implicit read of the original source to silence the verifier
498 // in the cases where some of the smaller VRs we're copying from might be
499 // undef, caused by the fact that the original, larger source VR might not
500 // be fully initialized at the time this COPY happens.
501 MIB.addReg(SrcReg, RegState::Implicit);
502
503 // If we are copying reversely, we should decrease the encoding.
504 SrcEncoding += (ReversedCopy ? -NumCopied : NumCopied);
505 DstEncoding += (ReversedCopy ? -NumCopied : NumCopied);
506 I += NumCopied;
507 }
508}
509
512 const DebugLoc &DL, Register DstReg,
513 Register SrcReg, bool KillSrc,
514 bool RenamableDest, bool RenamableSrc) const {
515 const TargetRegisterInfo *TRI = STI.getRegisterInfo();
516 RegState KillFlag = getKillRegState(KillSrc);
517
518 if (RISCV::GPRRegClass.contains(DstReg, SrcReg)) {
519 BuildMI(MBB, MBBI, DL, get(RISCV::ADDI), DstReg)
520 .addReg(SrcReg, KillFlag | getRenamableRegState(RenamableSrc))
521 .addImm(0);
522 return;
523 }
524
525 // Extracting from X0_Pair may create copies from DUMMY_REG_PAIR_WITH_X0.
526 if (SrcReg == RISCV::DUMMY_REG_PAIR_WITH_X0 &&
527 RISCV::GPRRegClass.contains(DstReg)) {
528 BuildMI(MBB, MBBI, DL, get(RISCV::ADDI), DstReg)
529 .addReg(RISCV::X0)
530 .addImm(0);
531 return;
532 }
533
534 if (RISCV::GPRF16RegClass.contains(DstReg, SrcReg)) {
535 BuildMI(MBB, MBBI, DL, get(RISCV::PseudoMV_FPR16INX), DstReg)
536 .addReg(SrcReg, KillFlag | getRenamableRegState(RenamableSrc));
537 return;
538 }
539
540 if (RISCV::GPRF32RegClass.contains(DstReg, SrcReg)) {
541 BuildMI(MBB, MBBI, DL, get(RISCV::PseudoMV_FPR32INX), DstReg)
542 .addReg(SrcReg, KillFlag | getRenamableRegState(RenamableSrc));
543 return;
544 }
545
546 if (RISCV::GPRPairRegClass.contains(DstReg, SrcReg)) {
547 if (!STI.is64Bit()) {
548 if (STI.hasStdExtZdinx()) {
549 // On RV32_Zdinx, FMV.D will move a pair of registers to another pair of
550 // registers, in one instruction.
551 BuildMI(MBB, MBBI, DL, get(RISCV::FSGNJ_D_IN32X), DstReg)
552 .addReg(SrcReg, getRenamableRegState(RenamableSrc))
553 .addReg(SrcReg, KillFlag | getRenamableRegState(RenamableSrc));
554 return;
555 }
556
557 if (STI.hasStdExtP()) {
558 // On RV32P, `padd.dw` is a GPR Pair Add
559 BuildMI(MBB, MBBI, DL, get(RISCV::PADD_DW), DstReg)
560 .addReg(RISCV::X0_Pair)
561 .addReg(SrcReg, KillFlag | getRenamableRegState(RenamableSrc));
562 return;
563 }
564 }
565
566 MCRegister EvenReg = TRI->getSubReg(SrcReg, RISCV::sub_gpr_even);
567 MCRegister OddReg = TRI->getSubReg(SrcReg, RISCV::sub_gpr_odd);
568 // We need to correct the odd register of X0_Pair.
569 if (OddReg == RISCV::DUMMY_REG_PAIR_WITH_X0)
570 OddReg = RISCV::X0;
571 assert(DstReg != RISCV::X0_Pair && "Cannot write to X0_Pair");
572
573 // Emit an ADDI for both parts of GPRPair.
574 BuildMI(MBB, MBBI, DL, get(RISCV::ADDI),
575 TRI->getSubReg(DstReg, RISCV::sub_gpr_even))
576 .addReg(EvenReg, KillFlag)
577 .addImm(0);
578 BuildMI(MBB, MBBI, DL, get(RISCV::ADDI),
579 TRI->getSubReg(DstReg, RISCV::sub_gpr_odd))
580 .addReg(OddReg, KillFlag)
581 .addImm(0);
582 return;
583 }
584
585 // Handle copy from csr
586 if (RISCV::VCSRRegClass.contains(SrcReg) &&
587 RISCV::GPRRegClass.contains(DstReg)) {
588 BuildMI(MBB, MBBI, DL, get(RISCV::CSRRS), DstReg)
589 .addImm(RISCVSysReg::lookupSysRegByName(TRI->getName(SrcReg))->Encoding)
590 .addReg(RISCV::X0);
591 return;
592 }
593
594 if (RISCV::FPR16RegClass.contains(DstReg, SrcReg)) {
595 unsigned Opc;
596 if (STI.hasStdExtZfh()) {
597 Opc = RISCV::FSGNJ_H;
598 } else {
599 assert(STI.hasStdExtF() &&
600 (STI.hasStdExtZfhmin() || STI.hasStdExtZfbfmin()) &&
601 "Unexpected extensions");
602 // Zfhmin/Zfbfmin doesn't have FSGNJ_H, replace FSGNJ_H with FSGNJ_S.
603 DstReg = TRI->getMatchingSuperReg(DstReg, RISCV::sub_16,
604 &RISCV::FPR32RegClass);
605 SrcReg = TRI->getMatchingSuperReg(SrcReg, RISCV::sub_16,
606 &RISCV::FPR32RegClass);
607 Opc = RISCV::FSGNJ_S;
608 }
609 BuildMI(MBB, MBBI, DL, get(Opc), DstReg)
610 .addReg(SrcReg, KillFlag)
611 .addReg(SrcReg, KillFlag);
612 return;
613 }
614
615 if (RISCV::FPR32RegClass.contains(DstReg, SrcReg)) {
616 BuildMI(MBB, MBBI, DL, get(RISCV::FSGNJ_S), DstReg)
617 .addReg(SrcReg, KillFlag)
618 .addReg(SrcReg, KillFlag);
619 return;
620 }
621
622 if (RISCV::FPR64RegClass.contains(DstReg, SrcReg)) {
623 BuildMI(MBB, MBBI, DL, get(RISCV::FSGNJ_D), DstReg)
624 .addReg(SrcReg, KillFlag)
625 .addReg(SrcReg, KillFlag);
626 return;
627 }
628
629 if (RISCV::FPR32RegClass.contains(DstReg) &&
630 RISCV::GPRRegClass.contains(SrcReg)) {
631 BuildMI(MBB, MBBI, DL, get(RISCV::FMV_W_X), DstReg)
632 .addReg(SrcReg, KillFlag);
633 return;
634 }
635
636 if (RISCV::GPRRegClass.contains(DstReg) &&
637 RISCV::FPR32RegClass.contains(SrcReg)) {
638 BuildMI(MBB, MBBI, DL, get(RISCV::FMV_X_W), DstReg)
639 .addReg(SrcReg, KillFlag);
640 return;
641 }
642
643 if (RISCV::FPR64RegClass.contains(DstReg) &&
644 RISCV::GPRRegClass.contains(SrcReg)) {
645 assert(STI.getXLen() == 64 && "Unexpected GPR size");
646 BuildMI(MBB, MBBI, DL, get(RISCV::FMV_D_X), DstReg)
647 .addReg(SrcReg, KillFlag);
648 return;
649 }
650
651 if (RISCV::GPRRegClass.contains(DstReg) &&
652 RISCV::FPR64RegClass.contains(SrcReg)) {
653 assert(STI.getXLen() == 64 && "Unexpected GPR size");
654 BuildMI(MBB, MBBI, DL, get(RISCV::FMV_X_D), DstReg)
655 .addReg(SrcReg, KillFlag);
656 return;
657 }
658
659 // VR->VR copies.
660 const TargetRegisterClass *RegClass =
661 TRI->getCommonMinimalPhysRegClass(SrcReg, DstReg);
662 if (RISCVRegisterInfo::isRVVRegClass(RegClass)) {
663 copyPhysRegVector(MBB, MBBI, DL, DstReg, SrcReg, KillSrc, RegClass);
664 return;
665 }
666
667 llvm_unreachable("Impossible reg-to-reg copy");
668}
669
672 Register SrcReg, bool IsKill, int FI,
673 const TargetRegisterClass *RC,
674 Register VReg,
675 MachineInstr::MIFlag Flags) const {
676 MachineFunction *MF = MBB.getParent();
677 MachineFrameInfo &MFI = MF->getFrameInfo();
678 Align Alignment = MFI.getObjectAlign(FI);
679
680 unsigned Opcode;
681 if (RISCV::GPRRegClass.hasSubClassEq(RC)) {
682 Opcode = RegInfo.getRegSizeInBits(RISCV::GPRRegClass) == 32 ? RISCV::SW
683 : RISCV::SD;
684 } else if (RISCV::GPRF16RegClass.hasSubClassEq(RC)) {
685 Opcode = RISCV::SH_INX;
686 } else if (RISCV::GPRF32RegClass.hasSubClassEq(RC)) {
687 Opcode = RISCV::SW_INX;
688 } else if (RISCV::GPRPairRegClass.hasSubClassEq(RC)) {
689 if (!STI.is64Bit() && STI.hasStdExtZilsd() &&
690 Alignment >= STI.getZilsdAlign()) {
691 Opcode = RISCV::SD_RV32;
692 } else {
693 Opcode = RISCV::PseudoRV32ZdinxSD;
694 }
695 } else if (RISCV::FPR16RegClass.hasSubClassEq(RC)) {
696 Opcode = RISCV::FSH;
697 } else if (RISCV::FPR32RegClass.hasSubClassEq(RC)) {
698 Opcode = RISCV::FSW;
699 } else if (RISCV::FPR64RegClass.hasSubClassEq(RC)) {
700 Opcode = RISCV::FSD;
701 } else if (RISCV::VRRegClass.hasSubClassEq(RC)) {
702 Opcode = RISCV::VS1R_V;
703 } else if (RISCV::VRM2RegClass.hasSubClassEq(RC)) {
704 Opcode = RISCV::VS2R_V;
705 } else if (RISCV::VRM4RegClass.hasSubClassEq(RC)) {
706 Opcode = RISCV::VS4R_V;
707 } else if (RISCV::VRM8RegClass.hasSubClassEq(RC)) {
708 Opcode = RISCV::VS8R_V;
709 } else if (RISCV::VRN2M1RegClass.hasSubClassEq(RC))
710 Opcode = RISCV::PseudoVSPILL2_M1;
711 else if (RISCV::VRN2M2RegClass.hasSubClassEq(RC))
712 Opcode = RISCV::PseudoVSPILL2_M2;
713 else if (RISCV::VRN2M4RegClass.hasSubClassEq(RC))
714 Opcode = RISCV::PseudoVSPILL2_M4;
715 else if (RISCV::VRN3M1RegClass.hasSubClassEq(RC))
716 Opcode = RISCV::PseudoVSPILL3_M1;
717 else if (RISCV::VRN3M2RegClass.hasSubClassEq(RC))
718 Opcode = RISCV::PseudoVSPILL3_M2;
719 else if (RISCV::VRN4M1RegClass.hasSubClassEq(RC))
720 Opcode = RISCV::PseudoVSPILL4_M1;
721 else if (RISCV::VRN4M2RegClass.hasSubClassEq(RC))
722 Opcode = RISCV::PseudoVSPILL4_M2;
723 else if (RISCV::VRN5M1RegClass.hasSubClassEq(RC))
724 Opcode = RISCV::PseudoVSPILL5_M1;
725 else if (RISCV::VRN6M1RegClass.hasSubClassEq(RC))
726 Opcode = RISCV::PseudoVSPILL6_M1;
727 else if (RISCV::VRN7M1RegClass.hasSubClassEq(RC))
728 Opcode = RISCV::PseudoVSPILL7_M1;
729 else if (RISCV::VRN8M1RegClass.hasSubClassEq(RC))
730 Opcode = RISCV::PseudoVSPILL8_M1;
731 else
732 llvm_unreachable("Can't store this register to stack slot");
733
737 TypeSize::getScalable(MFI.getObjectSize(FI)), Alignment);
738
740 BuildMI(MBB, I, DebugLoc(), get(Opcode))
741 .addReg(SrcReg, getKillRegState(IsKill))
742 .addFrameIndex(FI)
743 .addMemOperand(MMO)
744 .setMIFlag(Flags);
745 NumVRegSpilled += RegInfo.getRegSizeInBits(*RC) / RISCV::RVVBitsPerBlock;
746 } else {
749 MFI.getObjectSize(FI), Alignment);
750
751 BuildMI(MBB, I, DebugLoc(), get(Opcode))
752 .addReg(SrcReg, getKillRegState(IsKill))
753 .addFrameIndex(FI)
754 .addImm(0)
755 .addMemOperand(MMO)
756 .setMIFlag(Flags);
757 }
758}
759
762 Register DstReg, int FI,
763 const TargetRegisterClass *RC,
764 Register VReg, unsigned SubReg,
765 MachineInstr::MIFlag Flags) const {
766 MachineFunction *MF = MBB.getParent();
767 MachineFrameInfo &MFI = MF->getFrameInfo();
768 Align Alignment = MFI.getObjectAlign(FI);
769 DebugLoc DL =
770 Flags & MachineInstr::FrameDestroy ? MBB.findDebugLoc(I) : DebugLoc();
771
772 unsigned Opcode;
773 if (RISCV::GPRRegClass.hasSubClassEq(RC)) {
774 Opcode = RegInfo.getRegSizeInBits(RISCV::GPRRegClass) == 32 ? RISCV::LW
775 : RISCV::LD;
776 } else if (RISCV::GPRF16RegClass.hasSubClassEq(RC)) {
777 Opcode = RISCV::LH_INX;
778 } else if (RISCV::GPRF32RegClass.hasSubClassEq(RC)) {
779 Opcode = RISCV::LW_INX;
780 } else if (RISCV::GPRPairRegClass.hasSubClassEq(RC)) {
781 if (!STI.is64Bit() && STI.hasStdExtZilsd() &&
782 Alignment >= STI.getZilsdAlign()) {
783 Opcode = RISCV::LD_RV32;
784 } else {
785 Opcode = RISCV::PseudoRV32ZdinxLD;
786 }
787 } else if (RISCV::FPR16RegClass.hasSubClassEq(RC)) {
788 Opcode = RISCV::FLH;
789 } else if (RISCV::FPR32RegClass.hasSubClassEq(RC)) {
790 Opcode = RISCV::FLW;
791 } else if (RISCV::FPR64RegClass.hasSubClassEq(RC)) {
792 Opcode = RISCV::FLD;
793 } else if (RISCV::VRRegClass.hasSubClassEq(RC)) {
794 Opcode = RISCV::VL1RE8_V;
795 } else if (RISCV::VRM2RegClass.hasSubClassEq(RC)) {
796 Opcode = RISCV::VL2RE8_V;
797 } else if (RISCV::VRM4RegClass.hasSubClassEq(RC)) {
798 Opcode = RISCV::VL4RE8_V;
799 } else if (RISCV::VRM8RegClass.hasSubClassEq(RC)) {
800 Opcode = RISCV::VL8RE8_V;
801 } else if (RISCV::VRN2M1RegClass.hasSubClassEq(RC))
802 Opcode = RISCV::PseudoVRELOAD2_M1;
803 else if (RISCV::VRN2M2RegClass.hasSubClassEq(RC))
804 Opcode = RISCV::PseudoVRELOAD2_M2;
805 else if (RISCV::VRN2M4RegClass.hasSubClassEq(RC))
806 Opcode = RISCV::PseudoVRELOAD2_M4;
807 else if (RISCV::VRN3M1RegClass.hasSubClassEq(RC))
808 Opcode = RISCV::PseudoVRELOAD3_M1;
809 else if (RISCV::VRN3M2RegClass.hasSubClassEq(RC))
810 Opcode = RISCV::PseudoVRELOAD3_M2;
811 else if (RISCV::VRN4M1RegClass.hasSubClassEq(RC))
812 Opcode = RISCV::PseudoVRELOAD4_M1;
813 else if (RISCV::VRN4M2RegClass.hasSubClassEq(RC))
814 Opcode = RISCV::PseudoVRELOAD4_M2;
815 else if (RISCV::VRN5M1RegClass.hasSubClassEq(RC))
816 Opcode = RISCV::PseudoVRELOAD5_M1;
817 else if (RISCV::VRN6M1RegClass.hasSubClassEq(RC))
818 Opcode = RISCV::PseudoVRELOAD6_M1;
819 else if (RISCV::VRN7M1RegClass.hasSubClassEq(RC))
820 Opcode = RISCV::PseudoVRELOAD7_M1;
821 else if (RISCV::VRN8M1RegClass.hasSubClassEq(RC))
822 Opcode = RISCV::PseudoVRELOAD8_M1;
823 else
824 llvm_unreachable("Can't load this register from stack slot");
825
829 TypeSize::getScalable(MFI.getObjectSize(FI)), Alignment);
830
832 BuildMI(MBB, I, DL, get(Opcode), DstReg)
833 .addFrameIndex(FI)
834 .addMemOperand(MMO)
835 .setMIFlag(Flags);
836 NumVRegReloaded += RegInfo.getRegSizeInBits(*RC) / RISCV::RVVBitsPerBlock;
837 } else {
840 MFI.getObjectSize(FI), Alignment);
841
842 BuildMI(MBB, I, DL, get(Opcode), DstReg)
843 .addFrameIndex(FI)
844 .addImm(0)
845 .addMemOperand(MMO)
846 .setMIFlag(Flags);
847 }
848}
849std::optional<unsigned> getFoldedOpcode(MachineFunction &MF, MachineInstr &MI,
851 const RISCVSubtarget &ST) {
852
853 // The below optimizations narrow the load so they are only valid for little
854 // endian.
855 // TODO: Support big endian by adding an offset into the frame object?
856 if (MF.getDataLayout().isBigEndian())
857 return std::nullopt;
858
859 // Fold load from stack followed by sext.b/sext.h/sext.w/zext.b/zext.h/zext.w.
860 if (Ops.size() != 1 || Ops[0] != 1)
861 return std::nullopt;
862
863 switch (MI.getOpcode()) {
864 default:
865 if (RISCVInstrInfo::isSEXT_W(MI))
866 return RISCV::LW;
867 if (RISCVInstrInfo::isZEXT_W(MI))
868 return RISCV::LWU;
869 if (RISCVInstrInfo::isZEXT_B(MI))
870 return RISCV::LBU;
871 break;
872 case RISCV::SEXT_H:
873 return RISCV::LH;
874 case RISCV::SEXT_B:
875 return RISCV::LB;
876 case RISCV::ZEXT_H_RV32:
877 case RISCV::ZEXT_H_RV64:
878 return RISCV::LHU;
879 }
880
881 switch (RISCV::getRVVMCOpcode(MI.getOpcode())) {
882 default:
883 return std::nullopt;
884 case RISCV::VMV_X_S: {
885 unsigned Log2SEW =
886 MI.getOperand(RISCVII::getSEWOpNum(MI.getDesc())).getImm();
887 if (ST.getXLen() < (1U << Log2SEW))
888 return std::nullopt;
889 switch (Log2SEW) {
890 case 3:
891 return RISCV::LB;
892 case 4:
893 return RISCV::LH;
894 case 5:
895 return RISCV::LW;
896 case 6:
897 return RISCV::LD;
898 default:
899 llvm_unreachable("Unexpected SEW");
900 }
901 }
902 case RISCV::VFMV_F_S: {
903 unsigned Log2SEW =
904 MI.getOperand(RISCVII::getSEWOpNum(MI.getDesc())).getImm();
905 switch (Log2SEW) {
906 case 4:
907 return RISCV::FLH;
908 case 5:
909 return RISCV::FLW;
910 case 6:
911 return RISCV::FLD;
912 default:
913 llvm_unreachable("Unexpected SEW");
914 }
915 }
916 }
917}
918
919// This is the version used during InlineSpiller::spillAroundUses
922 ArrayRef<unsigned> Ops, int FrameIndex,
923 MachineInstr *&CopyMI, LiveIntervals *LIS,
924 VirtRegMap *VRM) const {
926 std::optional<unsigned> LoadOpc = getFoldedOpcode(MF, MI, Ops, STI);
927 if (!LoadOpc)
928 return nullptr;
929 Register DstReg = MI.getOperand(0).getReg();
930 return BuildMI(*MI.getParent(), InsertPt, MI.getDebugLoc(), get(*LoadOpc),
931 DstReg)
932 .addFrameIndex(FrameIndex)
933 .addImm(0);
934}
935
936static unsigned getLoadPredicatedOpcode(unsigned Opcode) {
937 switch (Opcode) {
938 case RISCV::LB:
939 return RISCV::PseudoCCLB;
940 case RISCV::LBU:
941 return RISCV::PseudoCCLBU;
942 case RISCV::LH:
943 return RISCV::PseudoCCLH;
944 case RISCV::LHU:
945 return RISCV::PseudoCCLHU;
946 case RISCV::LW:
947 return RISCV::PseudoCCLW;
948 case RISCV::LWU:
949 return RISCV::PseudoCCLWU;
950 case RISCV::LD:
951 return RISCV::PseudoCCLD;
952 case RISCV::QC_E_LB:
953 return RISCV::PseudoCCQC_E_LB;
954 case RISCV::QC_E_LBU:
955 return RISCV::PseudoCCQC_E_LBU;
956 case RISCV::QC_E_LH:
957 return RISCV::PseudoCCQC_E_LH;
958 case RISCV::QC_E_LHU:
959 return RISCV::PseudoCCQC_E_LHU;
960 case RISCV::QC_E_LW:
961 return RISCV::PseudoCCQC_E_LW;
962 default:
963 return 0;
964 }
965}
966
969 MachineInstr &LoadMI, MachineInstr *&CopyMI, LiveIntervals *LIS,
970 VirtRegMap *VRM) const {
972 // For now, only handle RISCV::PseudoCCMOVGPR.
973 if (MI.getOpcode() != RISCV::PseudoCCMOVGPR)
974 return nullptr;
975
976 unsigned PredOpc = getLoadPredicatedOpcode(LoadMI.getOpcode());
977
978 if (!STI.hasShortForwardBranchILoad() || !PredOpc)
979 return nullptr;
980
982 if (Ops.size() != 1 || (Ops[0] != 1 && Ops[0] != 2))
983 return nullptr;
984
985 bool Invert = Ops[0] == 2;
986 const MachineOperand &FalseReg = MI.getOperand(!Invert ? 2 : 1);
987 Register DestReg = MI.getOperand(0).getReg();
988 const TargetRegisterClass *PreviousClass = MRI.getRegClass(FalseReg.getReg());
989 if (!MRI.constrainRegClass(DestReg, PreviousClass))
990 return nullptr;
991
992 // Create a new predicated version of DefMI.
993 MachineInstrBuilder NewMI = BuildMI(*MI.getParent(), InsertPt,
994 MI.getDebugLoc(), get(PredOpc), DestReg);
995
996 // Copy the false register.
997 NewMI.add(FalseReg);
998
999 // Copy all the DefMI operands.
1000 const MCInstrDesc &DefDesc = LoadMI.getDesc();
1001 for (unsigned i = 1, e = DefDesc.getNumOperands(); i != e; ++i)
1002 NewMI.add(LoadMI.getOperand(i));
1003
1004 // Add branch opcode, inverting if necessary.
1005 unsigned BCC = MI.getOperand(MI.getNumExplicitOperands() - 3).getImm();
1006 if (!Invert)
1008 NewMI.addImm(BCC);
1009
1010 // Copy condition portion
1011 NewMI.add({MI.getOperand(MI.getNumExplicitOperands() - 2),
1012 MI.getOperand(MI.getNumExplicitOperands() - 1)});
1013 NewMI.cloneMemRefs(LoadMI);
1014 return NewMI;
1015}
1016
1019 const DebugLoc &DL, Register DstReg, uint64_t Val,
1020 MachineInstr::MIFlag Flag, bool DstRenamable,
1021 bool DstIsDead) const {
1022 Register SrcReg = RISCV::X0;
1023
1024 // For RV32, allow a sign or unsigned 32 bit value.
1025 if (!STI.is64Bit() && !isInt<32>(Val)) {
1026 // If have a uimm32 it will still fit in a register so we can allow it.
1027 if (!isUInt<32>(Val))
1028 report_fatal_error("Should only materialize 32-bit constants for RV32");
1029
1030 // Sign extend for generateInstSeq.
1031 Val = SignExtend64<32>(Val);
1032 }
1033
1035 assert(!Seq.empty());
1036
1037 bool SrcRenamable = false;
1038 unsigned Num = 0;
1039
1040 for (const RISCVMatInt::Inst &Inst : Seq) {
1041 bool LastItem = ++Num == Seq.size();
1042 RegState DstRegState = getDeadRegState(DstIsDead && LastItem) |
1043 getRenamableRegState(DstRenamable);
1044 RegState SrcRegState = getKillRegState(SrcReg != RISCV::X0) |
1045 getRenamableRegState(SrcRenamable);
1046 switch (Inst.getOpndKind()) {
1047 case RISCVMatInt::Imm:
1048 BuildMI(MBB, MBBI, DL, get(Inst.getOpcode()))
1049 .addReg(DstReg, RegState::Define | DstRegState)
1050 .addImm(Inst.getImm())
1051 .setMIFlag(Flag);
1052 break;
1053 case RISCVMatInt::RegX0:
1054 BuildMI(MBB, MBBI, DL, get(Inst.getOpcode()))
1055 .addReg(DstReg, RegState::Define | DstRegState)
1056 .addReg(SrcReg, SrcRegState)
1057 .addReg(RISCV::X0)
1058 .setMIFlag(Flag);
1059 break;
1061 BuildMI(MBB, MBBI, DL, get(Inst.getOpcode()))
1062 .addReg(DstReg, RegState::Define | DstRegState)
1063 .addReg(SrcReg, SrcRegState)
1064 .addReg(SrcReg, SrcRegState)
1065 .setMIFlag(Flag);
1066 break;
1068 BuildMI(MBB, MBBI, DL, get(Inst.getOpcode()))
1069 .addReg(DstReg, RegState::Define | DstRegState)
1070 .addReg(SrcReg, SrcRegState)
1071 .addImm(Inst.getImm())
1072 .setMIFlag(Flag);
1073 break;
1074 }
1075
1076 // Only the first instruction has X0 as its source.
1077 SrcReg = DstReg;
1078 SrcRenamable = DstRenamable;
1079 }
1080}
1081
1083 switch (Opc) {
1084 default:
1085 return RISCVCC::COND_INVALID;
1086 case RISCV::BEQ:
1087 case RISCV::BEQI:
1088 case RISCV::CV_BEQIMM:
1089 case RISCV::QC_BEQI:
1090 case RISCV::QC_E_BEQI:
1091 case RISCV::NDS_BBC:
1092 case RISCV::NDS_BEQC:
1093 return RISCVCC::COND_EQ;
1094 case RISCV::BNE:
1095 case RISCV::BNEI:
1096 case RISCV::QC_BNEI:
1097 case RISCV::QC_E_BNEI:
1098 case RISCV::CV_BNEIMM:
1099 case RISCV::NDS_BBS:
1100 case RISCV::NDS_BNEC:
1101 return RISCVCC::COND_NE;
1102 case RISCV::BLT:
1103 case RISCV::QC_BLTI:
1104 case RISCV::QC_E_BLTI:
1105 return RISCVCC::COND_LT;
1106 case RISCV::BGE:
1107 case RISCV::QC_BGEI:
1108 case RISCV::QC_E_BGEI:
1109 return RISCVCC::COND_GE;
1110 case RISCV::BLTU:
1111 case RISCV::QC_BLTUI:
1112 case RISCV::QC_E_BLTUI:
1113 return RISCVCC::COND_LTU;
1114 case RISCV::BGEU:
1115 case RISCV::QC_BGEUI:
1116 case RISCV::QC_E_BGEUI:
1117 return RISCVCC::COND_GEU;
1118 }
1119}
1120
1122 int64_t C1) {
1123 switch (CC) {
1124 default:
1125 llvm_unreachable("Unexpected CC");
1126 case RISCVCC::COND_EQ:
1127 return C0 == C1;
1128 case RISCVCC::COND_NE:
1129 return C0 != C1;
1130 case RISCVCC::COND_LT:
1131 return C0 < C1;
1132 case RISCVCC::COND_GE:
1133 return C0 >= C1;
1134 case RISCVCC::COND_LTU:
1135 return (uint64_t)C0 < (uint64_t)C1;
1136 case RISCVCC::COND_GEU:
1137 return (uint64_t)C0 >= (uint64_t)C1;
1138 }
1139}
1140
1141// The contents of values added to Cond are not examined outside of
1142// RISCVInstrInfo, giving us flexibility in what to push to it. For RISCV, we
1143// push BranchOpcode, Reg1, Reg2.
1146 // Block ends with fall-through condbranch.
1147 assert(LastInst.getDesc().isConditionalBranch() &&
1148 "Unknown conditional branch");
1149 Target = LastInst.getOperand(2).getMBB();
1150 Cond.push_back(MachineOperand::CreateImm(LastInst.getOpcode()));
1151 Cond.push_back(LastInst.getOperand(0));
1152 Cond.push_back(LastInst.getOperand(1));
1153}
1154
1155static unsigned getInverseXqcicmOpcode(unsigned Opcode) {
1156 switch (Opcode) {
1157 default:
1158 llvm_unreachable("Unexpected Opcode");
1159 case RISCV::QC_MVEQ:
1160 return RISCV::QC_MVNE;
1161 case RISCV::QC_MVNE:
1162 return RISCV::QC_MVEQ;
1163 case RISCV::QC_MVLT:
1164 return RISCV::QC_MVGE;
1165 case RISCV::QC_MVGE:
1166 return RISCV::QC_MVLT;
1167 case RISCV::QC_MVLTU:
1168 return RISCV::QC_MVGEU;
1169 case RISCV::QC_MVGEU:
1170 return RISCV::QC_MVLTU;
1171 case RISCV::QC_MVEQI:
1172 return RISCV::QC_MVNEI;
1173 case RISCV::QC_MVNEI:
1174 return RISCV::QC_MVEQI;
1175 case RISCV::QC_MVLTI:
1176 return RISCV::QC_MVGEI;
1177 case RISCV::QC_MVGEI:
1178 return RISCV::QC_MVLTI;
1179 case RISCV::QC_MVLTUI:
1180 return RISCV::QC_MVGEUI;
1181 case RISCV::QC_MVGEUI:
1182 return RISCV::QC_MVLTUI;
1183 }
1184}
1185
1186unsigned RISCVCC::getBrCond(RISCVCC::CondCode CC, unsigned SelectOpc) {
1187 switch (SelectOpc) {
1188 default:
1189 switch (CC) {
1190 default:
1191 llvm_unreachable("Unexpected condition code!");
1192 case RISCVCC::COND_EQ:
1193 return RISCV::BEQ;
1194 case RISCVCC::COND_NE:
1195 return RISCV::BNE;
1196 case RISCVCC::COND_LT:
1197 return RISCV::BLT;
1198 case RISCVCC::COND_GE:
1199 return RISCV::BGE;
1200 case RISCVCC::COND_LTU:
1201 return RISCV::BLTU;
1202 case RISCVCC::COND_GEU:
1203 return RISCV::BGEU;
1204 }
1205 break;
1206 case RISCV::Select_GPR_Using_CC_Imm5_Zibi:
1207 switch (CC) {
1208 default:
1209 llvm_unreachable("Unexpected condition code!");
1210 case RISCVCC::COND_EQ:
1211 return RISCV::BEQI;
1212 case RISCVCC::COND_NE:
1213 return RISCV::BNEI;
1214 }
1215 break;
1216 case RISCV::Select_GPR_Using_CC_SImm5_CV:
1217 switch (CC) {
1218 default:
1219 llvm_unreachable("Unexpected condition code!");
1220 case RISCVCC::COND_EQ:
1221 return RISCV::CV_BEQIMM;
1222 case RISCVCC::COND_NE:
1223 return RISCV::CV_BNEIMM;
1224 }
1225 break;
1226 case RISCV::Select_GPRNoX0_Using_CC_SImm5NonZero_QC:
1227 switch (CC) {
1228 default:
1229 llvm_unreachable("Unexpected condition code!");
1230 case RISCVCC::COND_EQ:
1231 return RISCV::QC_BEQI;
1232 case RISCVCC::COND_NE:
1233 return RISCV::QC_BNEI;
1234 case RISCVCC::COND_LT:
1235 return RISCV::QC_BLTI;
1236 case RISCVCC::COND_GE:
1237 return RISCV::QC_BGEI;
1238 }
1239 break;
1240 case RISCV::Select_GPRNoX0_Using_CC_UImm5NonZero_QC:
1241 switch (CC) {
1242 default:
1243 llvm_unreachable("Unexpected condition code!");
1244 case RISCVCC::COND_LTU:
1245 return RISCV::QC_BLTUI;
1246 case RISCVCC::COND_GEU:
1247 return RISCV::QC_BGEUI;
1248 }
1249 break;
1250 case RISCV::Select_GPRNoX0_Using_CC_SImm16NonZero_QC:
1251 switch (CC) {
1252 default:
1253 llvm_unreachable("Unexpected condition code!");
1254 case RISCVCC::COND_EQ:
1255 return RISCV::QC_E_BEQI;
1256 case RISCVCC::COND_NE:
1257 return RISCV::QC_E_BNEI;
1258 case RISCVCC::COND_LT:
1259 return RISCV::QC_E_BLTI;
1260 case RISCVCC::COND_GE:
1261 return RISCV::QC_E_BGEI;
1262 }
1263 break;
1264 case RISCV::Select_GPRNoX0_Using_CC_UImm16NonZero_QC:
1265 switch (CC) {
1266 default:
1267 llvm_unreachable("Unexpected condition code!");
1268 case RISCVCC::COND_LTU:
1269 return RISCV::QC_E_BLTUI;
1270 case RISCVCC::COND_GEU:
1271 return RISCV::QC_E_BGEUI;
1272 }
1273 break;
1274 case RISCV::Select_GPR_Using_CC_UImmLog2XLen_NDS:
1275 switch (CC) {
1276 default:
1277 llvm_unreachable("Unexpected condition code!");
1278 case RISCVCC::COND_EQ:
1279 return RISCV::NDS_BBC;
1280 case RISCVCC::COND_NE:
1281 return RISCV::NDS_BBS;
1282 }
1283 break;
1284 case RISCV::Select_GPR_Using_CC_UImm7_NDS:
1285 switch (CC) {
1286 default:
1287 llvm_unreachable("Unexpected condition code!");
1288 case RISCVCC::COND_EQ:
1289 return RISCV::NDS_BEQC;
1290 case RISCVCC::COND_NE:
1291 return RISCV::NDS_BNEC;
1292 }
1293 break;
1294 }
1295}
1296
1298 switch (CC) {
1299 default:
1300 llvm_unreachable("Unrecognized conditional branch");
1301 case RISCVCC::COND_EQ:
1302 return RISCVCC::COND_NE;
1303 case RISCVCC::COND_NE:
1304 return RISCVCC::COND_EQ;
1305 case RISCVCC::COND_LT:
1306 return RISCVCC::COND_GE;
1307 case RISCVCC::COND_GE:
1308 return RISCVCC::COND_LT;
1309 case RISCVCC::COND_LTU:
1310 return RISCVCC::COND_GEU;
1311 case RISCVCC::COND_GEU:
1312 return RISCVCC::COND_LTU;
1313 }
1314}
1315
1316// Return inverse branch
1317unsigned RISCVCC::getInverseBranchOpcode(unsigned BCC) {
1318 switch (BCC) {
1319 default:
1320 llvm_unreachable("Unexpected branch opcode!");
1321 case RISCV::BEQ:
1322 return RISCV::BNE;
1323 case RISCV::BEQI:
1324 return RISCV::BNEI;
1325 case RISCV::BNE:
1326 return RISCV::BEQ;
1327 case RISCV::BNEI:
1328 return RISCV::BEQI;
1329 case RISCV::BLT:
1330 return RISCV::BGE;
1331 case RISCV::BGE:
1332 return RISCV::BLT;
1333 case RISCV::BLTU:
1334 return RISCV::BGEU;
1335 case RISCV::BGEU:
1336 return RISCV::BLTU;
1337 case RISCV::CV_BEQIMM:
1338 return RISCV::CV_BNEIMM;
1339 case RISCV::CV_BNEIMM:
1340 return RISCV::CV_BEQIMM;
1341 case RISCV::QC_BEQI:
1342 return RISCV::QC_BNEI;
1343 case RISCV::QC_BNEI:
1344 return RISCV::QC_BEQI;
1345 case RISCV::QC_BLTI:
1346 return RISCV::QC_BGEI;
1347 case RISCV::QC_BGEI:
1348 return RISCV::QC_BLTI;
1349 case RISCV::QC_BLTUI:
1350 return RISCV::QC_BGEUI;
1351 case RISCV::QC_BGEUI:
1352 return RISCV::QC_BLTUI;
1353 case RISCV::QC_E_BEQI:
1354 return RISCV::QC_E_BNEI;
1355 case RISCV::QC_E_BNEI:
1356 return RISCV::QC_E_BEQI;
1357 case RISCV::QC_E_BLTI:
1358 return RISCV::QC_E_BGEI;
1359 case RISCV::QC_E_BGEI:
1360 return RISCV::QC_E_BLTI;
1361 case RISCV::QC_E_BLTUI:
1362 return RISCV::QC_E_BGEUI;
1363 case RISCV::QC_E_BGEUI:
1364 return RISCV::QC_E_BLTUI;
1365 case RISCV::NDS_BBC:
1366 return RISCV::NDS_BBS;
1367 case RISCV::NDS_BBS:
1368 return RISCV::NDS_BBC;
1369 case RISCV::NDS_BEQC:
1370 return RISCV::NDS_BNEC;
1371 case RISCV::NDS_BNEC:
1372 return RISCV::NDS_BEQC;
1373 }
1374}
1375
1378 MachineBasicBlock *&FBB,
1380 bool AllowModify) const {
1381 TBB = FBB = nullptr;
1382 Cond.clear();
1383
1384 // If the block has no terminators, it just falls into the block after it.
1385 MachineBasicBlock::iterator I = MBB.getLastNonDebugInstr();
1386 if (I == MBB.end() || !isUnpredicatedTerminator(*I))
1387 return false;
1388
1389 // Count the number of terminators and find the first unconditional or
1390 // indirect branch.
1391 MachineBasicBlock::iterator FirstUncondOrIndirectBr = MBB.end();
1392 int NumTerminators = 0;
1393 for (auto J = I.getReverse(); J != MBB.rend() && isUnpredicatedTerminator(*J);
1394 J++) {
1395 NumTerminators++;
1396 if (J->getDesc().isUnconditionalBranch() ||
1397 J->getDesc().isIndirectBranch()) {
1398 FirstUncondOrIndirectBr = J.getReverse();
1399 }
1400 }
1401
1402 // If AllowModify is true, we can erase any terminators after
1403 // FirstUncondOrIndirectBR.
1404 if (AllowModify && FirstUncondOrIndirectBr != MBB.end()) {
1405 while (std::next(FirstUncondOrIndirectBr) != MBB.end()) {
1406 std::next(FirstUncondOrIndirectBr)->eraseFromParent();
1407 NumTerminators--;
1408 }
1409 I = FirstUncondOrIndirectBr;
1410 }
1411
1412 // We can't handle blocks that end in an indirect branch.
1413 if (I->getDesc().isIndirectBranch())
1414 return true;
1415
1416 // We can't handle Generic branch opcodes from Global ISel.
1417 if (I->isPreISelOpcode())
1418 return true;
1419
1420 // We can't handle blocks with more than 2 terminators.
1421 if (NumTerminators > 2)
1422 return true;
1423
1424 // Handle a single unconditional branch.
1425 if (NumTerminators == 1 && I->getDesc().isUnconditionalBranch()) {
1427 return false;
1428 }
1429
1430 // Handle a single conditional branch.
1431 if (NumTerminators == 1 && I->getDesc().isConditionalBranch()) {
1433 return false;
1434 }
1435
1436 // Handle a conditional branch followed by an unconditional branch.
1437 if (NumTerminators == 2 && std::prev(I)->getDesc().isConditionalBranch() &&
1438 I->getDesc().isUnconditionalBranch()) {
1439 parseCondBranch(*std::prev(I), TBB, Cond);
1440 FBB = getBranchDestBlock(*I);
1441 return false;
1442 }
1443
1444 // Otherwise, we can't handle this.
1445 return true;
1446}
1447
1449 int *BytesRemoved) const {
1450 if (BytesRemoved)
1451 *BytesRemoved = 0;
1452 MachineBasicBlock::iterator I = MBB.getLastNonDebugInstr();
1453 if (I == MBB.end())
1454 return 0;
1455
1456 if (!I->getDesc().isUnconditionalBranch() &&
1457 !I->getDesc().isConditionalBranch())
1458 return 0;
1459
1460 // Remove the branch.
1461 if (BytesRemoved)
1462 *BytesRemoved += getInstSizeInBytes(*I);
1463 I->eraseFromParent();
1464
1465 I = MBB.end();
1466
1467 if (I == MBB.begin())
1468 return 1;
1469 --I;
1470 if (!I->getDesc().isConditionalBranch())
1471 return 1;
1472
1473 // Remove the branch.
1474 if (BytesRemoved)
1475 *BytesRemoved += getInstSizeInBytes(*I);
1476 I->eraseFromParent();
1477 return 2;
1478}
1479
1480// Inserts a branch into the end of the specific MachineBasicBlock, returning
1481// the number of instructions inserted.
1484 ArrayRef<MachineOperand> Cond, const DebugLoc &DL, int *BytesAdded) const {
1485 if (BytesAdded)
1486 *BytesAdded = 0;
1487
1488 // Shouldn't be a fall through.
1489 assert(TBB && "insertBranch must not be told to insert a fallthrough");
1490 assert((Cond.size() == 3 || Cond.size() == 0) &&
1491 "RISC-V branch conditions have two components!");
1492
1493 // Unconditional branch.
1494 if (Cond.empty()) {
1495 MachineInstr &MI = *BuildMI(&MBB, DL, get(RISCV::PseudoBR)).addMBB(TBB);
1496 if (BytesAdded)
1497 *BytesAdded += getInstSizeInBytes(MI);
1498 return 1;
1499 }
1500
1501 // Either a one or two-way conditional branch.
1502 MachineInstr &CondMI = *BuildMI(&MBB, DL, get(Cond[0].getImm()))
1503 .add(Cond[1])
1504 .add(Cond[2])
1505 .addMBB(TBB);
1506 if (BytesAdded)
1507 *BytesAdded += getInstSizeInBytes(CondMI);
1508
1509 // One-way conditional branch.
1510 if (!FBB)
1511 return 1;
1512
1513 // Two-way conditional branch.
1514 MachineInstr &MI = *BuildMI(&MBB, DL, get(RISCV::PseudoBR)).addMBB(FBB);
1515 if (BytesAdded)
1516 *BytesAdded += getInstSizeInBytes(MI);
1517 return 2;
1518}
1519
1521 MachineBasicBlock &DestBB,
1522 MachineBasicBlock &RestoreBB,
1523 const DebugLoc &DL, int64_t BrOffset,
1524 RegScavenger *RS) const {
1525 assert(RS && "RegScavenger required for long branching");
1526 assert(MBB.empty() &&
1527 "new block should be inserted for expanding unconditional branch");
1528 assert(MBB.pred_size() == 1);
1529 assert(RestoreBB.empty() &&
1530 "restore block should be inserted for restoring clobbered registers");
1531
1532 MachineFunction *MF = MBB.getParent();
1533 MachineRegisterInfo &MRI = MF->getRegInfo();
1536
1537 if (!isInt<32>(BrOffset))
1539 "Branch offsets outside of the signed 32-bit range not supported");
1540
1541 // FIXME: A virtual register must be used initially, as the register
1542 // scavenger won't work with empty blocks (SIInstrInfo::insertIndirectBranch
1543 // uses the same workaround).
1544 Register ScratchReg = MRI.createVirtualRegister(&RISCV::GPRJALRRegClass);
1545 auto II = MBB.end();
1546 // We may also update the jump target to RestoreBB later.
1547 MachineInstr &MI = *BuildMI(MBB, II, DL, get(RISCV::PseudoJump))
1548 .addReg(ScratchReg, RegState::Define | RegState::Dead)
1549 .addMBB(&DestBB, RISCVII::MO_CALL);
1550
1551 RS->enterBasicBlockEnd(MBB);
1552 // When cf-protection-branch is enabled, we must use t2 (x7) for software
1553 // guarded branches to hold the landing pad label.
1554 bool HasCFBranch =
1555 MF->getInfo<RISCVMachineFunctionInfo>()->hasCFProtectionBranch();
1556 const TargetRegisterClass *RC = &RISCV::GPRRegClass;
1557 if (HasCFBranch)
1558 RC = &RISCV::GPRX7RegClass;
1559 Register TmpGPR =
1560 RS->scavengeRegisterBackwards(*RC, MI.getIterator(),
1561 /*RestoreAfter=*/false, /*SpAdj=*/0,
1562 /*AllowSpill=*/false);
1563 if (TmpGPR.isValid())
1564 RS->setRegUsed(TmpGPR);
1565 else {
1566 // The case when there is no scavenged register needs special handling.
1567
1568 // Pick s11(or s1 for rve) because it doesn't make a difference.
1569 TmpGPR = STI.hasStdExtE() ? RISCV::X9 : RISCV::X27;
1570 // Force t2 if cf-protection-branch is enabled
1571 if (HasCFBranch)
1572 TmpGPR = RISCV::X7;
1573
1574 int FrameIndex = RVFI->getBranchRelaxationScratchFrameIndex();
1575 if (FrameIndex == -1)
1576 report_fatal_error("underestimated function size");
1577
1578 storeRegToStackSlot(MBB, MI, TmpGPR, /*IsKill=*/true, FrameIndex,
1579 &RISCV::GPRRegClass, Register());
1580 TRI->eliminateFrameIndex(std::prev(MI.getIterator()),
1581 /*SpAdj=*/0, /*FIOperandNum=*/1);
1582
1583 MI.getOperand(1).setMBB(&RestoreBB);
1584
1585 loadRegFromStackSlot(RestoreBB, RestoreBB.end(), TmpGPR, FrameIndex,
1586 &RISCV::GPRRegClass, Register());
1587 TRI->eliminateFrameIndex(RestoreBB.back(),
1588 /*SpAdj=*/0, /*FIOperandNum=*/1);
1589 }
1590
1591 MRI.replaceRegWith(ScratchReg, TmpGPR);
1592 MRI.clearVirtRegs();
1593}
1594
1597 assert((Cond.size() == 3) && "Invalid branch condition!");
1598
1600
1601 return false;
1602}
1603
1604// Return true if the instruction is a load immediate instruction (i.e.
1605// (ADDI x0, imm) or (BSETI x0, imm)).
1606static bool isLoadImm(const MachineInstr *MI, int64_t &Imm) {
1607 if (MI->getOpcode() == RISCV::ADDI && MI->getOperand(1).isReg() &&
1608 MI->getOperand(1).getReg() == RISCV::X0) {
1609 Imm = MI->getOperand(2).getImm();
1610 return true;
1611 }
1612 // BSETI can be used to create power of 2 constants. Only 2048 is currently
1613 // interesting because it is 1 more than the maximum ADDI constant.
1614 if (MI->getOpcode() == RISCV::BSETI && MI->getOperand(1).isReg() &&
1615 MI->getOperand(1).getReg() == RISCV::X0 &&
1616 MI->getOperand(2).getImm() == 11) {
1617 Imm = 2048;
1618 return true;
1619 }
1620 return false;
1621}
1622
1624 const MachineOperand &Op, int64_t &Imm) {
1625 // Either a load from immediate instruction or X0.
1626 if (!Op.isReg())
1627 return false;
1628
1629 Register Reg = Op.getReg();
1630 if (Reg == RISCV::X0) {
1631 Imm = 0;
1632 return true;
1633 }
1634
1635 if (!Reg.isVirtual())
1636 return false;
1637
1638 const MachineInstr *DefMI = MRI.getVRegDef(Reg);
1639 return DefMI && isLoadImm(DefMI, Imm);
1640}
1641
1643 bool IsSigned = false;
1644 bool IsEquality = false;
1645 switch (MI.getOpcode()) {
1646 default:
1647 return false;
1648 case RISCV::BEQ:
1649 case RISCV::BNE:
1650 IsEquality = true;
1651 break;
1652 case RISCV::BGE:
1653 case RISCV::BLT:
1654 IsSigned = true;
1655 break;
1656 case RISCV::BGEU:
1657 case RISCV::BLTU:
1658 break;
1659 }
1660
1661 MachineBasicBlock *MBB = MI.getParent();
1662 MachineRegisterInfo &MRI = MBB->getParent()->getRegInfo();
1663
1664 const MachineOperand &LHS = MI.getOperand(0);
1665 const MachineOperand &RHS = MI.getOperand(1);
1666 MachineBasicBlock *TBB = MI.getOperand(2).getMBB();
1667
1668 RISCVCC::CondCode CC = getCondFromBranchOpc(MI.getOpcode());
1670
1671 // Canonicalize conditional branches which can be constant folded into
1672 // beqz or bnez. We can't modify the CFG here.
1673 int64_t C0, C1;
1674 if (isFromLoadImm(MRI, LHS, C0) && isFromLoadImm(MRI, RHS, C1)) {
1675 unsigned NewOpc = evaluateCondBranch(CC, C0, C1) ? RISCV::BEQ : RISCV::BNE;
1676 // Build the new branch and remove the old one.
1677 BuildMI(*MBB, MI, MI.getDebugLoc(), get(NewOpc))
1678 .addReg(RISCV::X0)
1679 .addReg(RISCV::X0)
1680 .addMBB(TBB);
1681 MI.eraseFromParent();
1682 return true;
1683 }
1684
1685 if (IsEquality)
1686 return false;
1687
1688 // For two constants C0 and C1 from
1689 // ```
1690 // li Y, C0
1691 // li Z, C1
1692 // ```
1693 // 1. if C1 = C0 + 1
1694 // we can turn:
1695 // (a) blt Y, X -> bge X, Z
1696 // (b) bge Y, X -> blt X, Z
1697 //
1698 // 2. if C1 = C0 - 1
1699 // we can turn:
1700 // (a) blt X, Y -> bge Z, X
1701 // (b) bge X, Y -> blt Z, X
1702 //
1703 // To make sure this optimization is really beneficial, we only
1704 // optimize for cases where Y had only one use (i.e. only used by the branch).
1705 // Try to find the register for constant Z; return
1706 // invalid register otherwise.
1707 auto searchConst = [&](int64_t C1) -> Register {
1709 auto DefC1 = std::find_if(++II, E, [&](const MachineInstr &I) -> bool {
1710 int64_t Imm;
1711 return isLoadImm(&I, Imm) && Imm == C1 &&
1712 I.getOperand(0).getReg().isVirtual();
1713 });
1714 if (DefC1 != E)
1715 return DefC1->getOperand(0).getReg();
1716
1717 return Register();
1718 };
1719
1720 unsigned NewOpc = RISCVCC::getBrCond(getInverseBranchCondition(CC));
1721
1722 // Might be case 1.
1723 // Don't change 0 to 1 since we can use x0.
1724 // For unsigned cases changing -1U to 0 would be incorrect.
1725 // The incorrect case for signed would be INT_MAX, but isFromLoadImm can't
1726 // return that.
1727 if (isFromLoadImm(MRI, LHS, C0) && C0 != 0 && LHS.getReg().isVirtual() &&
1728 MRI.hasOneUse(LHS.getReg()) && (IsSigned || C0 != -1)) {
1729 assert((isInt<12>(C0) || C0 == 2048) && "Unexpected immediate");
1730 if (Register RegZ = searchConst(C0 + 1)) {
1731 BuildMI(*MBB, MI, MI.getDebugLoc(), get(NewOpc))
1732 .add(RHS)
1733 .addReg(RegZ)
1734 .addMBB(TBB);
1735 // We might extend the live range of Z, clear its kill flag to
1736 // account for this.
1737 MRI.clearKillFlags(RegZ);
1738 MI.eraseFromParent();
1739 return true;
1740 }
1741 }
1742
1743 // Might be case 2.
1744 // For signed cases we don't want to change 0 since we can use x0.
1745 // For unsigned cases changing 0 to -1U would be incorrect.
1746 // The incorrect case for signed would be INT_MIN, but isFromLoadImm can't
1747 // return that.
1748 if (isFromLoadImm(MRI, RHS, C0) && C0 != 0 && RHS.getReg().isVirtual() &&
1749 MRI.hasOneUse(RHS.getReg())) {
1750 assert((isInt<12>(C0) || C0 == 2048) && "Unexpected immediate");
1751 if (Register RegZ = searchConst(C0 - 1)) {
1752 BuildMI(*MBB, MI, MI.getDebugLoc(), get(NewOpc))
1753 .addReg(RegZ)
1754 .add(LHS)
1755 .addMBB(TBB);
1756 // We might extend the live range of Z, clear its kill flag to
1757 // account for this.
1758 MRI.clearKillFlags(RegZ);
1759 MI.eraseFromParent();
1760 return true;
1761 }
1762 }
1763
1764 return false;
1765}
1766
1769 assert(MI.getDesc().isBranch() && "Unexpected opcode!");
1770 // The branch target is always the last operand.
1771 int NumOp = MI.getNumExplicitOperands();
1772 return MI.getOperand(NumOp - 1).getMBB();
1773}
1774
1776 int64_t BrOffset) const {
1777 unsigned XLen = STI.getXLen();
1778 // Ideally we could determine the supported branch offset from the
1779 // RISCVII::FormMask, but this can't be used for Pseudo instructions like
1780 // PseudoBR.
1781 switch (BranchOp) {
1782 default:
1783 llvm_unreachable("Unexpected opcode!");
1784 case RISCV::NDS_BBC:
1785 case RISCV::NDS_BBS:
1786 case RISCV::NDS_BEQC:
1787 case RISCV::NDS_BNEC:
1788 return isInt<11>(BrOffset);
1789 case RISCV::BEQ:
1790 case RISCV::BNE:
1791 case RISCV::BLT:
1792 case RISCV::BGE:
1793 case RISCV::BLTU:
1794 case RISCV::BGEU:
1795 case RISCV::BEQI:
1796 case RISCV::BNEI:
1797 case RISCV::CV_BEQIMM:
1798 case RISCV::CV_BNEIMM:
1799 case RISCV::QC_BEQI:
1800 case RISCV::QC_BNEI:
1801 case RISCV::QC_BGEI:
1802 case RISCV::QC_BLTI:
1803 case RISCV::QC_BLTUI:
1804 case RISCV::QC_BGEUI:
1805 case RISCV::QC_E_BEQI:
1806 case RISCV::QC_E_BNEI:
1807 case RISCV::QC_E_BGEI:
1808 case RISCV::QC_E_BLTI:
1809 case RISCV::QC_E_BLTUI:
1810 case RISCV::QC_E_BGEUI:
1811 return isInt<13>(BrOffset);
1812 case RISCV::JAL:
1813 case RISCV::PseudoBR:
1814 return isInt<21>(BrOffset);
1815 case RISCV::PseudoJump:
1816 return isInt<32>(SignExtend64(BrOffset + 0x800, XLen));
1817 }
1818}
1819
1820// If the operation has a predicated pseudo instruction, return the pseudo
1821// instruction opcode. Otherwise, return RISCV::INSTRUCTION_LIST_END.
1822// TODO: Support more operations.
1823unsigned getPredicatedOpcode(unsigned Opcode) {
1824 // clang-format off
1825 switch (Opcode) {
1826 case RISCV::ADD: return RISCV::PseudoCCADD;
1827 case RISCV::SUB: return RISCV::PseudoCCSUB;
1828 case RISCV::SLL: return RISCV::PseudoCCSLL;
1829 case RISCV::SRL: return RISCV::PseudoCCSRL;
1830 case RISCV::SRA: return RISCV::PseudoCCSRA;
1831 case RISCV::AND: return RISCV::PseudoCCAND;
1832 case RISCV::OR: return RISCV::PseudoCCOR;
1833 case RISCV::XOR: return RISCV::PseudoCCXOR;
1834 case RISCV::MAX: return RISCV::PseudoCCMAX;
1835 case RISCV::MAXU: return RISCV::PseudoCCMAXU;
1836 case RISCV::MIN: return RISCV::PseudoCCMIN;
1837 case RISCV::MINU: return RISCV::PseudoCCMINU;
1838 case RISCV::MUL: return RISCV::PseudoCCMUL;
1839 case RISCV::LUI: return RISCV::PseudoCCLUI;
1840 case RISCV::QC_LI: return RISCV::PseudoCCQC_LI;
1841 case RISCV::QC_E_LI: return RISCV::PseudoCCQC_E_LI;
1842
1843 case RISCV::ADDI: return RISCV::PseudoCCADDI;
1844 case RISCV::SLLI: return RISCV::PseudoCCSLLI;
1845 case RISCV::SRLI: return RISCV::PseudoCCSRLI;
1846 case RISCV::SRAI: return RISCV::PseudoCCSRAI;
1847 case RISCV::ANDI: return RISCV::PseudoCCANDI;
1848 case RISCV::ORI: return RISCV::PseudoCCORI;
1849 case RISCV::XORI: return RISCV::PseudoCCXORI;
1850
1851 case RISCV::ADDW: return RISCV::PseudoCCADDW;
1852 case RISCV::SUBW: return RISCV::PseudoCCSUBW;
1853 case RISCV::SLLW: return RISCV::PseudoCCSLLW;
1854 case RISCV::SRLW: return RISCV::PseudoCCSRLW;
1855 case RISCV::SRAW: return RISCV::PseudoCCSRAW;
1856
1857 case RISCV::ADDIW: return RISCV::PseudoCCADDIW;
1858 case RISCV::SLLIW: return RISCV::PseudoCCSLLIW;
1859 case RISCV::SRLIW: return RISCV::PseudoCCSRLIW;
1860 case RISCV::SRAIW: return RISCV::PseudoCCSRAIW;
1861
1862 case RISCV::ANDN: return RISCV::PseudoCCANDN;
1863 case RISCV::ORN: return RISCV::PseudoCCORN;
1864 case RISCV::XNOR: return RISCV::PseudoCCXNOR;
1865
1866 case RISCV::NDS_BFOS: return RISCV::PseudoCCNDS_BFOS;
1867 case RISCV::NDS_BFOZ: return RISCV::PseudoCCNDS_BFOZ;
1868 }
1869 // clang-format on
1870
1871 return RISCV::INSTRUCTION_LIST_END;
1872}
1873
1874/// Identify instructions that can be folded into a CCMOV instruction, and
1875/// return the defining instruction.
1877 const MachineRegisterInfo &MRI,
1878 const TargetInstrInfo *TII,
1879 const RISCVSubtarget &STI) {
1880 if (!Reg.isVirtual())
1881 return nullptr;
1882 if (!MRI.hasOneNonDBGUse(Reg))
1883 return nullptr;
1884 MachineInstr *MI = MRI.getVRegDef(Reg);
1885 if (!MI)
1886 return nullptr;
1887
1888 if (!STI.hasShortForwardBranchIMinMax() &&
1889 (MI->getOpcode() == RISCV::MAX || MI->getOpcode() == RISCV::MIN ||
1890 MI->getOpcode() == RISCV::MINU || MI->getOpcode() == RISCV::MAXU))
1891 return nullptr;
1892
1893 if (!STI.hasShortForwardBranchIMul() && MI->getOpcode() == RISCV::MUL)
1894 return nullptr;
1895
1896 // Check if MI can be predicated and folded into the CCMOV.
1897 if (getPredicatedOpcode(MI->getOpcode()) == RISCV::INSTRUCTION_LIST_END)
1898 return nullptr;
1899 // Don't predicate li idiom.
1900 if (MI->getOpcode() == RISCV::ADDI && MI->getOperand(1).isReg() &&
1901 MI->getOperand(1).getReg() == RISCV::X0)
1902 return nullptr;
1903 // Check if MI has any other defs or physreg uses.
1904 for (const MachineOperand &MO : llvm::drop_begin(MI->operands())) {
1905 // Reject frame index operands, PEI can't handle the predicated pseudos.
1906 if (MO.isFI() || MO.isCPI() || MO.isJTI())
1907 return nullptr;
1908 if (!MO.isReg())
1909 continue;
1910 // MI can't have any tied operands, that would conflict with predication.
1911 if (MO.isTied())
1912 return nullptr;
1913 if (MO.isDef())
1914 return nullptr;
1915 // Allow constant physregs.
1916 if (MO.getReg().isPhysical() && !MRI.isConstantPhysReg(MO.getReg()))
1917 return nullptr;
1918 }
1919 bool DontMoveAcrossStores = true;
1920 if (!MI->isSafeToMove(DontMoveAcrossStores))
1921 return nullptr;
1922 return MI;
1923}
1924
1928 bool PreferFalse) const {
1929 assert(MI.getOpcode() == RISCV::PseudoCCMOVGPR &&
1930 "Unknown select instruction");
1931 if (!STI.hasShortForwardBranchIALU())
1932 return nullptr;
1933
1934 MachineRegisterInfo &MRI = MI.getParent()->getParent()->getRegInfo();
1936 canFoldAsPredicatedOp(MI.getOperand(2).getReg(), MRI, this, STI);
1937 bool Invert = !DefMI;
1938 if (!DefMI)
1939 DefMI = canFoldAsPredicatedOp(MI.getOperand(1).getReg(), MRI, this, STI);
1940 if (!DefMI)
1941 return nullptr;
1942
1943 // Find new register class to use.
1944 MachineOperand FalseReg = MI.getOperand(Invert ? 2 : 1);
1945 Register DestReg = MI.getOperand(0).getReg();
1946 const TargetRegisterClass *PreviousClass = MRI.getRegClass(FalseReg.getReg());
1947 if (!MRI.constrainRegClass(DestReg, PreviousClass))
1948 return nullptr;
1949
1950 unsigned PredOpc = getPredicatedOpcode(DefMI->getOpcode());
1951 assert(PredOpc != RISCV::INSTRUCTION_LIST_END && "Unexpected opcode!");
1952
1953 // Create a new predicated version of DefMI.
1954 MachineInstrBuilder NewMI =
1955 BuildMI(*MI.getParent(), MI, MI.getDebugLoc(), get(PredOpc), DestReg);
1956
1957 // Copy the false register.
1958 NewMI.add(FalseReg);
1959
1960 // Copy all the DefMI operands.
1961 const MCInstrDesc &DefDesc = DefMI->getDesc();
1962 for (unsigned i = 1, e = DefDesc.getNumOperands(); i != e; ++i)
1963 NewMI.add(DefMI->getOperand(i));
1964
1965 // Add branch opcode, inverting if necessary.
1966 unsigned BCCOpcode = MI.getOperand(MI.getNumExplicitOperands() - 3).getImm();
1967 if (Invert)
1968 BCCOpcode = RISCVCC::getInverseBranchOpcode(BCCOpcode);
1969 NewMI.addImm(BCCOpcode);
1970
1971 // Copy the condition portion.
1972 NewMI.add(MI.getOperand(MI.getNumExplicitOperands() - 2));
1973 NewMI.add(MI.getOperand(MI.getNumExplicitOperands() - 1));
1974
1975 // Update SeenMIs set: register newly created MI and erase removed DefMI.
1976 SeenMIs.insert(NewMI);
1977 SeenMIs.erase(DefMI);
1978
1979 // If MI is inside a loop, and DefMI is outside the loop, then kill flags on
1980 // DefMI would be invalid when transferred inside the loop. Checking for a
1981 // loop is expensive, but at least remove kill flags if they are in different
1982 // BBs.
1983 if (DefMI->getParent() != MI.getParent())
1984 NewMI->clearKillInfo();
1985
1986 // The caller will erase MI, but not DefMI.
1987 DefMI->eraseFromParent();
1988 return NewMI;
1989}
1990
1992 if (MI.isMetaInstruction())
1993 return 0;
1994
1995 unsigned Opcode = MI.getOpcode();
1996
1997 if (Opcode == TargetOpcode::INLINEASM ||
1998 Opcode == TargetOpcode::INLINEASM_BR) {
1999 const MachineFunction &MF = *MI.getParent()->getParent();
2000 return getInlineAsmLength(MI.getOperand(0).getSymbolName(),
2001 MF.getTarget().getMCAsmInfo());
2002 }
2003
2004 if (requiresNTLHint(MI)) {
2005 if (STI.hasStdExtZca()) {
2006 if (isCompressibleInst(MI, STI))
2007 return 4; // c.ntl.all + c.load/c.store
2008 return 6; // c.ntl.all + load/store
2009 }
2010 return 8; // ntl.all + load/store
2011 }
2012
2013 if (Opcode == TargetOpcode::BUNDLE)
2014 return getInstBundleSize(MI);
2015
2016 if (MI.getParent() && MI.getParent()->getParent()) {
2017 if (isCompressibleInst(MI, STI))
2018 return 2;
2019 }
2020
2021 switch (Opcode) {
2022 case RISCV::PseudoMV_FPR16INX:
2023 case RISCV::PseudoMV_FPR32INX:
2024 case RISCV::PseudoClearGPR:
2025 // MV is always compressible to either c.mv or c.li rd, 0.
2026 return STI.hasStdExtZca() ? 2 : 4;
2027 // Below cases are for short forward branch pseudos
2028 case RISCV::PseudoCCMOVGPRNoX0:
2029 return get(MI.getOperand(MI.getNumExplicitOperands() - 3).getImm())
2030 .getSize() +
2031 2;
2032 case RISCV::PseudoCCMOVGPR:
2033 case RISCV::PseudoCCADD:
2034 case RISCV::PseudoCCSUB:
2035 case RISCV::PseudoCCSLL:
2036 case RISCV::PseudoCCSRL:
2037 case RISCV::PseudoCCSRA:
2038 case RISCV::PseudoCCAND:
2039 case RISCV::PseudoCCOR:
2040 case RISCV::PseudoCCXOR:
2041 case RISCV::PseudoCCADDI:
2042 case RISCV::PseudoCCANDI:
2043 case RISCV::PseudoCCORI:
2044 case RISCV::PseudoCCXORI:
2045 case RISCV::PseudoCCLUI:
2046 case RISCV::PseudoCCSLLI:
2047 case RISCV::PseudoCCSRLI:
2048 case RISCV::PseudoCCSRAI:
2049 case RISCV::PseudoCCADDW:
2050 case RISCV::PseudoCCSUBW:
2051 case RISCV::PseudoCCSLLW:
2052 case RISCV::PseudoCCSRLW:
2053 case RISCV::PseudoCCSRAW:
2054 case RISCV::PseudoCCADDIW:
2055 case RISCV::PseudoCCSLLIW:
2056 case RISCV::PseudoCCSRLIW:
2057 case RISCV::PseudoCCSRAIW:
2058 case RISCV::PseudoCCANDN:
2059 case RISCV::PseudoCCORN:
2060 case RISCV::PseudoCCXNOR:
2061 case RISCV::PseudoCCMAX:
2062 case RISCV::PseudoCCMIN:
2063 case RISCV::PseudoCCMAXU:
2064 case RISCV::PseudoCCMINU:
2065 case RISCV::PseudoCCMUL:
2066 case RISCV::PseudoCCLB:
2067 case RISCV::PseudoCCLH:
2068 case RISCV::PseudoCCLW:
2069 case RISCV::PseudoCCLHU:
2070 case RISCV::PseudoCCLBU:
2071 case RISCV::PseudoCCLWU:
2072 case RISCV::PseudoCCLD:
2073 case RISCV::PseudoCCQC_LI:
2074 return get(MI.getOperand(MI.getNumExplicitOperands() - 3).getImm())
2075 .getSize() +
2076 4;
2077 case RISCV::PseudoCCQC_E_LI:
2078 case RISCV::PseudoCCQC_E_LB:
2079 case RISCV::PseudoCCQC_E_LH:
2080 case RISCV::PseudoCCQC_E_LW:
2081 case RISCV::PseudoCCQC_E_LHU:
2082 case RISCV::PseudoCCQC_E_LBU:
2083 return get(MI.getOperand(MI.getNumExplicitOperands() - 3).getImm())
2084 .getSize() +
2085 6;
2086 case TargetOpcode::STACKMAP:
2087 // The upper bound for a stackmap intrinsic is the full length of its shadow
2089 case TargetOpcode::PATCHPOINT:
2090 // The size of the patchpoint intrinsic is the number of bytes requested
2092 case TargetOpcode::STATEPOINT: {
2093 // The size of the statepoint intrinsic is the number of bytes requested
2094 unsigned NumBytes = StatepointOpers(&MI).getNumPatchBytes();
2095 // No patch bytes means at most a PseudoCall is emitted
2096 return std::max(NumBytes, 8U);
2097 }
2098 case TargetOpcode::PATCHABLE_FUNCTION_ENTER:
2099 case TargetOpcode::PATCHABLE_FUNCTION_EXIT:
2100 case TargetOpcode::PATCHABLE_TAIL_CALL: {
2101 const MachineFunction &MF = *MI.getParent()->getParent();
2102 const Function &F = MF.getFunction();
2103 if (Opcode == TargetOpcode::PATCHABLE_FUNCTION_ENTER &&
2104 F.hasFnAttribute("patchable-function-entry")) {
2105 unsigned Num =
2106 F.getFnAttributeAsParsedInteger("patchable-function-entry");
2107 // Number of C.NOP or NOP
2108 return (STI.hasStdExtZca() ? 2 : 4) * Num;
2109 }
2110 // XRay uses C.JAL + 21 or 33 C.NOP for each sled in RV32 and RV64,
2111 // respectively.
2112 return STI.is64Bit() ? 68 : 44;
2113 }
2114 default:
2115 return get(Opcode).getSize();
2116 }
2117}
2118
2120 const unsigned Opcode = MI.getOpcode();
2121 switch (Opcode) {
2122 default:
2123 break;
2124 case RISCV::FSGNJ_D:
2125 case RISCV::FSGNJ_S:
2126 case RISCV::FSGNJ_H:
2127 case RISCV::FSGNJ_D_INX:
2128 case RISCV::FSGNJ_D_IN32X:
2129 case RISCV::FSGNJ_S_INX:
2130 case RISCV::FSGNJ_H_INX:
2131 // The canonical floating-point move is fsgnj rd, rs, rs.
2132 return MI.getOperand(1).isReg() && MI.getOperand(2).isReg() &&
2133 MI.getOperand(1).getReg() == MI.getOperand(2).getReg();
2134 case RISCV::ADDI:
2135 case RISCV::ORI:
2136 case RISCV::XORI:
2137 return (MI.getOperand(1).isReg() &&
2138 MI.getOperand(1).getReg() == RISCV::X0) ||
2139 (MI.getOperand(2).isImm() && MI.getOperand(2).getImm() == 0);
2140 }
2141 return MI.isAsCheapAsAMove();
2142}
2143
2144std::optional<DestSourcePair>
2146 if (MI.isMoveReg())
2147 return DestSourcePair{MI.getOperand(0), MI.getOperand(1)};
2148 switch (MI.getOpcode()) {
2149 default:
2150 break;
2151 case RISCV::ADD:
2152 case RISCV::OR:
2153 case RISCV::XOR:
2154 if (MI.getOperand(1).isReg() && MI.getOperand(1).getReg() == RISCV::X0 &&
2155 MI.getOperand(2).isReg())
2156 return DestSourcePair{MI.getOperand(0), MI.getOperand(2)};
2157 if (MI.getOperand(2).isReg() && MI.getOperand(2).getReg() == RISCV::X0 &&
2158 MI.getOperand(1).isReg())
2159 return DestSourcePair{MI.getOperand(0), MI.getOperand(1)};
2160 break;
2161 case RISCV::ADDI:
2162 // Operand 1 can be a frameindex but callers expect registers
2163 if (MI.getOperand(1).isReg() && MI.getOperand(2).isImm() &&
2164 MI.getOperand(2).getImm() == 0)
2165 return DestSourcePair{MI.getOperand(0), MI.getOperand(1)};
2166 break;
2167 case RISCV::SUB:
2168 if (MI.getOperand(2).isReg() && MI.getOperand(2).getReg() == RISCV::X0 &&
2169 MI.getOperand(1).isReg())
2170 return DestSourcePair{MI.getOperand(0), MI.getOperand(1)};
2171 break;
2172 case RISCV::SH1ADD:
2173 case RISCV::SH1ADD_UW:
2174 case RISCV::SH2ADD:
2175 case RISCV::SH2ADD_UW:
2176 case RISCV::SH3ADD:
2177 case RISCV::SH3ADD_UW:
2178 if (MI.getOperand(1).isReg() && MI.getOperand(1).getReg() == RISCV::X0 &&
2179 MI.getOperand(2).isReg())
2180 return DestSourcePair{MI.getOperand(0), MI.getOperand(2)};
2181 break;
2182 case RISCV::FSGNJ_D:
2183 case RISCV::FSGNJ_S:
2184 case RISCV::FSGNJ_H:
2185 case RISCV::FSGNJ_D_INX:
2186 case RISCV::FSGNJ_D_IN32X:
2187 case RISCV::FSGNJ_S_INX:
2188 case RISCV::FSGNJ_H_INX:
2189 // The canonical floating-point move is fsgnj rd, rs, rs.
2190 if (MI.getOperand(1).isReg() && MI.getOperand(2).isReg() &&
2191 MI.getOperand(1).getReg() == MI.getOperand(2).getReg())
2192 return DestSourcePair{MI.getOperand(0), MI.getOperand(1)};
2193 break;
2194 }
2195 return std::nullopt;
2196}
2197
2199 if (ForceMachineCombinerStrategy.getNumOccurrences() == 0) {
2200 // The option is unused. Choose Local strategy only for in-order cores. When
2201 // scheduling model is unspecified, use MinInstrCount strategy as more
2202 // generic one.
2203 const auto &SchedModel = STI.getSchedModel();
2204 return (!SchedModel.hasInstrSchedModel() || SchedModel.isOutOfOrder())
2207 }
2208 // The strategy was forced by the option.
2210}
2211
2213 MachineInstr &Root, unsigned &Pattern,
2214 SmallVectorImpl<MachineInstr *> &InsInstrs) const {
2215 int16_t FrmOpIdx =
2216 RISCV::getNamedOperandIdx(Root.getOpcode(), RISCV::OpName::frm);
2217 if (FrmOpIdx < 0) {
2218 assert(all_of(InsInstrs,
2219 [](MachineInstr *MI) {
2220 return RISCV::getNamedOperandIdx(MI->getOpcode(),
2221 RISCV::OpName::frm) < 0;
2222 }) &&
2223 "New instructions require FRM whereas the old one does not have it");
2224 return;
2225 }
2226
2227 const MachineOperand &FRM = Root.getOperand(FrmOpIdx);
2228 MachineFunction &MF = *Root.getMF();
2229
2230 for (auto *NewMI : InsInstrs) {
2231 // We'd already added the FRM operand.
2232 if (static_cast<unsigned>(RISCV::getNamedOperandIdx(
2233 NewMI->getOpcode(), RISCV::OpName::frm)) != NewMI->getNumOperands())
2234 continue;
2235 MachineInstrBuilder MIB(MF, NewMI);
2236 MIB.add(FRM);
2237 if (FRM.getImm() == RISCVFPRndMode::DYN)
2238 MIB.addUse(RISCV::FRM, RegState::Implicit);
2239 }
2240}
2241
2242static bool isFADD(unsigned Opc) {
2243 switch (Opc) {
2244 default:
2245 return false;
2246 case RISCV::FADD_H:
2247 case RISCV::FADD_S:
2248 case RISCV::FADD_D:
2249 return true;
2250 }
2251}
2252
2253static bool isFSUB(unsigned Opc) {
2254 switch (Opc) {
2255 default:
2256 return false;
2257 case RISCV::FSUB_H:
2258 case RISCV::FSUB_S:
2259 case RISCV::FSUB_D:
2260 return true;
2261 }
2262}
2263
2264static bool isFMUL(unsigned Opc) {
2265 switch (Opc) {
2266 default:
2267 return false;
2268 case RISCV::FMUL_H:
2269 case RISCV::FMUL_S:
2270 case RISCV::FMUL_D:
2271 return true;
2272 }
2273}
2274
2275bool RISCVInstrInfo::isVectorAssociativeAndCommutative(const MachineInstr &Inst,
2276 bool Invert) const {
2277#define OPCODE_LMUL_CASE(OPC) \
2278 case RISCV::OPC##_M1: \
2279 case RISCV::OPC##_M2: \
2280 case RISCV::OPC##_M4: \
2281 case RISCV::OPC##_M8: \
2282 case RISCV::OPC##_MF2: \
2283 case RISCV::OPC##_MF4: \
2284 case RISCV::OPC##_MF8
2285
2286#define OPCODE_LMUL_MASK_CASE(OPC) \
2287 case RISCV::OPC##_M1_MASK: \
2288 case RISCV::OPC##_M2_MASK: \
2289 case RISCV::OPC##_M4_MASK: \
2290 case RISCV::OPC##_M8_MASK: \
2291 case RISCV::OPC##_MF2_MASK: \
2292 case RISCV::OPC##_MF4_MASK: \
2293 case RISCV::OPC##_MF8_MASK
2294
2295 unsigned Opcode = Inst.getOpcode();
2296 if (Invert) {
2297 if (auto InvOpcode = getInverseOpcode(Opcode))
2298 Opcode = *InvOpcode;
2299 else
2300 return false;
2301 }
2302
2303 // clang-format off
2304 switch (Opcode) {
2305 default:
2306 return false;
2307 OPCODE_LMUL_CASE(PseudoVADD_VV):
2308 OPCODE_LMUL_MASK_CASE(PseudoVADD_VV):
2309 OPCODE_LMUL_CASE(PseudoVMUL_VV):
2310 OPCODE_LMUL_MASK_CASE(PseudoVMUL_VV):
2311 return true;
2312 }
2313 // clang-format on
2314
2315#undef OPCODE_LMUL_MASK_CASE
2316#undef OPCODE_LMUL_CASE
2317}
2318
2319bool RISCVInstrInfo::areRVVInstsReassociable(const MachineInstr &Root,
2320 const MachineInstr &Prev) const {
2321 if (!areOpcodesEqualOrInverse(Root.getOpcode(), Prev.getOpcode()))
2322 return false;
2323
2324 assert(Root.getMF() == Prev.getMF());
2325 const MachineRegisterInfo *MRI = &Root.getMF()->getRegInfo();
2326 const TargetRegisterInfo *TRI = MRI->getTargetRegisterInfo();
2327
2328 // Make sure vtype operands are also the same.
2329 const MCInstrDesc &Desc = get(Root.getOpcode());
2330 const uint64_t TSFlags = Desc.TSFlags;
2331
2332 auto checkImmOperand = [&](unsigned OpIdx) {
2333 return Root.getOperand(OpIdx).getImm() == Prev.getOperand(OpIdx).getImm();
2334 };
2335
2336 auto checkRegOperand = [&](unsigned OpIdx) {
2337 return Root.getOperand(OpIdx).getReg() == Prev.getOperand(OpIdx).getReg();
2338 };
2339
2340 // PassThru
2341 // TODO: Potentially we can loosen the condition to consider Root to be
2342 // associable with Prev if Root has NoReg as passthru. In which case we
2343 // also need to loosen the condition on vector policies between these.
2344 if (!checkRegOperand(1))
2345 return false;
2346
2347 // SEW
2348 if (RISCVII::hasSEWOp(TSFlags) &&
2349 !checkImmOperand(RISCVII::getSEWOpNum(Desc)))
2350 return false;
2351
2352 // Mask
2353 if (RISCVII::usesMaskPolicy(TSFlags)) {
2354 const MachineBasicBlock *MBB = Root.getParent();
2357 Register MI1VReg;
2358
2359 bool SeenMI2 = false;
2360 for (auto End = MBB->rend(), It = It1; It != End; ++It) {
2361 if (It == It2) {
2362 SeenMI2 = true;
2363 if (!MI1VReg.isValid())
2364 // There is no V0 def between Root and Prev; they're sharing the
2365 // same V0.
2366 break;
2367 }
2368
2369 if (It->modifiesRegister(RISCV::V0, TRI)) {
2370 Register SrcReg = It->getOperand(1).getReg();
2371 // If it's not VReg it'll be more difficult to track its defs, so
2372 // bailing out here just to be safe.
2373 if (!SrcReg.isVirtual())
2374 return false;
2375
2376 if (!MI1VReg.isValid()) {
2377 // This is the V0 def for Root.
2378 MI1VReg = SrcReg;
2379 continue;
2380 }
2381
2382 // Some random mask updates.
2383 if (!SeenMI2)
2384 continue;
2385
2386 // This is the V0 def for Prev; check if it's the same as that of
2387 // Root.
2388 if (MI1VReg != SrcReg)
2389 return false;
2390 else
2391 break;
2392 }
2393 }
2394
2395 // If we haven't encountered Prev, it's likely that this function was
2396 // called in a wrong way (e.g. Root is before Prev).
2397 assert(SeenMI2 && "Prev is expected to appear before Root");
2398 }
2399
2400 // Tail / Mask policies
2401 if (RISCVII::hasVecPolicyOp(TSFlags) &&
2402 !checkImmOperand(RISCVII::getVecPolicyOpNum(Desc)))
2403 return false;
2404
2405 // VL
2406 if (RISCVII::hasVLOp(TSFlags)) {
2407 unsigned OpIdx = RISCVII::getVLOpNum(Desc);
2408 const MachineOperand &Op1 = Root.getOperand(OpIdx);
2409 const MachineOperand &Op2 = Prev.getOperand(OpIdx);
2410 if (Op1.getType() != Op2.getType())
2411 return false;
2412 switch (Op1.getType()) {
2414 if (Op1.getReg() != Op2.getReg())
2415 return false;
2416 break;
2418 if (Op1.getImm() != Op2.getImm())
2419 return false;
2420 break;
2421 default:
2422 llvm_unreachable("Unrecognized VL operand type");
2423 }
2424 }
2425
2426 // Rounding modes
2427 if (int Idx = RISCVII::getFRMOpNum(Desc); Idx >= 0 && !checkImmOperand(Idx))
2428 return false;
2429 if (int Idx = RISCVII::getVXRMOpNum(Desc); Idx >= 0 && !checkImmOperand(Idx))
2430 return false;
2431
2432 return true;
2433}
2434
2435// Most of our RVV pseudos have passthru operand, so the real operands
2436// start from index = 2.
2437bool RISCVInstrInfo::hasReassociableVectorSibling(const MachineInstr &Inst,
2438 bool &Commuted) const {
2439 const MachineBasicBlock *MBB = Inst.getParent();
2440 const MachineRegisterInfo &MRI = MBB->getParent()->getRegInfo();
2442 "Expect the present of passthrough operand.");
2443 MachineInstr *MI1 = MRI.getUniqueVRegDef(Inst.getOperand(2).getReg());
2444 MachineInstr *MI2 = MRI.getUniqueVRegDef(Inst.getOperand(3).getReg());
2445
2446 // If only one operand has the same or inverse opcode and it's the second
2447 // source operand, the operands must be commuted.
2448 Commuted = !areRVVInstsReassociable(Inst, *MI1) &&
2449 areRVVInstsReassociable(Inst, *MI2);
2450 if (Commuted)
2451 std::swap(MI1, MI2);
2452
2453 return areRVVInstsReassociable(Inst, *MI1) &&
2454 (isVectorAssociativeAndCommutative(*MI1) ||
2455 isVectorAssociativeAndCommutative(*MI1, /* Invert */ true)) &&
2457 MRI.hasOneNonDBGUse(MI1->getOperand(0).getReg());
2458}
2459
2461 const MachineInstr &Inst, const MachineBasicBlock *MBB) const {
2462 if (!isVectorAssociativeAndCommutative(Inst) &&
2463 !isVectorAssociativeAndCommutative(Inst, /*Invert=*/true))
2465
2466 const MachineOperand &Op1 = Inst.getOperand(2);
2467 const MachineOperand &Op2 = Inst.getOperand(3);
2468 const MachineRegisterInfo &MRI = MBB->getParent()->getRegInfo();
2469
2470 // We need virtual register definitions for the operands that we will
2471 // reassociate.
2472 MachineInstr *MI1 = nullptr;
2473 MachineInstr *MI2 = nullptr;
2474 if (Op1.isReg() && Op1.getReg().isVirtual())
2475 MI1 = MRI.getUniqueVRegDef(Op1.getReg());
2476 if (Op2.isReg() && Op2.getReg().isVirtual())
2477 MI2 = MRI.getUniqueVRegDef(Op2.getReg());
2478
2479 // And at least one operand must be defined in MBB.
2480 return MI1 && MI2 && (MI1->getParent() == MBB || MI2->getParent() == MBB);
2481}
2482
2484 const MachineInstr &Root, unsigned Pattern,
2485 std::array<unsigned, 5> &OperandIndices) const {
2487 if (RISCV::getRVVMCOpcode(Root.getOpcode())) {
2488 // Skip the passthrough operand, so increment all indices by one.
2489 for (unsigned I = 0; I < 5; ++I)
2490 ++OperandIndices[I];
2491 }
2492}
2493
2495 bool &Commuted) const {
2496 if (isVectorAssociativeAndCommutative(Inst) ||
2497 isVectorAssociativeAndCommutative(Inst, /*Invert=*/true))
2498 return hasReassociableVectorSibling(Inst, Commuted);
2499
2500 if (!TargetInstrInfo::hasReassociableSibling(Inst, Commuted))
2501 return false;
2502
2503 const MachineRegisterInfo &MRI = Inst.getMF()->getRegInfo();
2504 unsigned OperandIdx = Commuted ? 2 : 1;
2505 const MachineInstr &Sibling =
2506 *MRI.getVRegDef(Inst.getOperand(OperandIdx).getReg());
2507
2508 int16_t InstFrmOpIdx =
2509 RISCV::getNamedOperandIdx(Inst.getOpcode(), RISCV::OpName::frm);
2510 int16_t SiblingFrmOpIdx =
2511 RISCV::getNamedOperandIdx(Sibling.getOpcode(), RISCV::OpName::frm);
2512
2513 return (InstFrmOpIdx < 0 && SiblingFrmOpIdx < 0) ||
2514 RISCV::hasEqualFRM(Inst, Sibling);
2515}
2516
2518 bool Invert) const {
2519 if (isVectorAssociativeAndCommutative(Inst, Invert))
2520 return true;
2521
2522 unsigned Opc = Inst.getOpcode();
2523 if (Invert) {
2524 auto InverseOpcode = getInverseOpcode(Opc);
2525 if (!InverseOpcode)
2526 return false;
2527 Opc = *InverseOpcode;
2528 }
2529
2530 if (isFADD(Opc) || isFMUL(Opc))
2533
2534 switch (Opc) {
2535 default:
2536 return false;
2537 case RISCV::ADD:
2538 case RISCV::ADDW:
2539 case RISCV::AND:
2540 case RISCV::OR:
2541 case RISCV::XOR:
2542 // From RISC-V ISA spec, if both the high and low bits of the same product
2543 // are required, then the recommended code sequence is:
2544 //
2545 // MULH[[S]U] rdh, rs1, rs2
2546 // MUL rdl, rs1, rs2
2547 // (source register specifiers must be in same order and rdh cannot be the
2548 // same as rs1 or rs2)
2549 //
2550 // Microarchitectures can then fuse these into a single multiply operation
2551 // instead of performing two separate multiplies.
2552 // MachineCombiner may reassociate MUL operands and lose the fusion
2553 // opportunity.
2554 case RISCV::MUL:
2555 case RISCV::MULW:
2556 case RISCV::MIN:
2557 case RISCV::MINU:
2558 case RISCV::MAX:
2559 case RISCV::MAXU:
2560 case RISCV::FMIN_H:
2561 case RISCV::FMIN_S:
2562 case RISCV::FMIN_D:
2563 case RISCV::FMAX_H:
2564 case RISCV::FMAX_S:
2565 case RISCV::FMAX_D:
2566 return true;
2567 }
2568
2569 return false;
2570}
2571
2572std::optional<unsigned>
2573RISCVInstrInfo::getInverseOpcode(unsigned Opcode) const {
2574#define RVV_OPC_LMUL_CASE(OPC, INV) \
2575 case RISCV::OPC##_M1: \
2576 return RISCV::INV##_M1; \
2577 case RISCV::OPC##_M2: \
2578 return RISCV::INV##_M2; \
2579 case RISCV::OPC##_M4: \
2580 return RISCV::INV##_M4; \
2581 case RISCV::OPC##_M8: \
2582 return RISCV::INV##_M8; \
2583 case RISCV::OPC##_MF2: \
2584 return RISCV::INV##_MF2; \
2585 case RISCV::OPC##_MF4: \
2586 return RISCV::INV##_MF4; \
2587 case RISCV::OPC##_MF8: \
2588 return RISCV::INV##_MF8
2589
2590#define RVV_OPC_LMUL_MASK_CASE(OPC, INV) \
2591 case RISCV::OPC##_M1_MASK: \
2592 return RISCV::INV##_M1_MASK; \
2593 case RISCV::OPC##_M2_MASK: \
2594 return RISCV::INV##_M2_MASK; \
2595 case RISCV::OPC##_M4_MASK: \
2596 return RISCV::INV##_M4_MASK; \
2597 case RISCV::OPC##_M8_MASK: \
2598 return RISCV::INV##_M8_MASK; \
2599 case RISCV::OPC##_MF2_MASK: \
2600 return RISCV::INV##_MF2_MASK; \
2601 case RISCV::OPC##_MF4_MASK: \
2602 return RISCV::INV##_MF4_MASK; \
2603 case RISCV::OPC##_MF8_MASK: \
2604 return RISCV::INV##_MF8_MASK
2605
2606 switch (Opcode) {
2607 default:
2608 return std::nullopt;
2609 case RISCV::FADD_H:
2610 return RISCV::FSUB_H;
2611 case RISCV::FADD_S:
2612 return RISCV::FSUB_S;
2613 case RISCV::FADD_D:
2614 return RISCV::FSUB_D;
2615 case RISCV::FSUB_H:
2616 return RISCV::FADD_H;
2617 case RISCV::FSUB_S:
2618 return RISCV::FADD_S;
2619 case RISCV::FSUB_D:
2620 return RISCV::FADD_D;
2621 case RISCV::ADD:
2622 return RISCV::SUB;
2623 case RISCV::SUB:
2624 return RISCV::ADD;
2625 case RISCV::ADDW:
2626 return RISCV::SUBW;
2627 case RISCV::SUBW:
2628 return RISCV::ADDW;
2629 // clang-format off
2630 RVV_OPC_LMUL_CASE(PseudoVADD_VV, PseudoVSUB_VV);
2631 RVV_OPC_LMUL_MASK_CASE(PseudoVADD_VV, PseudoVSUB_VV);
2632 RVV_OPC_LMUL_CASE(PseudoVSUB_VV, PseudoVADD_VV);
2633 RVV_OPC_LMUL_MASK_CASE(PseudoVSUB_VV, PseudoVADD_VV);
2634 // clang-format on
2635 }
2636
2637#undef RVV_OPC_LMUL_MASK_CASE
2638#undef RVV_OPC_LMUL_CASE
2639}
2640
2642 const MachineOperand &MO,
2643 bool DoRegPressureReduce) {
2644 if (!MO.isReg() || !MO.getReg().isVirtual())
2645 return false;
2646 const MachineRegisterInfo &MRI = Root.getMF()->getRegInfo();
2647 MachineInstr *MI = MRI.getVRegDef(MO.getReg());
2648 if (!MI || !isFMUL(MI->getOpcode()))
2649 return false;
2650
2653 return false;
2654
2655 // Try combining even if fmul has more than one use as it eliminates
2656 // dependency between fadd(fsub) and fmul. However, it can extend liveranges
2657 // for fmul operands, so reject the transformation in register pressure
2658 // reduction mode.
2659 if (DoRegPressureReduce && !MRI.hasOneNonDBGUse(MI->getOperand(0).getReg()))
2660 return false;
2661
2662 // Do not combine instructions from different basic blocks.
2663 if (Root.getParent() != MI->getParent())
2664 return false;
2665 return RISCV::hasEqualFRM(Root, *MI);
2666}
2667
2669 SmallVectorImpl<unsigned> &Patterns,
2670 bool DoRegPressureReduce) {
2671 unsigned Opc = Root.getOpcode();
2672 bool IsFAdd = isFADD(Opc);
2673 if (!IsFAdd && !isFSUB(Opc))
2674 return false;
2675 bool Added = false;
2676 if (canCombineFPFusedMultiply(Root, Root.getOperand(1),
2677 DoRegPressureReduce)) {
2680 Added = true;
2681 }
2682 if (canCombineFPFusedMultiply(Root, Root.getOperand(2),
2683 DoRegPressureReduce)) {
2686 Added = true;
2687 }
2688 return Added;
2689}
2690
2691static bool getFPPatterns(MachineInstr &Root,
2692 SmallVectorImpl<unsigned> &Patterns,
2693 bool DoRegPressureReduce) {
2694 return getFPFusedMultiplyPatterns(Root, Patterns, DoRegPressureReduce);
2695}
2696
2697/// Utility routine that checks if \param MO is defined by an
2698/// \param CombineOpc instruction in the basic block \param MBB
2700 const MachineOperand &MO,
2701 unsigned CombineOpc) {
2702 const MachineRegisterInfo &MRI = MBB.getParent()->getRegInfo();
2703 const MachineInstr *MI = nullptr;
2704
2705 if (MO.isReg() && MO.getReg().isVirtual())
2706 MI = MRI.getUniqueVRegDef(MO.getReg());
2707 // And it needs to be in the trace (otherwise, it won't have a depth).
2708 if (!MI || MI->getParent() != &MBB || MI->getOpcode() != CombineOpc)
2709 return nullptr;
2710 // Must only used by the user we combine with.
2711 if (!MRI.hasOneNonDBGUse(MI->getOperand(0).getReg()))
2712 return nullptr;
2713
2714 return MI;
2715}
2716
2717/// Utility routine that checks if \param MO is defined by a SLLI in \param
2718/// MBB that can be combined by splitting across 2 SHXADD instructions. The
2719/// first SHXADD shift amount is given by \param OuterShiftAmt.
2721 const MachineOperand &MO,
2722 unsigned OuterShiftAmt) {
2723 const MachineInstr *ShiftMI = canCombine(MBB, MO, RISCV::SLLI);
2724 if (!ShiftMI)
2725 return false;
2726
2727 unsigned InnerShiftAmt = ShiftMI->getOperand(2).getImm();
2728 if (InnerShiftAmt < OuterShiftAmt || (InnerShiftAmt - OuterShiftAmt) > 3)
2729 return false;
2730
2731 return true;
2732}
2733
2734// Returns the shift amount from a SHXADD instruction. Returns 0 if the
2735// instruction is not a SHXADD.
2736static unsigned getSHXADDShiftAmount(unsigned Opc) {
2737 switch (Opc) {
2738 default:
2739 return 0;
2740 case RISCV::SH1ADD:
2741 return 1;
2742 case RISCV::SH2ADD:
2743 return 2;
2744 case RISCV::SH3ADD:
2745 return 3;
2746 }
2747}
2748
2749// Returns the shift amount from a SHXADD.UW instruction. Returns 0 if the
2750// instruction is not a SHXADD.UW.
2751static unsigned getSHXADDUWShiftAmount(unsigned Opc) {
2752 switch (Opc) {
2753 default:
2754 return 0;
2755 case RISCV::SH1ADD_UW:
2756 return 1;
2757 case RISCV::SH2ADD_UW:
2758 return 2;
2759 case RISCV::SH3ADD_UW:
2760 return 3;
2761 }
2762}
2763
2764// Look for opportunities to combine (sh3add Z, (add X, (slli Y, 5))) into
2765// (sh3add (sh2add Y, Z), X).
2766static bool getSHXADDPatterns(const MachineInstr &Root,
2767 SmallVectorImpl<unsigned> &Patterns) {
2768 unsigned ShiftAmt = getSHXADDShiftAmount(Root.getOpcode());
2769 if (!ShiftAmt)
2770 return false;
2771
2772 const MachineBasicBlock &MBB = *Root.getParent();
2773
2774 const MachineInstr *AddMI = canCombine(MBB, Root.getOperand(2), RISCV::ADD);
2775 if (!AddMI)
2776 return false;
2777
2778 bool Found = false;
2779 if (canCombineShiftIntoShXAdd(MBB, AddMI->getOperand(1), ShiftAmt)) {
2781 Found = true;
2782 }
2783 if (canCombineShiftIntoShXAdd(MBB, AddMI->getOperand(2), ShiftAmt)) {
2785 Found = true;
2786 }
2787
2788 return Found;
2789}
2790
2802
2804 MachineInstr &Root, SmallVectorImpl<unsigned> &Patterns,
2805 bool DoRegPressureReduce) const {
2806
2807 if (getFPPatterns(Root, Patterns, DoRegPressureReduce))
2808 return true;
2809
2810 if (getSHXADDPatterns(Root, Patterns))
2811 return true;
2812
2813 return TargetInstrInfo::getMachineCombinerPatterns(Root, Patterns,
2814 DoRegPressureReduce);
2815}
2816
2817static unsigned getFPFusedMultiplyOpcode(unsigned RootOpc, unsigned Pattern) {
2818 switch (RootOpc) {
2819 default:
2820 llvm_unreachable("Unexpected opcode");
2821 case RISCV::FADD_H:
2822 return RISCV::FMADD_H;
2823 case RISCV::FADD_S:
2824 return RISCV::FMADD_S;
2825 case RISCV::FADD_D:
2826 return RISCV::FMADD_D;
2827 case RISCV::FSUB_H:
2828 return Pattern == RISCVMachineCombinerPattern::FMSUB ? RISCV::FMSUB_H
2829 : RISCV::FNMSUB_H;
2830 case RISCV::FSUB_S:
2831 return Pattern == RISCVMachineCombinerPattern::FMSUB ? RISCV::FMSUB_S
2832 : RISCV::FNMSUB_S;
2833 case RISCV::FSUB_D:
2834 return Pattern == RISCVMachineCombinerPattern::FMSUB ? RISCV::FMSUB_D
2835 : RISCV::FNMSUB_D;
2836 }
2837}
2838
2839static unsigned getAddendOperandIdx(unsigned Pattern) {
2840 switch (Pattern) {
2841 default:
2842 llvm_unreachable("Unexpected pattern");
2845 return 2;
2848 return 1;
2849 }
2850}
2851
2853 unsigned Pattern,
2856 MachineFunction *MF = Root.getMF();
2857 MachineRegisterInfo &MRI = MF->getRegInfo();
2859
2860 MachineOperand &Mul1 = Prev.getOperand(1);
2861 MachineOperand &Mul2 = Prev.getOperand(2);
2862 MachineOperand &Dst = Root.getOperand(0);
2864
2865 Register DstReg = Dst.getReg();
2866 unsigned FusedOpc = getFPFusedMultiplyOpcode(Root.getOpcode(), Pattern);
2867 uint32_t IntersectedFlags = Root.getFlags() & Prev.getFlags();
2868 DebugLoc MergedLoc =
2870
2871 bool Mul1IsKill = Mul1.isKill();
2872 bool Mul2IsKill = Mul2.isKill();
2873 bool AddendIsKill = Addend.isKill();
2874
2875 // We need to clear kill flags since we may be extending the live range past
2876 // a kill. If the mul had kill flags, we can preserve those since we know
2877 // where the previous range stopped.
2878 MRI.clearKillFlags(Mul1.getReg());
2879 MRI.clearKillFlags(Mul2.getReg());
2880
2882 BuildMI(*MF, MergedLoc, TII->get(FusedOpc), DstReg)
2883 .addReg(Mul1.getReg(), getKillRegState(Mul1IsKill))
2884 .addReg(Mul2.getReg(), getKillRegState(Mul2IsKill))
2885 .addReg(Addend.getReg(), getKillRegState(AddendIsKill))
2886 .setMIFlags(IntersectedFlags);
2887
2888 InsInstrs.push_back(MIB);
2889 if (MRI.hasOneNonDBGUse(Prev.getOperand(0).getReg()))
2890 DelInstrs.push_back(&Prev);
2891 DelInstrs.push_back(&Root);
2892}
2893
2894// Combine patterns like (sh3add Z, (add X, (slli Y, 5))) to
2895// (sh3add (sh2add Y, Z), X) if the shift amount can be split across two
2896// shXadd instructions. The outer shXadd keeps its original opcode.
2897static void
2898genShXAddAddShift(MachineInstr &Root, unsigned AddOpIdx,
2901 DenseMap<Register, unsigned> &InstrIdxForVirtReg) {
2902 MachineFunction *MF = Root.getMF();
2903 MachineRegisterInfo &MRI = MF->getRegInfo();
2905
2906 unsigned OuterShiftAmt = getSHXADDShiftAmount(Root.getOpcode());
2907 assert(OuterShiftAmt != 0 && "Unexpected opcode");
2908
2909 MachineInstr *AddMI = MRI.getUniqueVRegDef(Root.getOperand(2).getReg());
2910 MachineInstr *ShiftMI =
2911 MRI.getUniqueVRegDef(AddMI->getOperand(AddOpIdx).getReg());
2912
2913 unsigned InnerShiftAmt = ShiftMI->getOperand(2).getImm();
2914 assert(InnerShiftAmt >= OuterShiftAmt && "Unexpected shift amount");
2915
2916 unsigned InnerOpc;
2917 switch (InnerShiftAmt - OuterShiftAmt) {
2918 default:
2919 llvm_unreachable("Unexpected shift amount");
2920 case 0:
2921 InnerOpc = RISCV::ADD;
2922 break;
2923 case 1:
2924 InnerOpc = RISCV::SH1ADD;
2925 break;
2926 case 2:
2927 InnerOpc = RISCV::SH2ADD;
2928 break;
2929 case 3:
2930 InnerOpc = RISCV::SH3ADD;
2931 break;
2932 }
2933
2934 const MachineOperand &X = AddMI->getOperand(3 - AddOpIdx);
2935 const MachineOperand &Y = ShiftMI->getOperand(1);
2936 const MachineOperand &Z = Root.getOperand(1);
2937
2938 Register NewVR = MRI.createVirtualRegister(&RISCV::GPRRegClass);
2939
2940 auto MIB1 = BuildMI(*MF, MIMetadata(Root), TII->get(InnerOpc), NewVR)
2941 .addReg(Y.getReg(), getKillRegState(Y.isKill()))
2942 .addReg(Z.getReg(), getKillRegState(Z.isKill()));
2943 auto MIB2 = BuildMI(*MF, MIMetadata(Root), TII->get(Root.getOpcode()),
2944 Root.getOperand(0).getReg())
2945 .addReg(NewVR, RegState::Kill)
2946 .addReg(X.getReg(), getKillRegState(X.isKill()));
2947
2948 InstrIdxForVirtReg.insert(std::make_pair(NewVR, 0));
2949 InsInstrs.push_back(MIB1);
2950 InsInstrs.push_back(MIB2);
2951 DelInstrs.push_back(ShiftMI);
2952 DelInstrs.push_back(AddMI);
2953 DelInstrs.push_back(&Root);
2954}
2955
2957 MachineInstr &Root, unsigned Pattern,
2960 DenseMap<Register, unsigned> &InstrIdxForVirtReg) const {
2961 MachineRegisterInfo &MRI = Root.getMF()->getRegInfo();
2962 switch (Pattern) {
2963 default:
2965 DelInstrs, InstrIdxForVirtReg);
2966 return;
2969 MachineInstr &Prev = *MRI.getVRegDef(Root.getOperand(1).getReg());
2970 combineFPFusedMultiply(Root, Prev, Pattern, InsInstrs, DelInstrs);
2971 return;
2972 }
2975 MachineInstr &Prev = *MRI.getVRegDef(Root.getOperand(2).getReg());
2976 combineFPFusedMultiply(Root, Prev, Pattern, InsInstrs, DelInstrs);
2977 return;
2978 }
2980 genShXAddAddShift(Root, 1, InsInstrs, DelInstrs, InstrIdxForVirtReg);
2981 return;
2983 genShXAddAddShift(Root, 2, InsInstrs, DelInstrs, InstrIdxForVirtReg);
2984 return;
2985 }
2986}
2987
2989 StringRef &ErrInfo) const {
2990 MCInstrDesc const &Desc = MI.getDesc();
2991
2992 for (const auto &[Index, Operand] : enumerate(Desc.operands())) {
2993 const MachineOperand &MO = MI.getOperand(Index);
2994 unsigned OpType = Operand.OperandType;
2995 switch (OpType) {
2996 default:
2997 if (OpType >= RISCVOp::OPERAND_FIRST_RISCV_IMM &&
2999 if (!MO.isImm()) {
3000 ErrInfo = "Expected an immediate operand.";
3001 return false;
3002 }
3003 int64_t Imm = MO.getImm();
3004 bool Ok;
3005 switch (OpType) {
3006 default:
3007 llvm_unreachable("Unexpected operand type");
3008
3009#define CASE_OPERAND_UIMM(NUM) \
3010 case RISCVOp::OPERAND_UIMM##NUM: \
3011 Ok = isUInt<NUM>(Imm); \
3012 break;
3013#define CASE_OPERAND_UIMM_LSB_ZEROS(BITS, SUFFIX) \
3014 case RISCVOp::OPERAND_UIMM##BITS##_LSB##SUFFIX: { \
3015 constexpr size_t NumZeros = sizeof(#SUFFIX) - 1; \
3016 Ok = isShiftedUInt<BITS - NumZeros, NumZeros>(Imm); \
3017 break; \
3018 }
3019#define CASE_OPERAND_SIMM(NUM) \
3020 case RISCVOp::OPERAND_SIMM##NUM: \
3021 Ok = isInt<NUM>(Imm); \
3022 break;
3023 // clang-format off
3048 // clang-format on
3050 Ok = isUInt<5>(Imm) && (Imm != 0);
3051 break;
3053 Ok = isUInt<5>(Imm) && (Imm > 3);
3054 break;
3056 Ok = Imm >= 1 && Imm <= 32;
3057 break;
3059 Ok = Imm >= 1 && Imm <= 64;
3060 break;
3062 Ok = Imm == STI.getXLen();
3063 break;
3065 Ok = isUInt<8>(Imm) && Imm >= 32;
3066 break;
3069 break;
3071 Ok = isShiftedInt<6, 4>(Imm) && (Imm != 0);
3072 break;
3074 Ok = isShiftedUInt<8, 2>(Imm) && (Imm != 0);
3075 break;
3077 Ok = isUInt<16>(Imm) && (Imm != 0);
3078 break;
3080 Ok = Imm == 3;
3081 break;
3083 Ok = Imm == 4;
3084 break;
3086 Ok = (isUInt<5>(Imm) && Imm != 0) || Imm == -1;
3087 break;
3088 // clang-format off
3096 // clang-format on
3098 Ok = Imm >= -15 && Imm <= 16;
3099 break;
3101 Ok = isInt<5>(Imm) && (Imm != 0);
3102 break;
3104 Ok = Imm != 0 && isInt<6>(Imm);
3105 break;
3108 break;
3111 break;
3113 Ok = isShiftedInt<7, 5>(Imm);
3114 break;
3116 Ok = isInt<16>(Imm) && (Imm != 0);
3117 break;
3119 Ok = isInt<20>(Imm);
3120 break;
3122 Ok = STI.is64Bit() ? isUInt<6>(Imm) : isUInt<5>(Imm);
3123 break;
3125 Ok = STI.is64Bit() ? isUInt<6>(Imm) : isUInt<5>(Imm);
3126 Ok = Ok && Imm != 0;
3127 break;
3129 Ok = (isUInt<5>(Imm) && Imm != 0) || (Imm >= 0xfffe0 && Imm <= 0xfffff);
3130 break;
3132 Ok = Imm >= 0 && Imm <= 10;
3133 break;
3135 Ok = Imm >= 0 && Imm <= 7;
3136 break;
3138 Ok = Imm >= 1 && Imm <= 10;
3139 break;
3141 Ok = Imm >= 2 && Imm <= 14;
3142 break;
3144 Ok = Imm >= RISCVZC::RA && Imm <= RISCVZC::RA_S0_S11;
3145 break;
3148 break;
3150 Ok = Imm >= 0 && Imm <= 48 && Imm % 16 == 0;
3151 break;
3154 break;
3156 Ok = Imm == RISCVFPRndMode::RTZ;
3157 break;
3160 break;
3162 Ok = Imm == XSMTVTypeMode::SMT_I8;
3163 break;
3165 Ok = Imm >= 0 && Imm < RISCVCC::COND_INVALID;
3166 break;
3169 break;
3172 Imm;
3173 break;
3175 Ok = (isUInt<5>(Imm) && RISCVVType::isValidSEW(1 << Imm));
3176 break;
3178 Ok = Imm == 0;
3179 break;
3182 if (RISCVII::usesVXRM(Desc.TSFlags))
3183 Ok = isUInt<2>(Imm);
3184 else
3186 break;
3189 break;
3191 Ok = Imm == 1 || Imm == 2 || Imm == 4;
3192 break;
3193 }
3194 if (!Ok) {
3195 ErrInfo = "Invalid immediate";
3196 return false;
3197 }
3198 }
3199 break;
3201 // TODO: We could be stricter about what non-register operands are
3202 // allowed.
3203 if (MO.isReg()) {
3204 ErrInfo = "Expected a non-register operand.";
3205 return false;
3206 }
3207 if (MO.isImm() && !isInt<12>(MO.getImm())) {
3208 ErrInfo = "Invalid immediate";
3209 return false;
3210 }
3211 break;
3214 // TODO: We could be stricter about what non-register operands are
3215 // allowed.
3216 if (MO.isReg()) {
3217 ErrInfo = "Expected a non-register operand.";
3218 return false;
3219 }
3220 if (MO.isImm() && !isUInt<20>(MO.getImm())) {
3221 ErrInfo = "Invalid immediate";
3222 return false;
3223 }
3224 break;
3226 // TODO: We could be stricter about what non-register operands are
3227 // allowed.
3228 if (MO.isReg()) {
3229 ErrInfo = "Expected a non-register operand.";
3230 return false;
3231 }
3232 if (MO.isImm() && !isInt<32>(MO.getImm())) {
3233 ErrInfo = "Invalid immediate";
3234 return false;
3235 }
3236 break;
3238 if (MO.isImm()) {
3239 int64_t Imm = MO.getImm();
3240 // VLMAX is represented as -1.
3241 if (!isUInt<5>(Imm) && Imm != -1) {
3242 ErrInfo = "Invalid immediate";
3243 return false;
3244 }
3245 } else if (!MO.isReg()) {
3246 ErrInfo = "Expected a register or immediate operand.";
3247 return false;
3248 }
3249 break;
3251 if (!MO.isReg() && !MO.isImm()) {
3252 ErrInfo = "Expected a register or immediate operand.";
3253 return false;
3254 }
3255 break;
3256 }
3257 }
3258
3259 const uint64_t TSFlags = Desc.TSFlags;
3260 if (RISCVII::hasVLOp(TSFlags)) {
3261 const MachineOperand &Op = MI.getOperand(RISCVII::getVLOpNum(Desc));
3262 if (!Op.isImm() && !Op.isReg()) {
3263 ErrInfo = "Invalid operand type for VL operand";
3264 return false;
3265 }
3266 if (Op.isReg() && Op.getReg().isValid()) {
3267 const MachineRegisterInfo &MRI = MI.getParent()->getParent()->getRegInfo();
3268 auto *RC = MRI.getRegClass(Op.getReg());
3269 if (!RISCV::GPRNoX0RegClass.hasSubClassEq(RC)) {
3270 ErrInfo = "Invalid register class for VL operand";
3271 return false;
3272 }
3273 }
3274 if (!RISCVII::hasSEWOp(TSFlags)) {
3275 ErrInfo = "VL operand w/o SEW operand?";
3276 return false;
3277 }
3278 }
3279 if (RISCVII::hasSEWOp(TSFlags)) {
3280 unsigned OpIdx = RISCVII::getSEWOpNum(Desc);
3281 if (!MI.getOperand(OpIdx).isImm()) {
3282 ErrInfo = "SEW value expected to be an immediate";
3283 return false;
3284 }
3285 uint64_t Log2SEW = MI.getOperand(OpIdx).getImm();
3286 if (Log2SEW > 31) {
3287 ErrInfo = "Unexpected SEW value";
3288 return false;
3289 }
3290 unsigned SEW = Log2SEW ? 1 << Log2SEW : 8;
3291 if (!RISCVVType::isValidSEW(SEW)) {
3292 ErrInfo = "Unexpected SEW value";
3293 return false;
3294 }
3295 }
3296 if (RISCVII::hasVecPolicyOp(TSFlags)) {
3297 unsigned OpIdx = RISCVII::getVecPolicyOpNum(Desc);
3298 if (!MI.getOperand(OpIdx).isImm()) {
3299 ErrInfo = "Policy operand expected to be an immediate";
3300 return false;
3301 }
3302 uint64_t Policy = MI.getOperand(OpIdx).getImm();
3304 ErrInfo = "Invalid Policy Value";
3305 return false;
3306 }
3307 if (!RISCVII::hasVLOp(TSFlags)) {
3308 ErrInfo = "policy operand w/o VL operand?";
3309 return false;
3310 }
3311
3312 // VecPolicy operands can only exist on instructions with passthru/merge
3313 // arguments. Note that not all arguments with passthru have vec policy
3314 // operands- some instructions have implicit policies.
3315 unsigned UseOpIdx;
3316 if (!MI.isRegTiedToUseOperand(0, &UseOpIdx)) {
3317 ErrInfo = "policy operand w/o tied operand?";
3318 return false;
3319 }
3320 }
3321
3322 if (int Idx = RISCVII::getFRMOpNum(Desc);
3323 Idx >= 0 && MI.getOperand(Idx).getImm() == RISCVFPRndMode::DYN &&
3324 !MI.readsRegister(RISCV::FRM, /*TRI=*/nullptr)) {
3325 ErrInfo = "dynamic rounding mode should read FRM";
3326 return false;
3327 }
3328
3329 return true;
3330}
3331
3333 const MachineInstr &AddrI,
3334 ExtAddrMode &AM) const {
3335 switch (MemI.getOpcode()) {
3336 default:
3337 return false;
3338 case RISCV::LB:
3339 case RISCV::LBU:
3340 case RISCV::LH:
3341 case RISCV::LH_INX:
3342 case RISCV::LHU:
3343 case RISCV::LW:
3344 case RISCV::LW_INX:
3345 case RISCV::LWU:
3346 case RISCV::LD:
3347 case RISCV::LD_RV32:
3348 case RISCV::FLH:
3349 case RISCV::FLW:
3350 case RISCV::FLD:
3351 case RISCV::SB:
3352 case RISCV::SH:
3353 case RISCV::SH_INX:
3354 case RISCV::SW:
3355 case RISCV::SW_INX:
3356 case RISCV::SD:
3357 case RISCV::SD_RV32:
3358 case RISCV::FSH:
3359 case RISCV::FSW:
3360 case RISCV::FSD:
3361 break;
3362 }
3363
3364 if (MemI.getOperand(0).getReg() == Reg)
3365 return false;
3366
3367 if (AddrI.getOpcode() != RISCV::ADDI || !AddrI.getOperand(1).isReg() ||
3368 !AddrI.getOperand(2).isImm())
3369 return false;
3370
3371 int64_t OldOffset = MemI.getOperand(2).getImm();
3372 int64_t Disp = AddrI.getOperand(2).getImm();
3373 int64_t NewOffset = OldOffset + Disp;
3374 if (!STI.is64Bit())
3375 NewOffset = SignExtend64<32>(NewOffset);
3376
3377 if (!isInt<12>(NewOffset))
3378 return false;
3379
3380 AM.BaseReg = AddrI.getOperand(1).getReg();
3381 AM.ScaledReg = 0;
3382 AM.Scale = 0;
3383 AM.Displacement = NewOffset;
3385 return true;
3386}
3387
3389 const ExtAddrMode &AM) const {
3390
3391 const DebugLoc &DL = MemI.getDebugLoc();
3392 MachineBasicBlock &MBB = *MemI.getParent();
3393
3394 assert(AM.ScaledReg == 0 && AM.Scale == 0 &&
3395 "Addressing mode not supported for folding");
3396
3397 return BuildMI(MBB, MemI, DL, get(MemI.getOpcode()))
3398 .addReg(MemI.getOperand(0).getReg(), getDefRegState(MemI.mayLoad()))
3399 .addReg(AM.BaseReg)
3400 .addImm(AM.Displacement)
3401 .setMemRefs(MemI.memoperands())
3402 .setMIFlags(MemI.getFlags());
3403}
3404
3405// TODO: At the moment, MIPS introduced paring of instructions operating with
3406// word or double word. This should be extended with more instructions when more
3407// vendors support load/store pairing.
3409 switch (Opc) {
3410 default:
3411 return false;
3412 case RISCV::SW:
3413 case RISCV::SD:
3414 case RISCV::LD:
3415 case RISCV::LW:
3416 return true;
3417 }
3418}
3419
3421 const TargetRegisterInfo *TRI) {
3422 // If this is a volatile load/store, don't mess with it.
3423 if (LdSt.hasOrderedMemoryRef() || LdSt.getNumExplicitOperands() != 3)
3424 return false;
3425
3426 if (LdSt.getOperand(1).isFI())
3427 return true;
3428
3429 assert(LdSt.getOperand(1).isReg() && "Expected a reg operand.");
3430 // Can't cluster if the instruction modifies the base register
3431 // or it is update form. e.g. ld x5,8(x5)
3432 if (LdSt.modifiesRegister(LdSt.getOperand(1).getReg(), TRI))
3433 return false;
3434
3435 if (!LdSt.getOperand(2).isImm())
3436 return false;
3437
3438 return true;
3439}
3440
3443 int64_t &Offset, bool &OffsetIsScalable, LocationSize &Width,
3444 const TargetRegisterInfo *TRI) const {
3445 if (!LdSt.mayLoadOrStore())
3446 return false;
3447
3448 // Conservatively, only handle scalar loads/stores for now.
3449 switch (LdSt.getOpcode()) {
3450 case RISCV::LB:
3451 case RISCV::LBU:
3452 case RISCV::SB:
3453 case RISCV::LH:
3454 case RISCV::LH_INX:
3455 case RISCV::LHU:
3456 case RISCV::FLH:
3457 case RISCV::SH:
3458 case RISCV::SH_INX:
3459 case RISCV::FSH:
3460 case RISCV::LW:
3461 case RISCV::LW_INX:
3462 case RISCV::LWU:
3463 case RISCV::FLW:
3464 case RISCV::SW:
3465 case RISCV::SW_INX:
3466 case RISCV::FSW:
3467 case RISCV::LD:
3468 case RISCV::LD_RV32:
3469 case RISCV::FLD:
3470 case RISCV::SD:
3471 case RISCV::SD_RV32:
3472 case RISCV::FSD:
3473 break;
3474 default:
3475 return false;
3476 }
3477 const MachineOperand *BaseOp;
3478 OffsetIsScalable = false;
3479 if (!getMemOperandWithOffsetWidth(LdSt, BaseOp, Offset, Width, TRI))
3480 return false;
3481 BaseOps.push_back(BaseOp);
3482 return true;
3483}
3484
3485// TODO: This was copied from SIInstrInfo. Could it be lifted to a common
3486// helper?
3489 const MachineInstr &MI2,
3491 // Only examine the first "base" operand of each instruction, on the
3492 // assumption that it represents the real base address of the memory access.
3493 // Other operands are typically offsets or indices from this base address.
3494 if (BaseOps1.front()->isIdenticalTo(*BaseOps2.front()))
3495 return true;
3496
3497 if (!MI1.hasOneMemOperand() || !MI2.hasOneMemOperand())
3498 return false;
3499
3500 auto MO1 = *MI1.memoperands_begin();
3501 auto MO2 = *MI2.memoperands_begin();
3502 if (MO1->getAddrSpace() != MO2->getAddrSpace())
3503 return false;
3504
3505 auto Base1 = MO1->getValue();
3506 auto Base2 = MO2->getValue();
3507 if (!Base1 || !Base2)
3508 return false;
3509 Base1 = getUnderlyingObject(Base1);
3510 Base2 = getUnderlyingObject(Base2);
3511
3512 if (isa<UndefValue>(Base1) || isa<UndefValue>(Base2))
3513 return false;
3514
3515 return Base1 == Base2;
3516}
3517
3519 ArrayRef<const MachineOperand *> BaseOps1, int64_t Offset1,
3520 bool OffsetIsScalable1, ArrayRef<const MachineOperand *> BaseOps2,
3521 int64_t Offset2, bool OffsetIsScalable2, unsigned ClusterSize,
3522 unsigned NumBytes) const {
3523 // If the mem ops (to be clustered) do not have the same base ptr, then they
3524 // should not be clustered
3525 if (!BaseOps1.empty() && !BaseOps2.empty()) {
3526 const MachineInstr &FirstLdSt = *BaseOps1.front()->getParent();
3527 const MachineInstr &SecondLdSt = *BaseOps2.front()->getParent();
3528 if (!memOpsHaveSameBasePtr(FirstLdSt, BaseOps1, SecondLdSt, BaseOps2))
3529 return false;
3530 } else if (!BaseOps1.empty() || !BaseOps2.empty()) {
3531 // If only one base op is empty, they do not have the same base ptr
3532 return false;
3533 }
3534
3535 unsigned CacheLineSize =
3536 BaseOps1.front()->getParent()->getMF()->getSubtarget().getCacheLineSize();
3537 // Assume a cache line size of 64 bytes if no size is set in RISCVSubtarget.
3539 // Cluster if the memory operations are on the same or a neighbouring cache
3540 // line, but limit the maximum ClusterSize to avoid creating too much
3541 // additional register pressure.
3542 return ClusterSize <= 4 && std::abs(Offset1 - Offset2) < CacheLineSize;
3543}
3544
3545// Set BaseReg (the base register operand), Offset (the byte offset being
3546// accessed) and the access Width of the passed instruction that reads/writes
3547// memory. Returns false if the instruction does not read/write memory or the
3548// BaseReg/Offset/Width can't be determined. Is not guaranteed to always
3549// recognise base operands and offsets in all cases.
3550// TODO: Add an IsScalable bool ref argument (like the equivalent AArch64
3551// function) and set it as appropriate.
3553 const MachineInstr &LdSt, const MachineOperand *&BaseReg, int64_t &Offset,
3554 LocationSize &Width, const TargetRegisterInfo *TRI) const {
3555 if (!LdSt.mayLoadOrStore())
3556 return false;
3557
3558 // Here we assume the standard RISC-V ISA, which uses a base+offset
3559 // addressing mode. You'll need to relax these conditions to support custom
3560 // load/store instructions.
3561 if (LdSt.getNumExplicitOperands() != 3)
3562 return false;
3563 if ((!LdSt.getOperand(1).isReg() && !LdSt.getOperand(1).isFI()) ||
3564 !LdSt.getOperand(2).isImm())
3565 return false;
3566
3567 if (!LdSt.hasOneMemOperand())
3568 return false;
3569
3570 Width = (*LdSt.memoperands_begin())->getSize();
3571 BaseReg = &LdSt.getOperand(1);
3572 Offset = LdSt.getOperand(2).getImm();
3573 return true;
3574}
3575
3577 const MachineInstr &MIa, const MachineInstr &MIb) const {
3578 assert(MIa.mayLoadOrStore() && "MIa must be a load or store.");
3579 assert(MIb.mayLoadOrStore() && "MIb must be a load or store.");
3580
3583 return false;
3584
3585 // Retrieve the base register, offset from the base register and width. Width
3586 // is the size of memory that is being loaded/stored (e.g. 1, 2, 4). If
3587 // base registers are identical, and the offset of a lower memory access +
3588 // the width doesn't overlap the offset of a higher memory access,
3589 // then the memory accesses are different.
3590 const TargetRegisterInfo *TRI = STI.getRegisterInfo();
3591 const MachineOperand *BaseOpA = nullptr, *BaseOpB = nullptr;
3592 int64_t OffsetA = 0, OffsetB = 0;
3594 WidthB = LocationSize::precise(0);
3595 if (getMemOperandWithOffsetWidth(MIa, BaseOpA, OffsetA, WidthA, TRI) &&
3596 getMemOperandWithOffsetWidth(MIb, BaseOpB, OffsetB, WidthB, TRI)) {
3597 if (BaseOpA->isIdenticalTo(*BaseOpB)) {
3598 int LowOffset = std::min(OffsetA, OffsetB);
3599 int HighOffset = std::max(OffsetA, OffsetB);
3600 LocationSize LowWidth = (LowOffset == OffsetA) ? WidthA : WidthB;
3601 if (LowWidth.hasValue() &&
3602 LowOffset + (int)LowWidth.getValue() <= HighOffset)
3603 return true;
3604 }
3605 }
3606 return false;
3607}
3608
3609std::pair<unsigned, unsigned>
3611 const unsigned Mask = RISCVII::MO_DIRECT_FLAG_MASK;
3612 return std::make_pair(TF & Mask, TF & ~Mask);
3613}
3614
3617 using namespace RISCVII;
3618 static const std::pair<unsigned, const char *> TargetFlags[] = {
3619 {MO_CALL, "riscv-call"},
3620 {MO_LO, "riscv-lo"},
3621 {MO_HI, "riscv-hi"},
3622 {MO_PCREL_LO, "riscv-pcrel-lo"},
3623 {MO_PCREL_HI, "riscv-pcrel-hi"},
3624 {MO_GOT_HI, "riscv-got-hi"},
3625 {MO_TPREL_LO, "riscv-tprel-lo"},
3626 {MO_TPREL_HI, "riscv-tprel-hi"},
3627 {MO_TPREL_ADD, "riscv-tprel-add"},
3628 {MO_TLS_GOT_HI, "riscv-tls-got-hi"},
3629 {MO_TLS_GD_HI, "riscv-tls-gd-hi"},
3630 {MO_TLSDESC_HI, "riscv-tlsdesc-hi"},
3631 {MO_TLSDESC_LOAD_LO, "riscv-tlsdesc-load-lo"},
3632 {MO_TLSDESC_ADD_LO, "riscv-tlsdesc-add-lo"},
3633 {MO_TLSDESC_CALL, "riscv-tlsdesc-call"},
3634 {MO_QC_ACCESS, "riscv-qc-access"},
3635 };
3636 return ArrayRef(TargetFlags);
3637}
3639 MachineFunction &MF, bool OutlineFromLinkOnceODRs) const {
3640 const Function &F = MF.getFunction();
3641
3642 // Can F be deduplicated by the linker? If it can, don't outline from it.
3643 if (!OutlineFromLinkOnceODRs && F.hasLinkOnceODRLinkage())
3644 return false;
3645
3646 // Don't outline from functions with section markings; the program could
3647 // expect that all the code is in the named section.
3648 if (F.hasSection())
3649 return false;
3650
3651 // It's safe to outline from MF.
3652 return true;
3653}
3654
3656 unsigned &Flags) const {
3657 // More accurate safety checking is done in getOutliningCandidateInfo.
3659}
3660
3661// Enum values indicating how an outlined call should be constructed.
3667
3672
3674 const MachineFunction *MF = MBB.getParent();
3675 const Function &F = MF->getFunction();
3676 return F.getFnAttribute("fentry-call").getValueAsBool() ||
3677 F.hasFnAttribute("patchable-function-entry");
3678}
3679
3681 MCRegister RegNo) {
3682 return MI.readsRegister(RegNo, TRI) ||
3683 MI.getDesc().hasImplicitUseOfPhysReg(RegNo);
3684}
3685
3687 const TargetRegisterInfo *TRI, MCRegister RegNo) {
3688 return MI.modifiesRegister(RegNo, TRI) ||
3689 MI.getDesc().hasImplicitDefOfPhysReg(RegNo);
3690}
3691
3693 if (!MBB.back().isReturn())
3694 return true;
3696 return true;
3697
3698 // If the candidate reads the pre-set register
3699 // that can be used for expanding PseudoTAIL instruction,
3700 // then we cannot insert tail call.
3701 const TargetSubtargetInfo &STI = MBB.getParent()->getSubtarget();
3702 const RISCVMachineFunctionInfo *RVFI =
3703 MBB.getParent()->getInfo<RISCVMachineFunctionInfo>();
3704 // When cf-protection-branch is active, the outliner will emit PseudoTAILX7
3705 // which always uses X7. Otherwise, PseudoTAIL is emitted and the register
3706 // is determined by Zicfilp at encode time.
3707 MCRegister TailExpandUseRegNo =
3708 RVFI->hasCFProtectionBranch()
3709 ? RISCV::X7
3711 for (const MachineInstr &MI : MBB) {
3712 if (isMIReadsReg(MI, STI.getRegisterInfo(), TailExpandUseRegNo))
3713 return true;
3714 if (isMIModifiesReg(MI, STI.getRegisterInfo(), TailExpandUseRegNo))
3715 break;
3716 }
3717 return false;
3718}
3719
3721 const TargetRegisterInfo &TRI) {
3722 // Candidate registers for saving X5: t1-t6
3723 static const MCPhysReg TempRegs[] = {
3724 RISCV::X6, // t1
3725 RISCV::X7, // t2
3726 RISCV::X28, // t3
3727 RISCV::X29, // t4
3728 RISCV::X30, // t5
3729 RISCV::X31 // t6
3730 };
3731
3732 const MachineFunction *MF = C.getMF();
3733 const MachineRegisterInfo &MRI = MF->getRegInfo();
3734
3735 for (MCPhysReg Reg : TempRegs) {
3736 if (MRI.isReserved(Reg))
3737 continue;
3738
3739 if (C.isAvailableAcrossAndOutOfSeq(Reg, TRI) &&
3740 C.isAvailableInsideSeq(Reg, TRI)) {
3741 return Reg;
3742 }
3743 }
3744
3745 return Register();
3746}
3747
3749 // If the expansion register for tail calls is live across the candidate
3750 // outlined call site, we cannot outline that candidate as the expansion
3751 // would clobber the register.
3752 const RISCVMachineFunctionInfo *RVFI =
3753 C.getMF()->getInfo<RISCVMachineFunctionInfo>();
3754 MCRegister TailExpandUseReg =
3755 RVFI->hasCFProtectionBranch()
3756 ? RISCV::X7
3757 : RISCVII::getTailExpandUseRegNo(STI.getFeatureBits());
3758 if (C.back().isReturn() &&
3759 !C.isAvailableAcrossAndOutOfSeq(TailExpandUseReg, RegInfo)) {
3760 LLVM_DEBUG(dbgs() << "MBB:\n" << *C.getMBB());
3761 LLVM_DEBUG(dbgs() << "Cannot be outlined between: " << C.front() << "and "
3762 << C.back());
3763 LLVM_DEBUG(dbgs() << "Because the tail-call register is live across "
3764 "the proposed outlined function call\n");
3765 return true;
3766 }
3767
3768 // If last instruction is return then we can rely on
3769 // the verification already performed in the getOutliningTypeImpl.
3770 if (C.back().isReturn()) {
3771 assert(!cannotInsertTailCall(*C.getMBB()) &&
3772 "The candidate who uses return instruction must be outlined "
3773 "using tail call");
3774 return false;
3775 }
3776
3777 // Filter out candidates where the X5 register (t0) can't be used to setup
3778 // the function call.
3779 if (!C.isAvailableInsideSeq(RISCV::X5, RegInfo))
3780 return true;
3781
3782 // If X5 is available in the region, use X5 directly (MachineOutlinerDefault).
3783 if (C.isAvailableAcrossAndOutOfSeq(RISCV::X5, RegInfo))
3784 return false;
3785
3786 // Otherwise, try to save X5 into t1-t6 (MachineOutlinerRegSave).
3788 return false;
3789
3790 return true;
3791}
3792
3793std::optional<std::unique_ptr<outliner::OutlinedFunction>>
3795 const MachineModuleInfo &MMI,
3796 std::vector<outliner::Candidate> &RepeatedSequenceLocs,
3797 unsigned MinRepeats) const {
3798
3799 // Analyze each candidate and erase the ones that are not viable.
3800 llvm::erase_if(RepeatedSequenceLocs, [this](auto Candidate) {
3801 return analyzeCandidate(Candidate);
3802 });
3803
3804 // If the sequence doesn't have enough candidates left, then we're done.
3805 if (RepeatedSequenceLocs.size() < MinRepeats)
3806 return std::nullopt;
3807
3808 // Each RepeatedSequenceLoc is identical.
3809 outliner::Candidate &Candidate = RepeatedSequenceLocs[0];
3810 unsigned InstrSizeCExt =
3811 Candidate.getMF()->getSubtarget<RISCVSubtarget>().hasStdExtZca() ? 2 : 4;
3812 unsigned CallOverhead = 0, FrameOverhead = 0;
3813
3814 // Count the number of CFI instructions in the candidate, if present.
3815 unsigned CFICount = 0;
3816 for (auto &I : Candidate) {
3817 if (I.isCFIInstruction())
3818 CFICount++;
3819 }
3820
3821 // Ensure CFI coverage matches: comparing the number of CFIs in the candidate
3822 // with the total number of CFIs in the parent function for each candidate.
3823 // Outlining only a subset of a function’s CFIs would split the unwind state
3824 // across two code regions and lead to incorrect address offsets between the
3825 // outlined body and the remaining code. To preserve correct unwind info, we
3826 // only outline when all CFIs in the function can be outlined together.
3827 for (outliner::Candidate &C : RepeatedSequenceLocs) {
3828 std::vector<MCCFIInstruction> CFIInstructions =
3829 C.getMF()->getFrameInstructions();
3830
3831 if (CFICount > 0 && CFICount != CFIInstructions.size())
3832 return std::nullopt;
3833 }
3834
3836 if (Candidate.back().isReturn()) {
3838 // tail call = auipc + jalr in the worst case without linker relaxation.
3839 // FIXME: This code suggests the JALR can be compressed - how?
3840 CallOverhead = 4 + InstrSizeCExt;
3841 // Using tail call we move ret instruction from caller to callee.
3842 FrameOverhead = 0;
3843 } else {
3844 // call t0, function = 8 bytes.
3845 CallOverhead = 8;
3846 // jr t0 = 4 bytes, 2 bytes if compressed instructions are enabled.
3847 FrameOverhead = InstrSizeCExt;
3848 }
3849
3850 // If we have CFI instructions, we can only outline if the outlined section
3851 // can be a tail call.
3852 if (MOCI != MachineOutlinerTailCall && CFICount > 0)
3853 return std::nullopt;
3854
3856 // Set per-candidate overhead based on X5 availability
3857 for (auto &C : RepeatedSequenceLocs) {
3858
3859 if (C.isAvailableAcrossAndOutOfSeq(RISCV::X5, RegInfo)) {
3860 // X5 is available, just need the call
3861 unsigned CandCallOverhead = 8;
3862 C.setCallInfo(MachineOutlinerDefault, CandCallOverhead);
3863 } else {
3864 // X5 unavailable, need save + call + restore
3865 // Save (2-4) + Call (8) + Restore (2-4)
3866 unsigned CandCallOverhead = InstrSizeCExt + 8 + InstrSizeCExt;
3867 C.setCallInfo(MachineOutlinerRegSave, CandCallOverhead);
3868 }
3869 }
3870 } else {
3871 for (auto &C : RepeatedSequenceLocs)
3872 C.setCallInfo(MOCI, CallOverhead);
3873 }
3874
3875 unsigned SequenceSize = 0;
3876 for (auto &MI : Candidate)
3877 SequenceSize += getInstSizeInBytes(MI);
3878
3879 return std::make_unique<outliner::OutlinedFunction>(
3880 RepeatedSequenceLocs, SequenceSize, FrameOverhead, MOCI);
3881}
3882
3886 unsigned Flags) const {
3887 MachineInstr &MI = *MBBI;
3888 MachineBasicBlock *MBB = MI.getParent();
3889 const TargetRegisterInfo *TRI =
3890 MBB->getParent()->getSubtarget().getRegisterInfo();
3891 const auto &F = MI.getMF()->getFunction();
3892
3893 // We can only outline CFI instructions if we will tail call the outlined
3894 // function, or fix up the CFI offsets. Currently, CFI instructions are
3895 // outlined only if in a tail call.
3896 if (MI.isCFIInstruction())
3898
3899 if (cannotInsertTailCall(*MBB) &&
3900 (MI.isReturn() || isMIModifiesReg(MI, TRI, RISCV::X5)))
3902
3903 // Make sure the operands don't reference something unsafe.
3904 for (const auto &MO : MI.operands()) {
3905
3906 // pcrel-hi and pcrel-lo can't put in separate sections, filter that out
3907 // if any possible.
3908 if (MO.getTargetFlags() == RISCVII::MO_PCREL_LO &&
3909 (MI.getMF()->getTarget().getFunctionSections() || F.hasComdat() ||
3910 F.hasSection() || F.getSectionPrefix()))
3912 }
3913
3914 if (isLPAD(MI))
3916
3918}
3919
3922 const outliner::OutlinedFunction &OF) const {
3923
3924 if (OF.FrameConstructionID == MachineOutlinerTailCall)
3925 return;
3926
3927 MBB.addLiveIn(RISCV::X5);
3928
3929 // Add in a return instruction to the end of the outlined frame.
3930 MBB.insert(MBB.end(), BuildMI(MF, DebugLoc(), get(RISCV::JALR))
3931 .addReg(RISCV::X0, RegState::Define)
3932 .addReg(RISCV::X5)
3933 .addImm(0));
3934}
3935
3939
3940 if (C.CallConstructionID == MachineOutlinerTailCall) {
3941 const RISCVMachineFunctionInfo *RVFI =
3943 unsigned TailOpc =
3944 RVFI->hasCFProtectionBranch() ? RISCV::PseudoTAILX7 : RISCV::PseudoTAIL;
3945 It = MBB.insert(It, BuildMI(MF, DebugLoc(), get(TailOpc))
3946 .addGlobalAddress(M.getNamedValue(MF.getName()),
3947 /*Offset=*/0, RISCVII::MO_CALL));
3948 return It;
3949 }
3950
3951 if (C.CallConstructionID == MachineOutlinerRegSave) {
3952 Register SaveReg = findRegisterToSaveX5To(C, RegInfo);
3953 assert(SaveReg && "Cannot find an available register to save/restore X5.");
3954
3955 // Save: ADDI SaveReg, X5, 0 (equivalent to MV SaveReg, X5)
3956 It = MBB.insert(It, BuildMI(MF, DebugLoc(), get(RISCV::ADDI), SaveReg)
3957 .addReg(RISCV::X5)
3958 .addImm(0));
3959 It++;
3960
3961 // Call: PseudoCALLReg X5
3962 It = MBB.insert(
3963 It, BuildMI(MF, DebugLoc(), get(RISCV::PseudoCALLReg), RISCV::X5)
3964 .addGlobalAddress(M.getNamedValue(MF.getName()), 0,
3966 MachineBasicBlock::iterator CallPt = It;
3967 It++;
3968
3969 // Restore: ADDI X5, SaveReg, 0 (equivalent to MV X5, SaveReg)
3970 It = MBB.insert(It, BuildMI(MF, DebugLoc(), get(RISCV::ADDI), RISCV::X5)
3971 .addReg(SaveReg)
3972 .addImm(0));
3973
3974 return CallPt;
3975 }
3976
3977 // Add in a call instruction to the outlined function at the given location.
3978 It = MBB.insert(It,
3979 BuildMI(MF, DebugLoc(), get(RISCV::PseudoCALLReg), RISCV::X5)
3980 .addGlobalAddress(M.getNamedValue(MF.getName()), 0,
3982 return It;
3983}
3984
3987 DebugLoc &DL,
3988 bool AllowSideEffects) const {
3989
3990 const MachineFunction &MF = *MBB.getParent();
3991 const RISCVRegisterInfo &TRI = *STI.getRegisterInfo();
3992
3993 if (TRI.isGeneralPurposeRegister(MF, Reg)) {
3994 BuildMI(MBB, Iter, DL, get(RISCV::PseudoClearGPR), Reg);
3995 } else if (RISCV::FPR32RegClass.contains(Reg)) {
3996 BuildMI(MBB, Iter, DL, get(RISCV::PseudoClearFPR32), Reg);
3997 } else if (RISCV::FPR64RegClass.contains(Reg)) {
3998 BuildMI(MBB, Iter, DL, get(RISCV::PseudoClearFPR64), Reg);
3999 } else if (RISCV::FPR128RegClass.contains(Reg)) {
4000 BuildMI(MBB, Iter, DL, get(RISCV::PseudoClearFPR128), Reg);
4001 } else if (RISCV::VRRegClass.contains(Reg)) {
4002 BuildMI(MBB, Iter, DL, get(RISCV::PseudoClearVR), Reg);
4003 } else {
4005 "buildClearRegister is not implemented for " + TRI.getRegAsmName(Reg));
4006 }
4007}
4008
4009std::optional<RegImmPair> RISCVInstrInfo::isAddImmediate(const MachineInstr &MI,
4010 Register Reg) const {
4011 // TODO: Handle cases where Reg is a super- or sub-register of the
4012 // destination register.
4013 const MachineOperand &Op0 = MI.getOperand(0);
4014 if (!Op0.isReg() || Reg != Op0.getReg())
4015 return std::nullopt;
4016
4017 // Don't consider ADDIW as a candidate because the caller may not be aware
4018 // of its sign extension behaviour.
4019 if (MI.getOpcode() == RISCV::ADDI && MI.getOperand(1).isReg() &&
4020 MI.getOperand(2).isImm())
4021 return RegImmPair{MI.getOperand(1).getReg(), MI.getOperand(2).getImm()};
4022
4023 return std::nullopt;
4024}
4025
4026// MIR printer helper function to annotate Operands with a comment.
4028 const MachineInstr &MI, const MachineOperand &Op, unsigned OpIdx,
4029 const TargetRegisterInfo *TRI) const {
4030 // Print a generic comment for this operand if there is one.
4031 std::string GenericComment =
4033 if (!GenericComment.empty())
4034 return GenericComment;
4035
4036 const MCInstrDesc &Desc = MI.getDesc();
4037 if (OpIdx >= Desc.getNumOperands())
4038 return std::string();
4039
4040 std::string Comment;
4041 raw_string_ostream OS(Comment);
4042
4043 const MCOperandInfo &OpInfo = Desc.operands()[OpIdx];
4044
4045 // Print the full VType operand of vsetvli/vsetivli instructions, and the SEW
4046 // operand of vector codegen pseudos.
4047 switch (OpInfo.OperandType) {
4050 unsigned Imm = Op.getImm();
4052 break;
4053 }
4055 unsigned Imm = Op.getImm();
4057 break;
4058 }
4060 unsigned Imm = Op.getImm();
4061 OS << "w" << Imm;
4062 break;
4063 }
4066 unsigned Log2SEW = Op.getImm();
4067 unsigned SEW = Log2SEW ? 1 << Log2SEW : 8;
4068 assert(RISCVVType::isValidSEW(SEW) && "Unexpected SEW");
4069 OS << "e" << SEW;
4070 break;
4071 }
4073 unsigned Policy = Op.getImm();
4075 "Invalid Policy Value");
4076 OS << (Policy & RISCVVType::TAIL_AGNOSTIC ? "ta" : "tu") << ", "
4077 << (Policy & RISCVVType::MASK_AGNOSTIC ? "ma" : "mu");
4078 break;
4079 }
4081 if (Op.isImm() && Op.getImm() == -1)
4082 OS << "vl=VLMAX";
4083 else
4084 OS << "vl";
4085 break;
4087 if (RISCVII::usesVXRM(Desc.TSFlags)) {
4089 auto VXRM = static_cast<RISCVVXRndMode::RoundingMode>(Op.getImm());
4090 OS << "vxrm=" << RISCVVXRndMode::roundingModeToString(VXRM);
4091 } else {
4093 auto FRM = static_cast<RISCVFPRndMode::RoundingMode>(Op.getImm());
4094 OS << "frm=" << RISCVFPRndMode::roundingModeToString(FRM);
4095 }
4096 break;
4097 }
4098
4099 return Comment;
4100}
4101
4102// clang-format off
4103#define CASE_RVV_OPCODE_UNMASK_LMUL(OP, LMUL) \
4104 RISCV::Pseudo##OP##_##LMUL
4105
4106#define CASE_RVV_OPCODE_MASK_LMUL(OP, LMUL) \
4107 RISCV::Pseudo##OP##_##LMUL##_MASK
4108
4109#define CASE_RVV_OPCODE_LMUL(OP, LMUL) \
4110 CASE_RVV_OPCODE_UNMASK_LMUL(OP, LMUL): \
4111 case CASE_RVV_OPCODE_MASK_LMUL(OP, LMUL)
4112
4113#define CASE_RVV_OPCODE_UNMASK_WIDEN(OP) \
4114 CASE_RVV_OPCODE_UNMASK_LMUL(OP, MF8): \
4115 case CASE_RVV_OPCODE_UNMASK_LMUL(OP, MF4): \
4116 case CASE_RVV_OPCODE_UNMASK_LMUL(OP, MF2): \
4117 case CASE_RVV_OPCODE_UNMASK_LMUL(OP, M1): \
4118 case CASE_RVV_OPCODE_UNMASK_LMUL(OP, M2): \
4119 case CASE_RVV_OPCODE_UNMASK_LMUL(OP, M4)
4120
4121#define CASE_RVV_OPCODE_UNMASK(OP) \
4122 CASE_RVV_OPCODE_UNMASK_WIDEN(OP): \
4123 case CASE_RVV_OPCODE_UNMASK_LMUL(OP, M8)
4124
4125#define CASE_RVV_OPCODE_MASK_WIDEN(OP) \
4126 CASE_RVV_OPCODE_MASK_LMUL(OP, MF8): \
4127 case CASE_RVV_OPCODE_MASK_LMUL(OP, MF4): \
4128 case CASE_RVV_OPCODE_MASK_LMUL(OP, MF2): \
4129 case CASE_RVV_OPCODE_MASK_LMUL(OP, M1): \
4130 case CASE_RVV_OPCODE_MASK_LMUL(OP, M2): \
4131 case CASE_RVV_OPCODE_MASK_LMUL(OP, M4)
4132
4133#define CASE_RVV_OPCODE_MASK(OP) \
4134 CASE_RVV_OPCODE_MASK_WIDEN(OP): \
4135 case CASE_RVV_OPCODE_MASK_LMUL(OP, M8)
4136
4137#define CASE_RVV_OPCODE_WIDEN(OP) \
4138 CASE_RVV_OPCODE_UNMASK_WIDEN(OP): \
4139 case CASE_RVV_OPCODE_MASK_WIDEN(OP)
4140
4141#define CASE_RVV_OPCODE(OP) \
4142 CASE_RVV_OPCODE_UNMASK(OP): \
4143 case CASE_RVV_OPCODE_MASK(OP)
4144// clang-format on
4145
4146// clang-format off
4147#define CASE_VMA_OPCODE_COMMON(OP, TYPE, LMUL) \
4148 RISCV::PseudoV##OP##_##TYPE##_##LMUL
4149
4150#define CASE_VMA_OPCODE_LMULS(OP, TYPE) \
4151 CASE_VMA_OPCODE_COMMON(OP, TYPE, MF8): \
4152 case CASE_VMA_OPCODE_COMMON(OP, TYPE, MF4): \
4153 case CASE_VMA_OPCODE_COMMON(OP, TYPE, MF2): \
4154 case CASE_VMA_OPCODE_COMMON(OP, TYPE, M1): \
4155 case CASE_VMA_OPCODE_COMMON(OP, TYPE, M2): \
4156 case CASE_VMA_OPCODE_COMMON(OP, TYPE, M4): \
4157 case CASE_VMA_OPCODE_COMMON(OP, TYPE, M8)
4158
4159// VFMA instructions are SEW specific.
4160#define CASE_VFMA_OPCODE_COMMON(OP, TYPE, LMUL, SEW) \
4161 RISCV::PseudoV##OP##_##TYPE##_##LMUL##_##SEW
4162
4163#define CASE_VFMA_OPCODE_LMULS_M1(OP, TYPE, SEW) \
4164 CASE_VFMA_OPCODE_COMMON(OP, TYPE, M1, SEW): \
4165 case CASE_VFMA_OPCODE_COMMON(OP, TYPE, M2, SEW): \
4166 case CASE_VFMA_OPCODE_COMMON(OP, TYPE, M4, SEW): \
4167 case CASE_VFMA_OPCODE_COMMON(OP, TYPE, M8, SEW)
4168
4169#define CASE_VFMA_OPCODE_LMULS_MF2(OP, TYPE, SEW) \
4170 CASE_VFMA_OPCODE_COMMON(OP, TYPE, MF2, SEW): \
4171 case CASE_VFMA_OPCODE_LMULS_M1(OP, TYPE, SEW)
4172
4173#define CASE_VFMA_OPCODE_LMULS_MF4(OP, TYPE, SEW) \
4174 CASE_VFMA_OPCODE_COMMON(OP, TYPE, MF4, SEW): \
4175 case CASE_VFMA_OPCODE_LMULS_MF2(OP, TYPE, SEW)
4176
4177#define CASE_VFMA_OPCODE_VV(OP) \
4178 CASE_VFMA_OPCODE_LMULS_MF4(OP, VV, E16): \
4179 case CASE_VFMA_OPCODE_LMULS_MF4(OP##_ALT, VV, E16): \
4180 case CASE_VFMA_OPCODE_LMULS_MF2(OP, VV, E32): \
4181 case CASE_VFMA_OPCODE_LMULS_M1(OP, VV, E64)
4182
4183#define CASE_VFMA_SPLATS(OP) \
4184 CASE_VFMA_OPCODE_LMULS_MF4(OP, VFPR16, E16): \
4185 case CASE_VFMA_OPCODE_LMULS_MF4(OP##_ALT, VFPR16, E16): \
4186 case CASE_VFMA_OPCODE_LMULS_MF2(OP, VFPR32, E32): \
4187 case CASE_VFMA_OPCODE_LMULS_M1(OP, VFPR64, E64)
4188// clang-format on
4189
4191 unsigned &SrcOpIdx1,
4192 unsigned &SrcOpIdx2) const {
4193 const MCInstrDesc &Desc = MI.getDesc();
4194 if (!Desc.isCommutable())
4195 return false;
4196
4197 switch (MI.getOpcode()) {
4198 case RISCV::TH_MVEQZ:
4199 case RISCV::TH_MVNEZ:
4200 // We can't commute operands if operand 2 (i.e., rs1 in
4201 // mveqz/mvnez rd,rs1,rs2) is the zero-register (as it is
4202 // not valid as the in/out-operand 1).
4203 if (MI.getOperand(2).getReg() == RISCV::X0)
4204 return false;
4205 // Operands 1 and 2 are commutable, if we switch the opcode.
4206 return fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, 1, 2);
4207 case RISCV::QC_SELECTIEQ:
4208 case RISCV::QC_SELECTINE:
4209 case RISCV::QC_SELECTIIEQ:
4210 case RISCV::QC_SELECTIINE:
4211 return fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, 1, 2);
4212 case RISCV::QC_MVEQ:
4213 case RISCV::QC_MVNE:
4214 case RISCV::QC_MVLT:
4215 case RISCV::QC_MVGE:
4216 case RISCV::QC_MVLTU:
4217 case RISCV::QC_MVGEU:
4218 case RISCV::QC_MVEQI:
4219 case RISCV::QC_MVNEI:
4220 case RISCV::QC_MVLTI:
4221 case RISCV::QC_MVGEI:
4222 case RISCV::QC_MVLTUI:
4223 case RISCV::QC_MVGEUI:
4224 return fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, 1, 4);
4225 case RISCV::TH_MULA:
4226 case RISCV::TH_MULAW:
4227 case RISCV::TH_MULAH:
4228 case RISCV::TH_MULS:
4229 case RISCV::TH_MULSW:
4230 case RISCV::TH_MULSH:
4231 // Operands 2 and 3 are commutable.
4232 return fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, 2, 3);
4233 case RISCV::PseudoCCMOVGPRNoX0:
4234 case RISCV::PseudoCCMOVGPR:
4235 // Operands 1 and 2 are commutable.
4236 return fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, 1, 2);
4237 case CASE_RVV_OPCODE(VADD_VV):
4238 case CASE_RVV_OPCODE(VAND_VV):
4239 case CASE_RVV_OPCODE(VOR_VV):
4240 case CASE_RVV_OPCODE(VXOR_VV):
4241 case CASE_RVV_OPCODE_MASK(VMSEQ_VV):
4242 case CASE_RVV_OPCODE_MASK(VMSNE_VV):
4243 case CASE_RVV_OPCODE(VMIN_VV):
4244 case CASE_RVV_OPCODE(VMINU_VV):
4245 case CASE_RVV_OPCODE(VMAX_VV):
4246 case CASE_RVV_OPCODE(VMAXU_VV):
4247 case CASE_RVV_OPCODE(VMUL_VV):
4248 case CASE_RVV_OPCODE(VMULH_VV):
4249 case CASE_RVV_OPCODE(VMULHU_VV):
4250 case CASE_RVV_OPCODE_WIDEN(VWADD_VV):
4251 case CASE_RVV_OPCODE_WIDEN(VWADDU_VV):
4252 case CASE_RVV_OPCODE_WIDEN(VWMUL_VV):
4253 case CASE_RVV_OPCODE_WIDEN(VWMULU_VV):
4254 case CASE_RVV_OPCODE_WIDEN(VWMACC_VV):
4255 case CASE_RVV_OPCODE_WIDEN(VWMACCU_VV):
4256 case CASE_RVV_OPCODE(VABD_VV):
4257 case CASE_RVV_OPCODE(VABDU_VV):
4258 case CASE_RVV_OPCODE_WIDEN(VWABDA_VV):
4259 case CASE_RVV_OPCODE_WIDEN(VWABDAU_VV):
4260 case CASE_RVV_OPCODE_UNMASK(VADC_VVM):
4261 case CASE_RVV_OPCODE(VSADD_VV):
4262 case CASE_RVV_OPCODE(VSADDU_VV):
4263 case CASE_RVV_OPCODE(VAADD_VV):
4264 case CASE_RVV_OPCODE(VAADDU_VV):
4265 case CASE_RVV_OPCODE(VSMUL_VV):
4266 case CASE_RVV_OPCODE_LMUL(VDOT4A_VV, MF2):
4267 case CASE_RVV_OPCODE_LMUL(VDOT4A_VV, M1):
4268 case CASE_RVV_OPCODE_LMUL(VDOT4A_VV, M2):
4269 case CASE_RVV_OPCODE_LMUL(VDOT4A_VV, M4):
4270 case CASE_RVV_OPCODE_LMUL(VDOT4A_VV, M8):
4271 case CASE_RVV_OPCODE_LMUL(VDOT4AU_VV, MF2):
4272 case CASE_RVV_OPCODE_LMUL(VDOT4AU_VV, M1):
4273 case CASE_RVV_OPCODE_LMUL(VDOT4AU_VV, M2):
4274 case CASE_RVV_OPCODE_LMUL(VDOT4AU_VV, M4):
4275 case CASE_RVV_OPCODE_LMUL(VDOT4AU_VV, M8):
4276 // Operands 2 and 3 are commutable.
4277 return fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, 2, 3);
4278 case CASE_VFMA_SPLATS(FMADD):
4279 case CASE_VFMA_SPLATS(FMSUB):
4280 case CASE_VFMA_SPLATS(FMACC):
4281 case CASE_VFMA_SPLATS(FMSAC):
4284 case CASE_VFMA_SPLATS(FNMACC):
4285 case CASE_VFMA_SPLATS(FNMSAC):
4286 case CASE_VFMA_OPCODE_VV(FMACC):
4287 case CASE_VFMA_OPCODE_VV(FMSAC):
4288 case CASE_VFMA_OPCODE_VV(FNMACC):
4289 case CASE_VFMA_OPCODE_VV(FNMSAC):
4290 case CASE_VMA_OPCODE_LMULS(MADD, VX):
4291 case CASE_VMA_OPCODE_LMULS(NMSUB, VX):
4292 case CASE_VMA_OPCODE_LMULS(MACC, VX):
4293 case CASE_VMA_OPCODE_LMULS(NMSAC, VX):
4294 case CASE_VMA_OPCODE_LMULS(MACC, VV):
4295 case CASE_VMA_OPCODE_LMULS(NMSAC, VV): {
4296 // If the tail policy is undisturbed we can't commute.
4297 assert(RISCVII::hasVecPolicyOp(MI.getDesc().TSFlags));
4298 if ((MI.getOperand(RISCVII::getVecPolicyOpNum(MI.getDesc())).getImm() &
4299 1) == 0)
4300 return false;
4301
4302 // For these instructions we can only swap operand 1 and operand 3 by
4303 // changing the opcode.
4304 unsigned CommutableOpIdx1 = 1;
4305 unsigned CommutableOpIdx2 = 3;
4306 if (!fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, CommutableOpIdx1,
4307 CommutableOpIdx2))
4308 return false;
4309 return true;
4310 }
4311 case CASE_VFMA_OPCODE_VV(FMADD):
4315 case CASE_VMA_OPCODE_LMULS(MADD, VV):
4316 case CASE_VMA_OPCODE_LMULS(NMSUB, VV): {
4317 // If the tail policy is undisturbed we can't commute.
4318 assert(RISCVII::hasVecPolicyOp(MI.getDesc().TSFlags));
4319 if ((MI.getOperand(RISCVII::getVecPolicyOpNum(MI.getDesc())).getImm() &
4320 1) == 0)
4321 return false;
4322
4323 // For these instructions we have more freedom. We can commute with the
4324 // other multiplicand or with the addend/subtrahend/minuend.
4325
4326 // Any fixed operand must be from source 1, 2 or 3.
4327 if (SrcOpIdx1 != CommuteAnyOperandIndex && SrcOpIdx1 > 3)
4328 return false;
4329 if (SrcOpIdx2 != CommuteAnyOperandIndex && SrcOpIdx2 > 3)
4330 return false;
4331
4332 // It both ops are fixed one must be the tied source.
4333 if (SrcOpIdx1 != CommuteAnyOperandIndex &&
4334 SrcOpIdx2 != CommuteAnyOperandIndex && SrcOpIdx1 != 1 && SrcOpIdx2 != 1)
4335 return false;
4336
4337 // Look for two different register operands assumed to be commutable
4338 // regardless of the FMA opcode. The FMA opcode is adjusted later if
4339 // needed.
4340 if (SrcOpIdx1 == CommuteAnyOperandIndex ||
4341 SrcOpIdx2 == CommuteAnyOperandIndex) {
4342 // At least one of operands to be commuted is not specified and
4343 // this method is free to choose appropriate commutable operands.
4344 unsigned CommutableOpIdx1 = SrcOpIdx1;
4345 if (SrcOpIdx1 == SrcOpIdx2) {
4346 // Both of operands are not fixed. Set one of commutable
4347 // operands to the tied source.
4348 CommutableOpIdx1 = 1;
4349 } else if (SrcOpIdx1 == CommuteAnyOperandIndex) {
4350 // Only one of the operands is not fixed.
4351 CommutableOpIdx1 = SrcOpIdx2;
4352 }
4353
4354 // CommutableOpIdx1 is well defined now. Let's choose another commutable
4355 // operand and assign its index to CommutableOpIdx2.
4356 unsigned CommutableOpIdx2;
4357 if (CommutableOpIdx1 != 1) {
4358 // If we haven't already used the tied source, we must use it now.
4359 CommutableOpIdx2 = 1;
4360 } else {
4361 Register Op1Reg = MI.getOperand(CommutableOpIdx1).getReg();
4362
4363 // The commuted operands should have different registers.
4364 // Otherwise, the commute transformation does not change anything and
4365 // is useless. We use this as a hint to make our decision.
4366 if (Op1Reg != MI.getOperand(2).getReg())
4367 CommutableOpIdx2 = 2;
4368 else
4369 CommutableOpIdx2 = 3;
4370 }
4371
4372 // Assign the found pair of commutable indices to SrcOpIdx1 and
4373 // SrcOpIdx2 to return those values.
4374 if (!fixCommutedOpIndices(SrcOpIdx1, SrcOpIdx2, CommutableOpIdx1,
4375 CommutableOpIdx2))
4376 return false;
4377 }
4378
4379 return true;
4380 }
4381 }
4382
4383 return TargetInstrInfo::findCommutedOpIndices(MI, SrcOpIdx1, SrcOpIdx2);
4384}
4385
4386// clang-format off
4387#define CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, LMUL) \
4388 case RISCV::PseudoV##OLDOP##_##TYPE##_##LMUL: \
4389 Opc = RISCV::PseudoV##NEWOP##_##TYPE##_##LMUL; \
4390 break;
4391
4392#define CASE_VMA_CHANGE_OPCODE_LMULS(OLDOP, NEWOP, TYPE) \
4393 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, MF8) \
4394 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, MF4) \
4395 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, MF2) \
4396 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M1) \
4397 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M2) \
4398 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M4) \
4399 CASE_VMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M8)
4400
4401// VFMA depends on SEW.
4402#define CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, LMUL, SEW) \
4403 case RISCV::PseudoV##OLDOP##_##TYPE##_##LMUL##_##SEW: \
4404 Opc = RISCV::PseudoV##NEWOP##_##TYPE##_##LMUL##_##SEW; \
4405 break;
4406
4407#define CASE_VFMA_CHANGE_OPCODE_LMULS_M1(OLDOP, NEWOP, TYPE, SEW) \
4408 CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M1, SEW) \
4409 CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M2, SEW) \
4410 CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M4, SEW) \
4411 CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, M8, SEW)
4412
4413#define CASE_VFMA_CHANGE_OPCODE_LMULS_MF2(OLDOP, NEWOP, TYPE, SEW) \
4414 CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, MF2, SEW) \
4415 CASE_VFMA_CHANGE_OPCODE_LMULS_M1(OLDOP, NEWOP, TYPE, SEW)
4416
4417#define CASE_VFMA_CHANGE_OPCODE_LMULS_MF4(OLDOP, NEWOP, TYPE, SEW) \
4418 CASE_VFMA_CHANGE_OPCODE_COMMON(OLDOP, NEWOP, TYPE, MF4, SEW) \
4419 CASE_VFMA_CHANGE_OPCODE_LMULS_MF2(OLDOP, NEWOP, TYPE, SEW)
4420
4421#define CASE_VFMA_CHANGE_OPCODE_VV(OLDOP, NEWOP) \
4422 CASE_VFMA_CHANGE_OPCODE_LMULS_MF4(OLDOP, NEWOP, VV, E16) \
4423 CASE_VFMA_CHANGE_OPCODE_LMULS_MF4(OLDOP##_ALT, NEWOP##_ALT, VV, E16) \
4424 CASE_VFMA_CHANGE_OPCODE_LMULS_MF2(OLDOP, NEWOP, VV, E32) \
4425 CASE_VFMA_CHANGE_OPCODE_LMULS_M1(OLDOP, NEWOP, VV, E64)
4426
4427#define CASE_VFMA_CHANGE_OPCODE_SPLATS(OLDOP, NEWOP) \
4428 CASE_VFMA_CHANGE_OPCODE_LMULS_MF4(OLDOP, NEWOP, VFPR16, E16) \
4429 CASE_VFMA_CHANGE_OPCODE_LMULS_MF4(OLDOP##_ALT, NEWOP##_ALT, VFPR16, E16) \
4430 CASE_VFMA_CHANGE_OPCODE_LMULS_MF2(OLDOP, NEWOP, VFPR32, E32) \
4431 CASE_VFMA_CHANGE_OPCODE_LMULS_M1(OLDOP, NEWOP, VFPR64, E64)
4432// clang-format on
4433
4435 bool NewMI,
4436 unsigned OpIdx1,
4437 unsigned OpIdx2) const {
4438 auto cloneIfNew = [NewMI](MachineInstr &MI) -> MachineInstr & {
4439 if (NewMI)
4440 return *MI.getParent()->getParent()->CloneMachineInstr(&MI);
4441 return MI;
4442 };
4443
4444 switch (MI.getOpcode()) {
4445 case RISCV::TH_MVEQZ:
4446 case RISCV::TH_MVNEZ: {
4447 auto &WorkingMI = cloneIfNew(MI);
4448 WorkingMI.setDesc(get(MI.getOpcode() == RISCV::TH_MVEQZ ? RISCV::TH_MVNEZ
4449 : RISCV::TH_MVEQZ));
4450 return TargetInstrInfo::commuteInstructionImpl(WorkingMI, false, OpIdx1,
4451 OpIdx2);
4452 }
4453 case RISCV::QC_SELECTIEQ:
4454 case RISCV::QC_SELECTINE:
4455 case RISCV::QC_SELECTIIEQ:
4456 case RISCV::QC_SELECTIINE:
4457 return TargetInstrInfo::commuteInstructionImpl(MI, NewMI, OpIdx1, OpIdx2);
4458 case RISCV::QC_MVEQ:
4459 case RISCV::QC_MVNE:
4460 case RISCV::QC_MVLT:
4461 case RISCV::QC_MVGE:
4462 case RISCV::QC_MVLTU:
4463 case RISCV::QC_MVGEU:
4464 case RISCV::QC_MVEQI:
4465 case RISCV::QC_MVNEI:
4466 case RISCV::QC_MVLTI:
4467 case RISCV::QC_MVGEI:
4468 case RISCV::QC_MVLTUI:
4469 case RISCV::QC_MVGEUI: {
4470 auto &WorkingMI = cloneIfNew(MI);
4471 WorkingMI.setDesc(get(getInverseXqcicmOpcode(MI.getOpcode())));
4472 return TargetInstrInfo::commuteInstructionImpl(WorkingMI, false, OpIdx1,
4473 OpIdx2);
4474 }
4475 case RISCV::PseudoCCMOVGPRNoX0:
4476 case RISCV::PseudoCCMOVGPR: {
4477 // CCMOV can be commuted by inverting the condition.
4478 unsigned BCC = MI.getOperand(MI.getNumExplicitOperands() - 3).getImm();
4480 auto &WorkingMI = cloneIfNew(MI);
4481 WorkingMI.getOperand(MI.getNumExplicitOperands() - 3).setImm(BCC);
4482 return TargetInstrInfo::commuteInstructionImpl(WorkingMI, /*NewMI*/ false,
4483 OpIdx1, OpIdx2);
4484 }
4485 case CASE_VFMA_SPLATS(FMACC):
4486 case CASE_VFMA_SPLATS(FMADD):
4487 case CASE_VFMA_SPLATS(FMSAC):
4488 case CASE_VFMA_SPLATS(FMSUB):
4489 case CASE_VFMA_SPLATS(FNMACC):
4491 case CASE_VFMA_SPLATS(FNMSAC):
4493 case CASE_VFMA_OPCODE_VV(FMACC):
4494 case CASE_VFMA_OPCODE_VV(FMSAC):
4495 case CASE_VFMA_OPCODE_VV(FNMACC):
4496 case CASE_VFMA_OPCODE_VV(FNMSAC):
4497 case CASE_VMA_OPCODE_LMULS(MADD, VX):
4498 case CASE_VMA_OPCODE_LMULS(NMSUB, VX):
4499 case CASE_VMA_OPCODE_LMULS(MACC, VX):
4500 case CASE_VMA_OPCODE_LMULS(NMSAC, VX):
4501 case CASE_VMA_OPCODE_LMULS(MACC, VV):
4502 case CASE_VMA_OPCODE_LMULS(NMSAC, VV): {
4503 // It only make sense to toggle these between clobbering the
4504 // addend/subtrahend/minuend one of the multiplicands.
4505 assert((OpIdx1 == 1 || OpIdx2 == 1) && "Unexpected opcode index");
4506 assert((OpIdx1 == 3 || OpIdx2 == 3) && "Unexpected opcode index");
4507 unsigned Opc;
4508 switch (MI.getOpcode()) {
4509 default:
4510 llvm_unreachable("Unexpected opcode");
4511 CASE_VFMA_CHANGE_OPCODE_SPLATS(FMACC, FMADD)
4512 CASE_VFMA_CHANGE_OPCODE_SPLATS(FMADD, FMACC)
4519 CASE_VFMA_CHANGE_OPCODE_VV(FMACC, FMADD)
4523 CASE_VMA_CHANGE_OPCODE_LMULS(MACC, MADD, VX)
4524 CASE_VMA_CHANGE_OPCODE_LMULS(MADD, MACC, VX)
4525 CASE_VMA_CHANGE_OPCODE_LMULS(NMSAC, NMSUB, VX)
4526 CASE_VMA_CHANGE_OPCODE_LMULS(NMSUB, NMSAC, VX)
4527 CASE_VMA_CHANGE_OPCODE_LMULS(MACC, MADD, VV)
4528 CASE_VMA_CHANGE_OPCODE_LMULS(NMSAC, NMSUB, VV)
4529 }
4530
4531 auto &WorkingMI = cloneIfNew(MI);
4532 WorkingMI.setDesc(get(Opc));
4533 return TargetInstrInfo::commuteInstructionImpl(WorkingMI, /*NewMI=*/false,
4534 OpIdx1, OpIdx2);
4535 }
4536 case CASE_VFMA_OPCODE_VV(FMADD):
4540 case CASE_VMA_OPCODE_LMULS(MADD, VV):
4541 case CASE_VMA_OPCODE_LMULS(NMSUB, VV): {
4542 assert((OpIdx1 == 1 || OpIdx2 == 1) && "Unexpected opcode index");
4543 // If one of the operands, is the addend we need to change opcode.
4544 // Otherwise we're just swapping 2 of the multiplicands.
4545 if (OpIdx1 == 3 || OpIdx2 == 3) {
4546 unsigned Opc;
4547 switch (MI.getOpcode()) {
4548 default:
4549 llvm_unreachable("Unexpected opcode");
4550 CASE_VFMA_CHANGE_OPCODE_VV(FMADD, FMACC)
4554 CASE_VMA_CHANGE_OPCODE_LMULS(MADD, MACC, VV)
4555 CASE_VMA_CHANGE_OPCODE_LMULS(NMSUB, NMSAC, VV)
4556 }
4557
4558 auto &WorkingMI = cloneIfNew(MI);
4559 WorkingMI.setDesc(get(Opc));
4560 return TargetInstrInfo::commuteInstructionImpl(WorkingMI, /*NewMI=*/false,
4561 OpIdx1, OpIdx2);
4562 }
4563 // Let the default code handle it.
4564 break;
4565 }
4566 }
4567
4568 return TargetInstrInfo::commuteInstructionImpl(MI, NewMI, OpIdx1, OpIdx2);
4569}
4570
4571#undef CASE_VMA_CHANGE_OPCODE_COMMON
4572#undef CASE_VMA_CHANGE_OPCODE_LMULS
4573#undef CASE_VFMA_CHANGE_OPCODE_COMMON
4574#undef CASE_VFMA_CHANGE_OPCODE_LMULS_M1
4575#undef CASE_VFMA_CHANGE_OPCODE_LMULS_MF2
4576#undef CASE_VFMA_CHANGE_OPCODE_LMULS_MF4
4577#undef CASE_VFMA_CHANGE_OPCODE_VV
4578#undef CASE_VFMA_CHANGE_OPCODE_SPLATS
4579
4580#undef CASE_RVV_OPCODE_UNMASK_LMUL
4581#undef CASE_RVV_OPCODE_MASK_LMUL
4582#undef CASE_RVV_OPCODE_LMUL
4583#undef CASE_RVV_OPCODE_UNMASK_WIDEN
4584#undef CASE_RVV_OPCODE_UNMASK
4585#undef CASE_RVV_OPCODE_MASK_WIDEN
4586#undef CASE_RVV_OPCODE_MASK
4587#undef CASE_RVV_OPCODE_WIDEN
4588#undef CASE_RVV_OPCODE
4589
4590#undef CASE_VMA_OPCODE_COMMON
4591#undef CASE_VMA_OPCODE_LMULS
4592#undef CASE_VFMA_OPCODE_COMMON
4593#undef CASE_VFMA_OPCODE_LMULS_M1
4594#undef CASE_VFMA_OPCODE_LMULS_MF2
4595#undef CASE_VFMA_OPCODE_LMULS_MF4
4596#undef CASE_VFMA_OPCODE_VV
4597#undef CASE_VFMA_SPLATS
4598
4600 switch (MI.getOpcode()) {
4601 default:
4602 break;
4603 case RISCV::ADD:
4604 case RISCV::OR:
4605 case RISCV::XOR:
4606 // Normalize (so we hit the next if clause).
4607 // add/[x]or rd, zero, rs => add/[x]or rd, rs, zero
4608 if (MI.getOperand(1).getReg() == RISCV::X0)
4609 commuteInstruction(MI);
4610 // add/[x]or rd, rs, zero => addi rd, rs, 0
4611 if (MI.getOperand(2).getReg() == RISCV::X0) {
4612 MI.getOperand(2).ChangeToImmediate(0);
4613 MI.setDesc(get(RISCV::ADDI));
4614 return true;
4615 }
4616 // xor rd, rs, rs => addi rd, zero, 0
4617 if (MI.getOpcode() == RISCV::XOR &&
4618 MI.getOperand(1).getReg() == MI.getOperand(2).getReg()) {
4619 MI.getOperand(1).setReg(RISCV::X0);
4620 MI.getOperand(2).ChangeToImmediate(0);
4621 MI.setDesc(get(RISCV::ADDI));
4622 return true;
4623 }
4624 break;
4625 case RISCV::ORI:
4626 case RISCV::XORI:
4627 // [x]ori rd, zero, N => addi rd, zero, N
4628 if (MI.getOperand(1).getReg() == RISCV::X0) {
4629 MI.setDesc(get(RISCV::ADDI));
4630 return true;
4631 }
4632 break;
4633 case RISCV::SUB:
4634 // sub rd, rs, zero => addi rd, rs, 0
4635 if (MI.getOperand(2).getReg() == RISCV::X0) {
4636 MI.getOperand(2).ChangeToImmediate(0);
4637 MI.setDesc(get(RISCV::ADDI));
4638 return true;
4639 }
4640 break;
4641 case RISCV::SUBW:
4642 // subw rd, rs, zero => addiw rd, rs, 0
4643 if (MI.getOperand(2).getReg() == RISCV::X0) {
4644 MI.getOperand(2).ChangeToImmediate(0);
4645 MI.setDesc(get(RISCV::ADDIW));
4646 return true;
4647 }
4648 break;
4649 case RISCV::ADDW:
4650 // Normalize (so we hit the next if clause).
4651 // addw rd, zero, rs => addw rd, rs, zero
4652 if (MI.getOperand(1).getReg() == RISCV::X0)
4653 commuteInstruction(MI);
4654 // addw rd, rs, zero => addiw rd, rs, 0
4655 if (MI.getOperand(2).getReg() == RISCV::X0) {
4656 MI.getOperand(2).ChangeToImmediate(0);
4657 MI.setDesc(get(RISCV::ADDIW));
4658 return true;
4659 }
4660 break;
4661 case RISCV::SH1ADD:
4662 case RISCV::SH1ADD_UW:
4663 case RISCV::SH2ADD:
4664 case RISCV::SH2ADD_UW:
4665 case RISCV::SH3ADD:
4666 case RISCV::SH3ADD_UW:
4667 // shNadd[.uw] rd, zero, rs => addi rd, rs, 0
4668 if (MI.getOperand(1).getReg() == RISCV::X0) {
4669 MI.removeOperand(1);
4670 MI.addOperand(MachineOperand::CreateImm(0));
4671 MI.setDesc(get(RISCV::ADDI));
4672 return true;
4673 }
4674 // shNadd[.uw] rd, rs, zero => slli[.uw] rd, rs, N
4675 if (MI.getOperand(2).getReg() == RISCV::X0) {
4676 MI.removeOperand(2);
4677 unsigned Opc = MI.getOpcode();
4678 if (Opc == RISCV::SH1ADD_UW || Opc == RISCV::SH2ADD_UW ||
4679 Opc == RISCV::SH3ADD_UW) {
4681 MI.setDesc(get(RISCV::SLLI_UW));
4682 return true;
4683 }
4685 MI.setDesc(get(RISCV::SLLI));
4686 return true;
4687 }
4688 break;
4689 case RISCV::AND:
4690 case RISCV::MUL:
4691 case RISCV::MULH:
4692 case RISCV::MULHSU:
4693 case RISCV::MULHU:
4694 case RISCV::MULW:
4695 // and rd, zero, rs => addi rd, zero, 0
4696 // mul* rd, zero, rs => addi rd, zero, 0
4697 // and rd, rs, zero => addi rd, zero, 0
4698 // mul* rd, rs, zero => addi rd, zero, 0
4699 if (MI.getOperand(1).getReg() == RISCV::X0 ||
4700 MI.getOperand(2).getReg() == RISCV::X0) {
4701 MI.getOperand(1).setReg(RISCV::X0);
4702 MI.getOperand(2).ChangeToImmediate(0);
4703 MI.setDesc(get(RISCV::ADDI));
4704 return true;
4705 }
4706 break;
4707 case RISCV::ANDI:
4708 // andi rd, zero, C => addi rd, zero, 0
4709 if (MI.getOperand(1).getReg() == RISCV::X0) {
4710 MI.getOperand(2).setImm(0);
4711 MI.setDesc(get(RISCV::ADDI));
4712 return true;
4713 }
4714 break;
4715 case RISCV::SLL:
4716 case RISCV::SRL:
4717 case RISCV::SRA:
4718 // shift rd, zero, rs => addi rd, zero, 0
4719 if (MI.getOperand(1).getReg() == RISCV::X0) {
4720 MI.getOperand(2).ChangeToImmediate(0);
4721 MI.setDesc(get(RISCV::ADDI));
4722 return true;
4723 }
4724 // shift rd, rs, zero => addi rd, rs, 0
4725 if (MI.getOperand(2).getReg() == RISCV::X0) {
4726 MI.getOperand(2).ChangeToImmediate(0);
4727 MI.setDesc(get(RISCV::ADDI));
4728 return true;
4729 }
4730 break;
4731 case RISCV::SLLW:
4732 case RISCV::SRLW:
4733 case RISCV::SRAW:
4734 // shiftw rd, zero, rs => addi rd, zero, 0
4735 if (MI.getOperand(1).getReg() == RISCV::X0) {
4736 MI.getOperand(2).ChangeToImmediate(0);
4737 MI.setDesc(get(RISCV::ADDI));
4738 return true;
4739 }
4740 break;
4741 case RISCV::SLLI:
4742 case RISCV::SRLI:
4743 case RISCV::SRAI:
4744 case RISCV::SLLIW:
4745 case RISCV::SRLIW:
4746 case RISCV::SRAIW:
4747 case RISCV::SLLI_UW:
4748 // shiftimm rd, zero, N => addi rd, zero, 0
4749 if (MI.getOperand(1).getReg() == RISCV::X0) {
4750 MI.getOperand(2).setImm(0);
4751 MI.setDesc(get(RISCV::ADDI));
4752 return true;
4753 }
4754 break;
4755 case RISCV::SLTU:
4756 case RISCV::ADD_UW:
4757 // sltu rd, zero, zero => addi rd, zero, 0
4758 // add.uw rd, zero, zero => addi rd, zero, 0
4759 if (MI.getOperand(1).getReg() == RISCV::X0 &&
4760 MI.getOperand(2).getReg() == RISCV::X0) {
4761 MI.getOperand(2).ChangeToImmediate(0);
4762 MI.setDesc(get(RISCV::ADDI));
4763 return true;
4764 }
4765 // add.uw rd, zero, rs => addi rd, rs, 0
4766 if (MI.getOpcode() == RISCV::ADD_UW &&
4767 MI.getOperand(1).getReg() == RISCV::X0) {
4768 MI.removeOperand(1);
4769 MI.addOperand(MachineOperand::CreateImm(0));
4770 MI.setDesc(get(RISCV::ADDI));
4771 }
4772 break;
4773 case RISCV::SLTIU:
4774 // sltiu rd, zero, NZC => addi rd, zero, 1
4775 // sltiu rd, zero, 0 => addi rd, zero, 0
4776 if (MI.getOperand(1).getReg() == RISCV::X0) {
4777 MI.getOperand(2).setImm(MI.getOperand(2).getImm() != 0);
4778 MI.setDesc(get(RISCV::ADDI));
4779 return true;
4780 }
4781 break;
4782 case RISCV::SEXT_H:
4783 case RISCV::SEXT_B:
4784 case RISCV::ZEXT_H_RV32:
4785 case RISCV::ZEXT_H_RV64:
4786 // sext.[hb] rd, zero => addi rd, zero, 0
4787 // zext.h rd, zero => addi rd, zero, 0
4788 if (MI.getOperand(1).getReg() == RISCV::X0) {
4789 MI.addOperand(MachineOperand::CreateImm(0));
4790 MI.setDesc(get(RISCV::ADDI));
4791 return true;
4792 }
4793 break;
4794 case RISCV::MIN:
4795 case RISCV::MINU:
4796 case RISCV::MAX:
4797 case RISCV::MAXU:
4798 // min|max rd, rs, rs => addi rd, rs, 0
4799 if (MI.getOperand(1).getReg() == MI.getOperand(2).getReg()) {
4800 MI.getOperand(2).ChangeToImmediate(0);
4801 MI.setDesc(get(RISCV::ADDI));
4802 return true;
4803 }
4804 break;
4805 case RISCV::BEQ:
4806 case RISCV::BNE:
4807 // b{eq,ne} zero, rs, imm => b{eq,ne} rs, zero, imm
4808 if (MI.getOperand(0).getReg() == RISCV::X0) {
4809 MachineOperand MO0 = MI.getOperand(0);
4810 MI.removeOperand(0);
4811 MI.insert(MI.operands_begin() + 1, {MO0});
4812 }
4813 break;
4814 case RISCV::BLTU:
4815 // bltu zero, rs, imm => bne rs, zero, imm
4816 if (MI.getOperand(0).getReg() == RISCV::X0) {
4817 MachineOperand MO0 = MI.getOperand(0);
4818 MI.removeOperand(0);
4819 MI.insert(MI.operands_begin() + 1, {MO0});
4820 MI.setDesc(get(RISCV::BNE));
4821 }
4822 break;
4823 case RISCV::BGEU:
4824 // bgeu zero, rs, imm => beq rs, zero, imm
4825 if (MI.getOperand(0).getReg() == RISCV::X0) {
4826 MachineOperand MO0 = MI.getOperand(0);
4827 MI.removeOperand(0);
4828 MI.insert(MI.operands_begin() + 1, {MO0});
4829 MI.setDesc(get(RISCV::BEQ));
4830 }
4831 break;
4832 }
4833 return false;
4834}
4835
4836// clang-format off
4837#define CASE_WIDEOP_OPCODE_COMMON(OP, LMUL) \
4838 RISCV::PseudoV##OP##_##LMUL##_TIED
4839
4840#define CASE_WIDEOP_OPCODE_LMULS(OP) \
4841 CASE_WIDEOP_OPCODE_COMMON(OP, MF8): \
4842 case CASE_WIDEOP_OPCODE_COMMON(OP, MF4): \
4843 case CASE_WIDEOP_OPCODE_COMMON(OP, MF2): \
4844 case CASE_WIDEOP_OPCODE_COMMON(OP, M1): \
4845 case CASE_WIDEOP_OPCODE_COMMON(OP, M2): \
4846 case CASE_WIDEOP_OPCODE_COMMON(OP, M4)
4847
4848#define CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, LMUL) \
4849 case RISCV::PseudoV##OP##_##LMUL##_TIED: \
4850 NewOpc = RISCV::PseudoV##OP##_##LMUL; \
4851 break;
4852
4853#define CASE_WIDEOP_CHANGE_OPCODE_LMULS(OP) \
4854 CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF8) \
4855 CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF4) \
4856 CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF2) \
4857 CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, M1) \
4858 CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, M2) \
4859 CASE_WIDEOP_CHANGE_OPCODE_COMMON(OP, M4)
4860
4861// FP Widening Ops may by SEW aware. Create SEW aware cases for these cases.
4862#define CASE_FP_WIDEOP_OPCODE_COMMON(OP, LMUL, SEW) \
4863 RISCV::PseudoV##OP##_##LMUL##_##SEW##_TIED
4864
4865#define CASE_FP_WIDEOP_OPCODE_LMULS(OP) \
4866 CASE_FP_WIDEOP_OPCODE_COMMON(OP, MF4, E16): \
4867 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, MF2, E16): \
4868 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, MF2, E32): \
4869 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M1, E16): \
4870 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M1, E32): \
4871 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M2, E16): \
4872 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M2, E32): \
4873 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M4, E16): \
4874 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M4, E32) \
4875
4876#define CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, LMUL, SEW) \
4877 case RISCV::PseudoV##OP##_##LMUL##_##SEW##_TIED: \
4878 NewOpc = RISCV::PseudoV##OP##_##LMUL##_##SEW; \
4879 break;
4880
4881#define CASE_FP_WIDEOP_CHANGE_OPCODE_LMULS(OP) \
4882 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF4, E16) \
4883 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF2, E16) \
4884 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF2, E32) \
4885 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M1, E16) \
4886 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M1, E32) \
4887 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M2, E16) \
4888 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M2, E32) \
4889 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M4, E16) \
4890 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M4, E32) \
4891
4892#define CASE_FP_WIDEOP_OPCODE_LMULS_ALT(OP) \
4893 CASE_FP_WIDEOP_OPCODE_COMMON(OP, MF4, E16): \
4894 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, MF2, E16): \
4895 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M1, E16): \
4896 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M2, E16): \
4897 case CASE_FP_WIDEOP_OPCODE_COMMON(OP, M4, E16)
4898
4899#define CASE_FP_WIDEOP_CHANGE_OPCODE_LMULS_ALT(OP) \
4900 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF4, E16) \
4901 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, MF2, E16) \
4902 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M1, E16) \
4903 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M2, E16) \
4904 CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON(OP, M4, E16)
4905// clang-format on
4906
4908 LiveVariables *LV,
4909 LiveIntervals *LIS) const {
4911 switch (MI.getOpcode()) {
4912 default:
4913 return nullptr;
4914 case CASE_FP_WIDEOP_OPCODE_LMULS_ALT(FWADD_ALT_WV):
4915 case CASE_FP_WIDEOP_OPCODE_LMULS_ALT(FWSUB_ALT_WV):
4916 case CASE_FP_WIDEOP_OPCODE_LMULS(FWADD_WV):
4917 case CASE_FP_WIDEOP_OPCODE_LMULS(FWSUB_WV): {
4918 assert(RISCVII::hasVecPolicyOp(MI.getDesc().TSFlags) &&
4919 MI.getNumExplicitOperands() == 7 &&
4920 "Expect 7 explicit operands rd, rs2, rs1, rm, vl, sew, policy");
4921 // If the tail policy is undisturbed we can't convert.
4922 if ((MI.getOperand(RISCVII::getVecPolicyOpNum(MI.getDesc())).getImm() &
4923 1) == 0)
4924 return nullptr;
4925 // clang-format off
4926 unsigned NewOpc;
4927 switch (MI.getOpcode()) {
4928 default:
4929 llvm_unreachable("Unexpected opcode");
4934 }
4935 // clang-format on
4936
4937 MachineBasicBlock &MBB = *MI.getParent();
4938 MIB = BuildMI(MBB, MI, MI.getDebugLoc(), get(NewOpc))
4939 .add(MI.getOperand(0))
4940 .addReg(MI.getOperand(0).getReg(), RegState::Undef)
4941 .add(MI.getOperand(1))
4942 .add(MI.getOperand(2))
4943 .add(MI.getOperand(3))
4944 .add(MI.getOperand(4))
4945 .add(MI.getOperand(5))
4946 .add(MI.getOperand(6));
4947 break;
4948 }
4949 case CASE_WIDEOP_OPCODE_LMULS(WADD_WV):
4950 case CASE_WIDEOP_OPCODE_LMULS(WADDU_WV):
4951 case CASE_WIDEOP_OPCODE_LMULS(WSUB_WV):
4952 case CASE_WIDEOP_OPCODE_LMULS(WSUBU_WV): {
4953 // If the tail policy is undisturbed we can't convert.
4954 assert(RISCVII::hasVecPolicyOp(MI.getDesc().TSFlags) &&
4955 MI.getNumExplicitOperands() == 6);
4956 if ((MI.getOperand(RISCVII::getVecPolicyOpNum(MI.getDesc())).getImm() &
4957 1) == 0)
4958 return nullptr;
4959
4960 // clang-format off
4961 unsigned NewOpc;
4962 switch (MI.getOpcode()) {
4963 default:
4964 llvm_unreachable("Unexpected opcode");
4969 }
4970 // clang-format on
4971
4972 MachineBasicBlock &MBB = *MI.getParent();
4973 MIB = BuildMI(MBB, MI, MI.getDebugLoc(), get(NewOpc))
4974 .add(MI.getOperand(0))
4975 .addReg(MI.getOperand(0).getReg(), RegState::Undef)
4976 .add(MI.getOperand(1))
4977 .add(MI.getOperand(2))
4978 .add(MI.getOperand(3))
4979 .add(MI.getOperand(4))
4980 .add(MI.getOperand(5));
4981 break;
4982 }
4983 }
4984 MIB.copyImplicitOps(MI);
4985
4986 if (LV) {
4987 unsigned NumOps = MI.getNumOperands();
4988 for (unsigned I = 1; I < NumOps; ++I) {
4989 MachineOperand &Op = MI.getOperand(I);
4990 if (Op.isReg() && Op.isKill())
4991 LV->replaceKillInstruction(Op.getReg(), MI, *MIB);
4992 }
4993 }
4994
4995 if (LIS) {
4996 SlotIndex Idx = LIS->ReplaceMachineInstrInMaps(MI, *MIB);
4997
4998 if (MI.getOperand(0).isEarlyClobber()) {
4999 // Use operand 1 was tied to early-clobber def operand 0, so its live
5000 // interval could have ended at an early-clobber slot. Now they are not
5001 // tied we need to update it to the normal register slot.
5002 LiveInterval &LI = LIS->getInterval(MI.getOperand(1).getReg());
5004 if (S->end == Idx.getRegSlot(true))
5005 S->end = Idx.getRegSlot();
5006 }
5007 }
5008
5009 return MIB;
5010}
5011
5012#undef CASE_WIDEOP_OPCODE_COMMON
5013#undef CASE_WIDEOP_OPCODE_LMULS
5014#undef CASE_WIDEOP_CHANGE_OPCODE_COMMON
5015#undef CASE_WIDEOP_CHANGE_OPCODE_LMULS
5016#undef CASE_FP_WIDEOP_OPCODE_COMMON
5017#undef CASE_FP_WIDEOP_OPCODE_LMULS
5018#undef CASE_FP_WIDEOP_CHANGE_OPCODE_COMMON
5019#undef CASE_FP_WIDEOP_CHANGE_OPCODE_LMULS
5020
5023 Register DestReg, uint32_t Amount,
5024 MachineInstr::MIFlag Flag) const {
5025 MachineRegisterInfo &MRI = MF.getRegInfo();
5026 if (llvm::has_single_bit(Amount)) {
5027 uint32_t ShiftAmount = Log2_32(Amount);
5028 if (ShiftAmount == 0)
5029 return;
5030 BuildMI(MBB, II, DL, get(RISCV::SLLI), DestReg)
5031 .addReg(DestReg, RegState::Kill)
5032 .addImm(ShiftAmount)
5033 .setMIFlag(Flag);
5034 } else if (int ShXAmount, ShiftAmount;
5035 STI.hasShlAdd(3) &&
5036 (ShXAmount = isShifted359(Amount, ShiftAmount)) != 0) {
5037 // We can use Zba SHXADD+SLLI instructions for multiply in some cases.
5038 unsigned Opc;
5039 switch (ShXAmount) {
5040 case 1:
5041 Opc = RISCV::SH1ADD;
5042 break;
5043 case 2:
5044 Opc = RISCV::SH2ADD;
5045 break;
5046 case 3:
5047 Opc = RISCV::SH3ADD;
5048 break;
5049 default:
5050 llvm_unreachable("unexpected result of isShifted359");
5051 }
5052 if (ShiftAmount)
5053 BuildMI(MBB, II, DL, get(RISCV::SLLI), DestReg)
5054 .addReg(DestReg, RegState::Kill)
5055 .addImm(ShiftAmount)
5056 .setMIFlag(Flag);
5057 BuildMI(MBB, II, DL, get(Opc), DestReg)
5058 .addReg(DestReg, RegState::Kill)
5059 .addReg(DestReg)
5060 .setMIFlag(Flag);
5061 } else if (llvm::has_single_bit(Amount - 1)) {
5062 Register ScaledRegister = MRI.createVirtualRegister(&RISCV::GPRRegClass);
5063 uint32_t ShiftAmount = Log2_32(Amount - 1);
5064 BuildMI(MBB, II, DL, get(RISCV::SLLI), ScaledRegister)
5065 .addReg(DestReg)
5066 .addImm(ShiftAmount)
5067 .setMIFlag(Flag);
5068 BuildMI(MBB, II, DL, get(RISCV::ADD), DestReg)
5069 .addReg(ScaledRegister, RegState::Kill)
5070 .addReg(DestReg, RegState::Kill)
5071 .setMIFlag(Flag);
5072 } else if (llvm::has_single_bit(Amount + 1)) {
5073 Register ScaledRegister = MRI.createVirtualRegister(&RISCV::GPRRegClass);
5074 uint32_t ShiftAmount = Log2_32(Amount + 1);
5075 BuildMI(MBB, II, DL, get(RISCV::SLLI), ScaledRegister)
5076 .addReg(DestReg)
5077 .addImm(ShiftAmount)
5078 .setMIFlag(Flag);
5079 BuildMI(MBB, II, DL, get(RISCV::SUB), DestReg)
5080 .addReg(ScaledRegister, RegState::Kill)
5081 .addReg(DestReg, RegState::Kill)
5082 .setMIFlag(Flag);
5083 } else if (STI.hasStdExtZmmul()) {
5084 Register N = MRI.createVirtualRegister(&RISCV::GPRRegClass);
5085 movImm(MBB, II, DL, N, Amount, Flag);
5086 BuildMI(MBB, II, DL, get(RISCV::MUL), DestReg)
5087 .addReg(DestReg, RegState::Kill)
5089 .setMIFlag(Flag);
5090 } else {
5091 Register Acc;
5092 uint32_t PrevShiftAmount = 0;
5093 for (uint32_t ShiftAmount = 0; Amount >> ShiftAmount; ShiftAmount++) {
5094 if (Amount & (1U << ShiftAmount)) {
5095 if (ShiftAmount)
5096 BuildMI(MBB, II, DL, get(RISCV::SLLI), DestReg)
5097 .addReg(DestReg, RegState::Kill)
5098 .addImm(ShiftAmount - PrevShiftAmount)
5099 .setMIFlag(Flag);
5100 if (Amount >> (ShiftAmount + 1)) {
5101 // If we don't have an accmulator yet, create it and copy DestReg.
5102 if (!Acc) {
5103 Acc = MRI.createVirtualRegister(&RISCV::GPRRegClass);
5104 BuildMI(MBB, II, DL, get(TargetOpcode::COPY), Acc)
5105 .addReg(DestReg)
5106 .setMIFlag(Flag);
5107 } else {
5108 BuildMI(MBB, II, DL, get(RISCV::ADD), Acc)
5109 .addReg(Acc, RegState::Kill)
5110 .addReg(DestReg)
5111 .setMIFlag(Flag);
5112 }
5113 }
5114 PrevShiftAmount = ShiftAmount;
5115 }
5116 }
5117 assert(Acc && "Expected valid accumulator");
5118 BuildMI(MBB, II, DL, get(RISCV::ADD), DestReg)
5119 .addReg(DestReg, RegState::Kill)
5120 .addReg(Acc, RegState::Kill)
5121 .setMIFlag(Flag);
5122 }
5123}
5124
5127 static const std::pair<MachineMemOperand::Flags, const char *> TargetFlags[] =
5128 {{MONontemporalBit0, "riscv-nontemporal-domain-bit-0"},
5129 {MONontemporalBit1, "riscv-nontemporal-domain-bit-1"}};
5130 return ArrayRef(TargetFlags);
5131}
5132
5134 return OptLevel >= CodeGenOptLevel::Aggressive
5135 ? STI.getTailDupAggressiveThreshold()
5136 : 2;
5137}
5138
5140 // RVV lacks any support for immediate addressing for stack addresses, so be
5141 // conservative.
5142 unsigned Opcode = MI.getOpcode();
5143 if (!RISCVVPseudosTable::getPseudoInfo(Opcode) &&
5145 return false;
5146 return true;
5147}
5148
5149/// Return true if \p MI is a copy that will be lowered to one or more vmvNr.vs.
5151 const MachineInstr &MI) {
5152 return MI.isCopy() && MI.getOperand(0).getReg().isPhysical() &&
5154 TRI->getMinimalPhysRegClass(MI.getOperand(0).getReg()));
5155}
5156
5157std::optional<std::pair<unsigned, unsigned>>
5159 switch (Opcode) {
5160 default:
5161 return std::nullopt;
5162 case RISCV::PseudoVSPILL2_M1:
5163 case RISCV::PseudoVRELOAD2_M1:
5164 return std::make_pair(2u, 1u);
5165 case RISCV::PseudoVSPILL2_M2:
5166 case RISCV::PseudoVRELOAD2_M2:
5167 return std::make_pair(2u, 2u);
5168 case RISCV::PseudoVSPILL2_M4:
5169 case RISCV::PseudoVRELOAD2_M4:
5170 return std::make_pair(2u, 4u);
5171 case RISCV::PseudoVSPILL3_M1:
5172 case RISCV::PseudoVRELOAD3_M1:
5173 return std::make_pair(3u, 1u);
5174 case RISCV::PseudoVSPILL3_M2:
5175 case RISCV::PseudoVRELOAD3_M2:
5176 return std::make_pair(3u, 2u);
5177 case RISCV::PseudoVSPILL4_M1:
5178 case RISCV::PseudoVRELOAD4_M1:
5179 return std::make_pair(4u, 1u);
5180 case RISCV::PseudoVSPILL4_M2:
5181 case RISCV::PseudoVRELOAD4_M2:
5182 return std::make_pair(4u, 2u);
5183 case RISCV::PseudoVSPILL5_M1:
5184 case RISCV::PseudoVRELOAD5_M1:
5185 return std::make_pair(5u, 1u);
5186 case RISCV::PseudoVSPILL6_M1:
5187 case RISCV::PseudoVRELOAD6_M1:
5188 return std::make_pair(6u, 1u);
5189 case RISCV::PseudoVSPILL7_M1:
5190 case RISCV::PseudoVRELOAD7_M1:
5191 return std::make_pair(7u, 1u);
5192 case RISCV::PseudoVSPILL8_M1:
5193 case RISCV::PseudoVRELOAD8_M1:
5194 return std::make_pair(8u, 1u);
5195 }
5196}
5197
5198bool RISCV::hasEqualFRM(const MachineInstr &MI1, const MachineInstr &MI2) {
5199 int16_t MI1FrmOpIdx =
5200 RISCV::getNamedOperandIdx(MI1.getOpcode(), RISCV::OpName::frm);
5201 int16_t MI2FrmOpIdx =
5202 RISCV::getNamedOperandIdx(MI2.getOpcode(), RISCV::OpName::frm);
5203 if (MI1FrmOpIdx < 0 || MI2FrmOpIdx < 0)
5204 return false;
5205 MachineOperand FrmOp1 = MI1.getOperand(MI1FrmOpIdx);
5206 MachineOperand FrmOp2 = MI2.getOperand(MI2FrmOpIdx);
5207 return FrmOp1.getImm() == FrmOp2.getImm();
5208}
5209
5210std::optional<unsigned>
5211RISCV::getVectorLowDemandedScalarBits(unsigned Opcode, unsigned Log2SEW) {
5212 switch (Opcode) {
5213 default:
5214 return std::nullopt;
5215
5216 // 11.6. Vector Single-Width Shift Instructions
5217 case RISCV::VSLL_VX:
5218 case RISCV::VSRL_VX:
5219 case RISCV::VSRA_VX:
5220 // 12.4. Vector Single-Width Scaling Shift Instructions
5221 case RISCV::VSSRL_VX:
5222 case RISCV::VSSRA_VX:
5223 // Zvbb
5224 case RISCV::VROL_VX:
5225 case RISCV::VROR_VX:
5226 // Only the low lg2(SEW) bits of the shift-amount value are used.
5227 return Log2SEW;
5228
5229 // 11.7 Vector Narrowing Integer Right Shift Instructions
5230 case RISCV::VNSRL_WX:
5231 case RISCV::VNSRA_WX:
5232 // 12.5. Vector Narrowing Fixed-Point Clip Instructions
5233 case RISCV::VNCLIPU_WX:
5234 case RISCV::VNCLIP_WX:
5235 // Zvbb
5236 case RISCV::VWSLL_VX:
5237 // Only the low lg2(2*SEW) bits of the shift-amount value are used.
5238 return Log2SEW + 1;
5239
5240 // 11.1. Vector Single-Width Integer Add and Subtract
5241 case RISCV::VADD_VX:
5242 case RISCV::VSUB_VX:
5243 case RISCV::VRSUB_VX:
5244 // 11.2. Vector Widening Integer Add/Subtract
5245 case RISCV::VWADDU_VX:
5246 case RISCV::VWSUBU_VX:
5247 case RISCV::VWADD_VX:
5248 case RISCV::VWSUB_VX:
5249 case RISCV::VWADDU_WX:
5250 case RISCV::VWSUBU_WX:
5251 case RISCV::VWADD_WX:
5252 case RISCV::VWSUB_WX:
5253 // 11.4. Vector Integer Add-with-Carry / Subtract-with-Borrow Instructions
5254 case RISCV::VADC_VXM:
5255 case RISCV::VADC_VIM:
5256 case RISCV::VMADC_VXM:
5257 case RISCV::VMADC_VIM:
5258 case RISCV::VMADC_VX:
5259 case RISCV::VSBC_VXM:
5260 case RISCV::VMSBC_VXM:
5261 case RISCV::VMSBC_VX:
5262 // 11.5 Vector Bitwise Logical Instructions
5263 case RISCV::VAND_VX:
5264 case RISCV::VOR_VX:
5265 case RISCV::VXOR_VX:
5266 // 11.8. Vector Integer Compare Instructions
5267 case RISCV::VMSEQ_VX:
5268 case RISCV::VMSNE_VX:
5269 case RISCV::VMSLTU_VX:
5270 case RISCV::VMSLT_VX:
5271 case RISCV::VMSLEU_VX:
5272 case RISCV::VMSLE_VX:
5273 case RISCV::VMSGTU_VX:
5274 case RISCV::VMSGT_VX:
5275 // 11.9. Vector Integer Min/Max Instructions
5276 case RISCV::VMINU_VX:
5277 case RISCV::VMIN_VX:
5278 case RISCV::VMAXU_VX:
5279 case RISCV::VMAX_VX:
5280 // 11.10. Vector Single-Width Integer Multiply Instructions
5281 case RISCV::VMUL_VX:
5282 case RISCV::VMULH_VX:
5283 case RISCV::VMULHU_VX:
5284 case RISCV::VMULHSU_VX:
5285 // 11.11. Vector Integer Divide Instructions
5286 case RISCV::VDIVU_VX:
5287 case RISCV::VDIV_VX:
5288 case RISCV::VREMU_VX:
5289 case RISCV::VREM_VX:
5290 // 11.12. Vector Widening Integer Multiply Instructions
5291 case RISCV::VWMUL_VX:
5292 case RISCV::VWMULU_VX:
5293 case RISCV::VWMULSU_VX:
5294 // 11.13. Vector Single-Width Integer Multiply-Add Instructions
5295 case RISCV::VMACC_VX:
5296 case RISCV::VNMSAC_VX:
5297 case RISCV::VMADD_VX:
5298 case RISCV::VNMSUB_VX:
5299 // 11.14. Vector Widening Integer Multiply-Add Instructions
5300 case RISCV::VWMACCU_VX:
5301 case RISCV::VWMACC_VX:
5302 case RISCV::VWMACCSU_VX:
5303 case RISCV::VWMACCUS_VX:
5304 // 11.15. Vector Integer Merge Instructions
5305 case RISCV::VMERGE_VXM:
5306 // 11.16. Vector Integer Move Instructions
5307 case RISCV::VMV_V_X:
5308 // 12.1. Vector Single-Width Saturating Add and Subtract
5309 case RISCV::VSADDU_VX:
5310 case RISCV::VSADD_VX:
5311 case RISCV::VSSUBU_VX:
5312 case RISCV::VSSUB_VX:
5313 // 12.2. Vector Single-Width Averaging Add and Subtract
5314 case RISCV::VAADDU_VX:
5315 case RISCV::VAADD_VX:
5316 case RISCV::VASUBU_VX:
5317 case RISCV::VASUB_VX:
5318 // 12.3. Vector Single-Width Fractional Multiply with Rounding and Saturation
5319 case RISCV::VSMUL_VX:
5320 // 16.1. Integer Scalar Move Instructions
5321 case RISCV::VMV_S_X:
5322 // Zvbb
5323 case RISCV::VANDN_VX:
5324 return 1U << Log2SEW;
5325 }
5326}
5327
5328unsigned RISCV::getRVVMCOpcode(unsigned RVVPseudoOpcode) {
5330 RISCVVPseudosTable::getPseudoInfo(RVVPseudoOpcode);
5331 if (!RVV)
5332 return 0;
5333 return RVV->BaseInstr;
5334}
5335
5336unsigned RISCV::getDestLog2EEW(const MCInstrDesc &Desc, unsigned Log2SEW) {
5337 unsigned DestEEW =
5339 // EEW = 1
5340 if (DestEEW == 0)
5341 return 0;
5342 // EEW = SEW * n
5343 unsigned Scaled = Log2SEW + (DestEEW - 1);
5344 assert(Scaled >= 3 && Scaled <= 6);
5345 return Scaled;
5346}
5347
5348static std::optional<int64_t> getEffectiveImm(const MachineRegisterInfo &MRI,
5349 const MachineOperand &MO) {
5350 assert(MO.isImm() || MO.getReg().isVirtual());
5351 if (MO.isImm())
5352 return MO.getImm();
5353 const MachineInstr *Def = MRI.getVRegDef(MO.getReg());
5354 int64_t Imm;
5355 if (isLoadImm(Def, Imm))
5356 return Imm;
5357 return std::nullopt;
5358}
5359
5360/// Given two VL operands, do we know that LHS <= RHS? Must be used in SSA form.
5362 const MachineOperand &LHS, const MachineOperand &RHS) {
5363 assert((LHS.isImm() || MRI.isSSA()) && (RHS.isImm() || MRI.isSSA()));
5364 if (LHS.isReg() && RHS.isReg() && LHS.getReg().isVirtual() &&
5365 LHS.getReg() == RHS.getReg())
5366 return true;
5367 if (RHS.isImm() && RHS.getImm() == RISCV::VLMaxSentinel)
5368 return true;
5369 if (LHS.isImm() && LHS.getImm() == 0)
5370 return true;
5371 if (LHS.isImm() && LHS.getImm() == RISCV::VLMaxSentinel)
5372 return false;
5373 std::optional<int64_t> LHSImm = getEffectiveImm(MRI, LHS),
5374 RHSImm = getEffectiveImm(MRI, RHS);
5375 if (!LHSImm || !RHSImm)
5376 return false;
5377 return LHSImm <= RHSImm;
5378}
5379
5380namespace {
5381class RISCVPipelinerLoopInfo : public TargetInstrInfo::PipelinerLoopInfo {
5382 const MachineInstr *LHS;
5383 const MachineInstr *RHS;
5385
5386public:
5387 RISCVPipelinerLoopInfo(const MachineInstr *LHS, const MachineInstr *RHS,
5389 : LHS(LHS), RHS(RHS), Cond(Cond.begin(), Cond.end()) {}
5390
5391 bool shouldIgnoreForPipelining(const MachineInstr *MI) const override {
5392 // Make the instructions for loop control be placed in stage 0.
5393 // The predecessors of LHS/RHS are considered by the caller.
5394 if (LHS && MI == LHS)
5395 return true;
5396 if (RHS && MI == RHS)
5397 return true;
5398 return false;
5399 }
5400
5401 std::optional<bool> createTripCountGreaterCondition(
5402 int TC, MachineBasicBlock &MBB,
5403 SmallVectorImpl<MachineOperand> &CondParam) override {
5404 // A branch instruction will be inserted as "if (Cond) goto epilogue".
5405 // Cond is normalized for such use.
5406 // The predecessors of the branch are assumed to have already been inserted.
5407 CondParam = Cond;
5408 return {};
5409 }
5410
5411 void setPreheader(MachineBasicBlock *NewPreheader) override {}
5412
5413 void adjustTripCount(int TripCountAdjust) override {}
5414};
5415} // namespace
5416
5417std::unique_ptr<TargetInstrInfo::PipelinerLoopInfo>
5419 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
5421 if (analyzeBranch(*LoopBB, TBB, FBB, Cond, /*AllowModify=*/false))
5422 return nullptr;
5423
5424 // Infinite loops are not supported
5425 if (TBB == LoopBB && FBB == LoopBB)
5426 return nullptr;
5427
5428 // Must be conditional branch
5429 if (FBB == nullptr)
5430 return nullptr;
5431
5432 assert((TBB == LoopBB || FBB == LoopBB) &&
5433 "The Loop must be a single-basic-block loop");
5434
5435 // Normalization for createTripCountGreaterCondition()
5436 if (TBB == LoopBB)
5438
5439 const MachineRegisterInfo &MRI = LoopBB->getParent()->getRegInfo();
5440 auto FindRegDef = [&MRI](MachineOperand &Op) -> const MachineInstr * {
5441 if (!Op.isReg())
5442 return nullptr;
5443 Register Reg = Op.getReg();
5444 if (!Reg.isVirtual())
5445 return nullptr;
5446 return MRI.getVRegDef(Reg);
5447 };
5448
5449 const MachineInstr *LHS = FindRegDef(Cond[1]);
5450 const MachineInstr *RHS = FindRegDef(Cond[2]);
5451 if (LHS && LHS->isPHI())
5452 return nullptr;
5453 if (RHS && RHS->isPHI())
5454 return nullptr;
5455
5456 return std::make_unique<RISCVPipelinerLoopInfo>(LHS, RHS, Cond);
5457}
5458
5459// FIXME: We should remove this if we have a default generic scheduling model.
5461 unsigned RVVMCOpcode = RISCV::getRVVMCOpcode(Opc);
5462 Opc = RVVMCOpcode ? RVVMCOpcode : Opc;
5463 switch (Opc) {
5464 default:
5465 return false;
5466 // Integer div/rem.
5467 case RISCV::DIV:
5468 case RISCV::DIVW:
5469 case RISCV::DIVU:
5470 case RISCV::DIVUW:
5471 case RISCV::REM:
5472 case RISCV::REMW:
5473 case RISCV::REMU:
5474 case RISCV::REMUW:
5475 // Floating-point div/sqrt.
5476 case RISCV::FDIV_H:
5477 case RISCV::FDIV_S:
5478 case RISCV::FDIV_D:
5479 case RISCV::FDIV_H_INX:
5480 case RISCV::FDIV_S_INX:
5481 case RISCV::FDIV_D_INX:
5482 case RISCV::FDIV_D_IN32X:
5483 case RISCV::FSQRT_H:
5484 case RISCV::FSQRT_S:
5485 case RISCV::FSQRT_D:
5486 case RISCV::FSQRT_H_INX:
5487 case RISCV::FSQRT_S_INX:
5488 case RISCV::FSQRT_D_INX:
5489 case RISCV::FSQRT_D_IN32X:
5490 // Vector integer div/rem
5491 case RISCV::VDIV_VV:
5492 case RISCV::VDIV_VX:
5493 case RISCV::VDIVU_VV:
5494 case RISCV::VDIVU_VX:
5495 case RISCV::VREM_VV:
5496 case RISCV::VREM_VX:
5497 case RISCV::VREMU_VV:
5498 case RISCV::VREMU_VX:
5499 // Vector floating-point div/sqrt.
5500 case RISCV::VFDIV_VV:
5501 case RISCV::VFDIV_VF:
5502 case RISCV::VFRDIV_VF:
5503 case RISCV::VFSQRT_V:
5504 case RISCV::VFRSQRT7_V:
5505 return true;
5506 }
5507}
5508
5509bool RISCVInstrInfo::isVRegCopy(const MachineInstr *MI, unsigned LMul) const {
5510 if (MI->getOpcode() != TargetOpcode::COPY)
5511 return false;
5512 const MachineRegisterInfo &MRI = MI->getMF()->getRegInfo();
5514
5515 Register DstReg = MI->getOperand(0).getReg();
5516 const TargetRegisterClass *RC = DstReg.isVirtual()
5517 ? MRI.getRegClass(DstReg)
5518 : TRI->getMinimalPhysRegClass(DstReg);
5519
5521 return false;
5522
5523 if (!LMul)
5524 return true;
5525
5526 // TODO: Perhaps we could distinguish segment register classes (e.g. VRN3M2)
5527 // in the future.
5528 auto [RCLMul, RCFractional] =
5530 return (!RCFractional && LMul == RCLMul) || (RCFractional && LMul == 1);
5531}
5532
5534 if (MI.memoperands_empty())
5535 return false;
5536
5537 MachineMemOperand *MMO = *(MI.memoperands_begin());
5538 if (!MMO->isNonTemporal())
5539 return false;
5540
5541 return true;
5542}
5543
5545 const MachineBasicBlock::iterator &To) {
5546 assert(To == From.getParent()->end() || From.getParent() == To->getParent());
5547 SmallVector<Register> PhysUses, PhysDefs;
5548 for (const MachineOperand &MO : From.all_uses())
5549 if (MO.getReg().isPhysical())
5550 PhysUses.push_back(MO.getReg());
5551 for (const MachineOperand &MO : From.all_defs())
5552 if (MO.getReg().isPhysical())
5553 PhysDefs.push_back(MO.getReg());
5554 bool SawStore = false;
5555 for (auto II = std::next(From.getIterator()); II != To; II++) {
5556 for (Register PhysReg : PhysUses)
5557 if (II->definesRegister(PhysReg, nullptr))
5558 return false;
5559 for (Register PhysReg : PhysDefs)
5560 if (II->definesRegister(PhysReg, nullptr) ||
5561 II->readsRegister(PhysReg, nullptr))
5562 return false;
5563 II->isSafeToMove(SawStore);
5564 if (SawStore)
5565 break;
5566 }
5567 return From.isSafeToMove(SawStore);
5568}
MachineInstrBuilder MachineInstrBuilder & DefMI
static void parseCondBranch(MachineInstr *LastInst, MachineBasicBlock *&Target, SmallVectorImpl< MachineOperand > &Cond)
@ MachineOutlinerTailCall
Emit a save, restore, call, and return.
@ MachineOutlinerRegSave
Emit a call and tail-call.
@ MachineOutlinerDefault
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
SmallVector< int16_t, MAX_SRC_OPERANDS_NUM > OperandIndices
unsigned Imm
unsigned uint64_t
@ Scaled
MachineBasicBlock & MBB
MachineBasicBlock MachineBasicBlock::iterator DebugLoc DL
MachineBasicBlock MachineBasicBlock::iterator MBBI
basic Basic Alias true
#define X(NUM, ENUM, NAME)
Definition ELF.h:857
static GCRegistry::Add< ShadowStackGC > C("shadow-stack", "Very portable GC for uncooperative code generators")
#define clEnumValN(ENUMVAL, FLAGNAME, DESC)
const HexagonInstrInfo * TII
#define _
IRTranslator LLVM IR MI
Module.h This file contains the declarations for the Module class.
const size_t AbstractManglingParser< Derived, Alloc >::NumOps
const AbstractManglingParser< Derived, Alloc >::OperatorInfo AbstractManglingParser< Derived, Alloc >::Ops[]
#define F(x, y, z)
Definition MD5.cpp:54
#define I(x, y, z)
Definition MD5.cpp:57
Register Reg
Register const TargetRegisterInfo * TRI
Promote Memory to Register
Definition Mem2Reg.cpp:110
This file provides utility analysis objects describing memory locations.
uint64_t IntrinsicInst * II
static std::optional< int64_t > getEffectiveImm(const MachineRegisterInfo &MRI, const MachineOperand &MO)
static bool cannotInsertTailCall(const MachineBasicBlock &MBB)
#define CASE_VFMA_CHANGE_OPCODE_SPLATS(OLDOP, NEWOP)
#define CASE_FP_WIDEOP_CHANGE_OPCODE_LMULS_ALT(OP)
#define CASE_FP_WIDEOP_OPCODE_LMULS(OP)
#define CASE_OPERAND_SIMM(NUM)
static std::optional< unsigned > getLMULForRVVWholeLoadStore(unsigned Opcode)
#define CASE_VFMA_CHANGE_OPCODE_VV(OLDOP, NEWOP)
static unsigned getFPFusedMultiplyOpcode(unsigned RootOpc, unsigned Pattern)
std::optional< unsigned > getFoldedOpcode(MachineFunction &MF, MachineInstr &MI, ArrayRef< unsigned > Ops, const RISCVSubtarget &ST)
#define RVV_OPC_LMUL_CASE(OPC, INV)
#define CASE_FP_WIDEOP_CHANGE_OPCODE_LMULS(OP)
static bool forwardCopyWillClobberTuple(unsigned DstReg, unsigned SrcReg, unsigned NumRegs)
static void combineFPFusedMultiply(MachineInstr &Root, MachineInstr &Prev, unsigned Pattern, SmallVectorImpl< MachineInstr * > &InsInstrs, SmallVectorImpl< MachineInstr * > &DelInstrs)
static unsigned getAddendOperandIdx(unsigned Pattern)
#define CASE_RVV_OPCODE_UNMASK(OP)
#define CASE_WIDEOP_CHANGE_OPCODE_LMULS(OP)
static cl::opt< bool > PreferWholeRegisterMove("riscv-prefer-whole-register-move", cl::init(false), cl::Hidden, cl::desc("Prefer whole register move for vector registers."))
#define CASE_VFMA_SPLATS(OP)
unsigned getPredicatedOpcode(unsigned Opcode)
#define CASE_FP_WIDEOP_OPCODE_LMULS_ALT(OP)
#define CASE_WIDEOP_OPCODE_LMULS(OP)
static bool isMIReadsReg(const MachineInstr &MI, const TargetRegisterInfo *TRI, MCRegister RegNo)
#define OPCODE_LMUL_MASK_CASE(OPC)
#define CASE_OPERAND_UIMM_LSB_ZEROS(BITS, SUFFIX)
static bool isFSUB(unsigned Opc)
#define CASE_VMA_CHANGE_OPCODE_LMULS(OLDOP, NEWOP, TYPE)
#define CASE_RVV_OPCODE(OP)
#define CASE_VFMA_OPCODE_VV(OP)
static cl::opt< bool > OutlinerEnableRegSave("riscv-outliner-regsave", cl::init(true), cl::Hidden, cl::desc("Enable RegSave strategy in machine outliner (save X5 to a " "temporary register when X5 is live across outlined calls)."))
MachineOutlinerConstructionID
#define CASE_RVV_OPCODE_WIDEN(OP)
static unsigned getLoadPredicatedOpcode(unsigned Opcode)
static unsigned getSHXADDUWShiftAmount(unsigned Opc)
#define CASE_VMA_OPCODE_LMULS(OP, TYPE)
static bool isConvertibleToVMV_V_V(const RISCVSubtarget &STI, const MachineBasicBlock &MBB, MachineBasicBlock::const_iterator MBBI, MachineBasicBlock::const_iterator &DefMBBI, RISCVVType::VLMUL LMul)
static bool isFMUL(unsigned Opc)
static unsigned getInverseXqcicmOpcode(unsigned Opcode)
static bool getFPPatterns(MachineInstr &Root, SmallVectorImpl< unsigned > &Patterns, bool DoRegPressureReduce)
#define OPCODE_LMUL_CASE(OPC)
#define CASE_OPERAND_UIMM(NUM)
static Register findRegisterToSaveX5To(outliner::Candidate &C, const TargetRegisterInfo &TRI)
static bool canCombineShiftIntoShXAdd(const MachineBasicBlock &MBB, const MachineOperand &MO, unsigned OuterShiftAmt)
Utility routine that checks if.
static bool isCandidatePatchable(const MachineBasicBlock &MBB)
static bool isFADD(unsigned Opc)
static void genShXAddAddShift(MachineInstr &Root, unsigned AddOpIdx, SmallVectorImpl< MachineInstr * > &InsInstrs, SmallVectorImpl< MachineInstr * > &DelInstrs, DenseMap< Register, unsigned > &InstrIdxForVirtReg)
static bool isLoadImm(const MachineInstr *MI, int64_t &Imm)
static bool isMIModifiesReg(const MachineInstr &MI, const TargetRegisterInfo *TRI, MCRegister RegNo)
#define CASE_RVV_OPCODE_LMUL(OP, LMUL)
static bool canCombineFPFusedMultiply(const MachineInstr &Root, const MachineOperand &MO, bool DoRegPressureReduce)
static bool getSHXADDPatterns(const MachineInstr &Root, SmallVectorImpl< unsigned > &Patterns)
static bool getFPFusedMultiplyPatterns(MachineInstr &Root, SmallVectorImpl< unsigned > &Patterns, bool DoRegPressureReduce)
static cl::opt< MachineTraceStrategy > ForceMachineCombinerStrategy("riscv-force-machine-combiner-strategy", cl::Hidden, cl::desc("Force machine combiner to use a specific strategy for machine " "trace metrics evaluation."), cl::init(MachineTraceStrategy::TS_NumStrategies), cl::values(clEnumValN(MachineTraceStrategy::TS_Local, "local", "Local strategy."), clEnumValN(MachineTraceStrategy::TS_MinInstrCount, "min-instr", "MinInstrCount strategy.")))
static unsigned getSHXADDShiftAmount(unsigned Opc)
#define CASE_RVV_OPCODE_MASK(OP)
#define RVV_OPC_LMUL_MASK_CASE(OPC, INV)
static MachineInstr * canFoldAsPredicatedOp(Register Reg, const MachineRegisterInfo &MRI, const TargetInstrInfo *TII, const RISCVSubtarget &STI)
Identify instructions that can be folded into a CCMOV instruction, and return the defining instructio...
const SmallVectorImpl< MachineOperand > MachineBasicBlock * TBB
const SmallVectorImpl< MachineOperand > & Cond
This file declares the machine register scavenger class.
static bool memOpsHaveSameBasePtr(const MachineInstr &MI1, ArrayRef< const MachineOperand * > BaseOps1, const MachineInstr &MI2, ArrayRef< const MachineOperand * > BaseOps2)
This file contains some templates that are useful if you are working with the STL at all.
static bool contains(SmallPtrSetImpl< ConstantExpr * > &Cache, ConstantExpr *Expr, Constant *C)
Definition Value.cpp:484
This file defines the SmallVector class.
This file defines the 'Statistic' class, which is designed to be an easy way to expose various metric...
#define STATISTIC(VARNAME, DESC)
Definition Statistic.h:171
#define LLVM_DEBUG(...)
Definition Debug.h:119
static TableGen::Emitter::Opt Y("gen-skeleton-entry", EmitSkeleton, "Generate example skeleton entry")
static bool canCombine(MachineBasicBlock &MBB, MachineOperand &MO, unsigned CombineOpc=0)
static cl::opt< unsigned > CacheLineSize("cache-line-size", cl::init(0), cl::Hidden, cl::desc("Use this to override the target cache line size when " "specified by the user."))
Value * RHS
Value * LHS
Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition ArrayRef.h:40
const T & front() const
Get the first element.
Definition ArrayRef.h:144
bool empty() const
Check if the array is empty.
Definition ArrayRef.h:136
static LLVM_ABI DILocation * getMergedLocation(DILocation *LocA, DILocation *LocB)
Attempts to merge LocA and LocB into a single location; see DebugLoc::getMergedLocation for more deta...
bool isBigEndian() const
Definition DataLayout.h:218
A debug info location.
Definition DebugLoc.h:126
std::pair< iterator, bool > insert(const std::pair< KeyT, ValueT > &KV)
Definition DenseMap.h:312
bool hasMinSize() const
Optimize this function for minimum size (-Oz).
Definition Function.h:696
LiveInterval - This class represents the liveness of a register, or stack slot.
LiveInterval & getInterval(Register Reg)
SlotIndex ReplaceMachineInstrInMaps(MachineInstr &MI, MachineInstr &NewMI)
const Segment * getSegmentContaining(SlotIndex Idx) const
Return the segment that contains the specified index, or null if there is none.
LLVM_ABI void replaceKillInstruction(Register Reg, MachineInstr &OldMI, MachineInstr &NewMI)
replaceKillInstruction - Update register kill info by replacing a kill instruction with a new one.
bool hasValue() const
static LocationSize precise(uint64_t Value)
TypeSize getValue() const
MCInstBuilder & addReg(MCRegister Reg)
Add a new register operand.
MCInstBuilder & addImm(int64_t Val)
Add a new integer immediate operand.
Instances of this class represent a single low-level machine instruction.
Definition MCInst.h:188
Describe properties that are true of each instruction in the target description file.
unsigned getNumOperands() const
Return the number of declared MachineOperands for this MachineInstruction.
bool isConditionalBranch() const
Return true if this is a branch which may fall through to the next instruction or may transfer contro...
This holds information about one operand of a machine instruction, indicating the register class for ...
Definition MCInstrDesc.h:88
const uint8_t TSFlags
Configurable target specific flags.
Wrapper class representing physical registers. Should be passed by value.
Definition MCRegister.h:41
const FeatureBitset & getFeatureBits() const
Set of metadata that should be preserved when using BuildMI().
MachineInstrBundleIterator< const MachineInstr > const_iterator
MachineInstrBundleIterator< MachineInstr, true > reverse_iterator
const MachineFunction * getParent() const
Return the MachineFunction containing this basic block.
MachineInstrBundleIterator< MachineInstr > iterator
MachineInstrBundleIterator< const MachineInstr, true > const_reverse_iterator
The MachineFrameInfo class represents an abstract stack frame until prolog/epilog code is inserted.
void setStackID(int ObjectIdx, uint8_t ID)
Align getObjectAlign(int ObjectIdx) const
Return the alignment of the specified stack object.
int64_t getObjectSize(int ObjectIdx) const
Return the size of the specified object.
const TargetSubtargetInfo & getSubtarget() const
getSubtarget - Return the subtarget for which this machine code is being compiled.
StringRef getName() const
getName - Return the name of the corresponding LLVM function.
MachineFrameInfo & getFrameInfo()
getFrameInfo - Return the frame info object for the current function.
MachineRegisterInfo & getRegInfo()
getRegInfo - Return information about the registers currently in use.
const DataLayout & getDataLayout() const
Return the DataLayout attached to the Module associated to this MF.
Function & getFunction()
Return the LLVM function that this machine code represents.
Ty * getInfo()
getInfo - Keep track of various per-function pieces of information for backends that would like to do...
MachineMemOperand * getMachineMemOperand(MachinePointerInfo PtrInfo, MachineMemOperand::Flags F, LLT MemTy, Align BaseAlignment, const MMOMetadata &Metadata=MMOMetadata(), SyncScope::ID SSID=SyncScope::System, AtomicOrdering Ordering=AtomicOrdering::NotAtomic, AtomicOrdering FailureOrdering=AtomicOrdering::NotAtomic)
getMachineMemOperand - Allocate a new MachineMemOperand.
const TargetMachine & getTarget() const
getTarget - Return the target machine this machine code is compiled with
const MachineInstrBuilder & setMemRefs(ArrayRef< MachineMemOperand * > MMOs) const
const MachineInstrBuilder & addUse(Register RegNo, RegState Flags={}, unsigned SubReg=0) const
Add a virtual register use operand.
const MachineInstrBuilder & addReg(Register RegNo, RegState Flags={}, unsigned SubReg=0) const
Add a new virtual register operand.
const MachineInstrBuilder & setMIFlag(MachineInstr::MIFlag Flag) const
const MachineInstrBuilder & addImm(int64_t Val) const
Add a new immediate operand.
const MachineInstrBuilder & add(const MachineOperand &MO) const
const MachineInstrBuilder & addFrameIndex(int Idx) const
const MachineInstrBuilder & addMBB(MachineBasicBlock *MBB, unsigned TargetFlags=0) const
const MachineInstrBuilder & cloneMemRefs(const MachineInstr &OtherMI) const
const MachineInstrBuilder & setMIFlags(unsigned Flags) const
const MachineInstrBuilder & copyImplicitOps(const MachineInstr &OtherMI) const
Copy all the implicit operands from OtherMI onto this one.
const MachineInstrBuilder & addMemOperand(MachineMemOperand *MMO) const
reverse_iterator getReverse() const
Get a reverse iterator to the same node.
Representation of each machine instruction.
unsigned getOpcode() const
Returns the opcode of this MachineInstr.
bool isReturn(QueryType Type=AnyInBundle) const
bool mayLoadOrStore(QueryType Type=AnyInBundle) const
Return true if this instruction could possibly read or modify memory.
const MachineBasicBlock * getParent() const
filtered_mop_range all_defs()
Returns an iterator range over all operands that are (explicit or implicit) register defs.
bool getFlag(MIFlag Flag) const
Return whether an MI flag is set.
LLVM_ABI bool isSafeToMove(bool &SawStore) const
Return true if it is safe to move this instruction.
LLVM_ABI unsigned getNumExplicitOperands() const
Returns the number of non-implicit operands.
bool modifiesRegister(Register Reg, const TargetRegisterInfo *TRI) const
Return true if the MachineInstr modifies (fully define or partially define) the specified register.
bool mayLoad(QueryType Type=AnyInBundle) const
Return true if this instruction could possibly read memory.
const MCInstrDesc & getDesc() const
Returns the target instruction descriptor of this MachineInstr.
LLVM_ABI bool hasUnmodeledSideEffects() const
Return true if this instruction has side effects that are not modeled by mayLoad / mayStore,...
bool hasOneMemOperand() const
Return true if this instruction has exactly one MachineMemOperand.
mmo_iterator memoperands_begin() const
Access to memory operands of the instruction.
LLVM_ABI bool hasOrderedMemoryRef() const
Return true if this instruction may have an ordered or volatile memory reference, or if the informati...
LLVM_ABI const MachineFunction * getMF() const
Return the function that contains the basic block that this instruction belongs to.
ArrayRef< MachineMemOperand * > memoperands() const
Access to memory operands of the instruction.
const DebugLoc & getDebugLoc() const
Returns the debug location id of this MachineInstr.
filtered_mop_range all_uses()
Returns an iterator range over all operands that are (explicit or implicit) register uses.
const MachineOperand & getOperand(unsigned i) const
uint32_t getFlags() const
Return the MI flags bitvector.
LLVM_ABI void clearKillInfo()
Clears kill flags on all operands.
A description of a memory reference used in the backend.
@ MOLoad
The memory access reads data.
@ MOStore
The memory access writes data.
This class contains meta information specific to a module.
MachineOperand class - Representation of each machine instruction operand.
int64_t getImm() const
bool isReg() const
isReg - Tests if this is a MO_Register operand.
MachineBasicBlock * getMBB() const
bool isImm() const
isImm - Tests if this is a MO_Immediate operand.
static MachineOperand CreateImm(int64_t Val)
MachineOperandType getType() const
getType - Returns the MachineOperandType for this operand.
Register getReg() const
getReg - Returns the register number.
bool isFI() const
isFI - Tests if this is a MO_FrameIndex operand.
LLVM_ABI bool isIdenticalTo(const MachineOperand &Other) const
Returns true if this operand is identical to the specified operand except for liveness related flags ...
@ MO_Immediate
Immediate operand.
@ MO_Register
Register operand.
MachineRegisterInfo - Keep track of information for virtual and physical registers,...
LLVM_ABI bool hasOneNonDBGUse(Register RegNo) const
hasOneNonDBGUse - Return true if there is exactly one non-Debug use of the specified register.
const TargetRegisterClass * getRegClass(Register Reg) const
Return the register class of the specified virtual register.
LLVM_ABI void clearKillFlags(Register Reg) const
clearKillFlags - Iterate over all the uses of the given register and clear the kill flag from the Mac...
LLVM_ABI LLVM_READONLY MachineInstr * getVRegDef(Register Reg) const
getVRegDef - Return the machine instr that defines the specified virtual register or null if none is ...
bool isReserved(MCRegister PhysReg) const
isReserved - Returns true when PhysReg is a reserved register.
LLVM_ABI Register createVirtualRegister(const TargetRegisterClass *RegClass, StringRef Name="")
createVirtualRegister - Create and return a new virtual register in the function with the specified r...
bool hasOneUse(Register RegNo) const
hasOneUse - Return true if there is exactly one instruction using the specified register.
LLVM_ABI void clearVirtRegs()
clearVirtRegs - Remove all virtual registers (after physreg assignment).
const TargetRegisterInfo * getTargetRegisterInfo() const
LLVM_ABI bool isConstantPhysReg(MCRegister PhysReg) const
Returns true if PhysReg is unallocatable and constant throughout the function.
LLVM_ABI const TargetRegisterClass * constrainRegClass(Register Reg, const TargetRegisterClass *RC, unsigned MinNumRegs=0)
constrainRegClass - Constrain the register class of the specified virtual register to be a common sub...
LLVM_ABI void replaceRegWith(Register FromReg, Register ToReg)
replaceRegWith - Replace all instances of FromReg with ToReg in the machine function.
LLVM_ABI LLVM_READONLY MachineInstr * getUniqueVRegDef(Register Reg) const
getUniqueVRegDef - Return the unique machine instr that defines the specified virtual register or nul...
A Module instance is used to store all the information related to an LLVM module.
Definition Module.h:68
MI-level patchpoint operands.
Definition StackMaps.h:77
uint32_t getNumPatchBytes() const
Return the number of patchable bytes the given patchpoint should emit.
Definition StackMaps.h:105
void storeRegToStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, Register SrcReg, bool IsKill, int FrameIndex, const TargetRegisterClass *RC, Register VReg, MachineInstr::MIFlag Flags=MachineInstr::NoFlags) const override
static bool isSafeToMove(const MachineInstr &From, const MachineBasicBlock::iterator &To)
Return true if moving From down to To won't cause any physical register reads or writes to be clobber...
MachineInstr * convertToThreeAddress(MachineInstr &MI, LiveVariables *LV, LiveIntervals *LIS) const override
Register isLoadFromStackSlot(const MachineInstr &MI, int &FrameIndex) const override
std::optional< std::unique_ptr< outliner::OutlinedFunction > > getOutliningCandidateInfo(const MachineModuleInfo &MMI, std::vector< outliner::Candidate > &RepeatedSequenceLocs, unsigned MinRepeats) const override
unsigned removeBranch(MachineBasicBlock &MBB, int *BytesRemoved=nullptr) const override
void genAlternativeCodeSequence(MachineInstr &Root, unsigned Pattern, SmallVectorImpl< MachineInstr * > &InsInstrs, SmallVectorImpl< MachineInstr * > &DelInstrs, DenseMap< Register, unsigned > &InstrIdxForVirtReg) const override
void movImm(MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, const DebugLoc &DL, Register DstReg, uint64_t Val, MachineInstr::MIFlag Flag=MachineInstr::NoFlags, bool DstRenamable=false, bool DstIsDead=false) const
MachineInstr * emitLdStWithAddr(MachineInstr &MemI, const ExtAddrMode &AM) const override
void mulImm(MachineFunction &MF, MachineBasicBlock &MBB, MachineBasicBlock::iterator II, const DebugLoc &DL, Register DestReg, uint32_t Amt, MachineInstr::MIFlag Flag) const
Generate code to multiply the value in DestReg by Amt - handles all the common optimizations for this...
static bool isPairableLdStInstOpc(unsigned Opc)
Return true if pairing the given load or store may be paired with another.
RISCVInstrInfo(const RISCVSubtarget &STI)
void loadRegFromStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, Register DstReg, int FrameIndex, const TargetRegisterClass *RC, Register VReg, unsigned SubReg=0, MachineInstr::MIFlag Flags=MachineInstr::NoFlags) const override
bool isFunctionSafeToOutlineFrom(MachineFunction &MF, bool OutlineFromLinkOnceODRs) const override
std::unique_ptr< TargetInstrInfo::PipelinerLoopInfo > analyzeLoopForPipelining(MachineBasicBlock *LoopBB) const override
unsigned insertBranch(MachineBasicBlock &MBB, MachineBasicBlock *TBB, MachineBasicBlock *FBB, ArrayRef< MachineOperand > Cond, const DebugLoc &dl, int *BytesAdded=nullptr) const override
bool hasReassociableSibling(const MachineInstr &Inst, bool &Commuted) const override
static bool isLdStSafeToPair(const MachineInstr &LdSt, const TargetRegisterInfo *TRI)
void copyPhysRegVector(MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, const DebugLoc &DL, MCRegister DstReg, MCRegister SrcReg, bool KillSrc, const TargetRegisterClass *RegClass) const
bool isReMaterializableImpl(const MachineInstr &MI) const override
MachineInstr * optimizeSelect(MachineInstr &MI, SmallPtrSetImpl< MachineInstr * > &SeenMIs, bool) const override
bool isVRegCopy(const MachineInstr *MI, unsigned LMul=0) const
Return true if MI is a COPY to a vector register of a specific LMul, or any kind of vector registers ...
bool canFoldIntoAddrMode(const MachineInstr &MemI, Register Reg, const MachineInstr &AddrI, ExtAddrMode &AM) const override
void insertIndirectBranch(MachineBasicBlock &MBB, MachineBasicBlock &NewDestBB, MachineBasicBlock &RestoreBB, const DebugLoc &DL, int64_t BrOffset, RegScavenger *RS) const override
bool isAsCheapAsAMove(const MachineInstr &MI) const override
bool verifyInstruction(const MachineInstr &MI, StringRef &ErrInfo) const override
bool getMemOperandWithOffsetWidth(const MachineInstr &LdSt, const MachineOperand *&BaseOp, int64_t &Offset, LocationSize &Width, const TargetRegisterInfo *TRI) const
unsigned getTailDuplicateSize(CodeGenOptLevel OptLevel) const override
void getReassociateOperandIndices(const MachineInstr &Root, unsigned Pattern, std::array< unsigned, 5 > &OperandIndices) const override
const RISCVSubtarget & STI
Register isStoreToStackSlot(const MachineInstr &MI, int &FrameIndex) const override
std::optional< unsigned > getInverseOpcode(unsigned Opcode) const override
bool simplifyInstruction(MachineInstr &MI) const override
ArrayRef< std::pair< unsigned, const char * > > getSerializableDirectMachineOperandTargetFlags() const override
outliner::InstrType getOutliningTypeImpl(const MachineModuleInfo &MMI, MachineBasicBlock::iterator &MBBI, unsigned Flags) const override
MachineTraceStrategy getMachineCombinerTraceStrategy() const override
unsigned getInstSizeInBytes(const MachineInstr &MI) const override
std::optional< RegImmPair > isAddImmediate(const MachineInstr &MI, Register Reg) const override
bool reverseBranchCondition(SmallVectorImpl< MachineOperand > &Cond) const override
ArrayRef< std::pair< MachineMemOperand::Flags, const char * > > getSerializableMachineMemOperandTargetFlags() const override
MCInst getNop() const override
bool analyzeCandidate(outliner::Candidate &C) const
bool isMBBSafeToOutlineFrom(MachineBasicBlock &MBB, unsigned &Flags) const override
bool getMemOperandsWithOffsetWidth(const MachineInstr &MI, SmallVectorImpl< const MachineOperand * > &BaseOps, int64_t &Offset, bool &OffsetIsScalable, LocationSize &Width, const TargetRegisterInfo *TRI) const override
void buildOutlinedFrame(MachineBasicBlock &MBB, MachineFunction &MF, const outliner::OutlinedFunction &OF) const override
bool requiresNTLHint(const MachineInstr &MI) const
Return true if the instruction requires an NTL hint to be emitted.
void finalizeInsInstrs(MachineInstr &Root, unsigned &Pattern, SmallVectorImpl< MachineInstr * > &InsInstrs) const override
std::pair< unsigned, unsigned > decomposeMachineOperandsTargetFlags(unsigned TF) const override
MachineInstr * commuteInstructionImpl(MachineInstr &MI, bool NewMI, unsigned OpIdx1, unsigned OpIdx2) const override
bool hasReassociableOperands(const MachineInstr &Inst, const MachineBasicBlock *MBB) const override
MachineBasicBlock * getBranchDestBlock(const MachineInstr &MI) const override
std::string createMIROperandComment(const MachineInstr &MI, const MachineOperand &Op, unsigned OpIdx, const TargetRegisterInfo *TRI) const override
bool shouldOutlineFromFunctionByDefault(MachineFunction &MF) const override
void copyPhysReg(MachineBasicBlock &MBB, MachineBasicBlock::iterator MBBI, const DebugLoc &DL, Register DstReg, Register SrcReg, bool KillSrc, bool RenamableDest=false, bool RenamableSrc=false) const override
bool findCommutedOpIndices(const MachineInstr &MI, unsigned &SrcOpIdx1, unsigned &SrcOpIdx2) const override
bool analyzeBranch(MachineBasicBlock &MBB, MachineBasicBlock *&TBB, MachineBasicBlock *&FBB, SmallVectorImpl< MachineOperand > &Cond, bool AllowModify) const override
MachineBasicBlock::iterator insertOutlinedCall(Module &M, MachineBasicBlock &MBB, MachineBasicBlock::iterator &It, MachineFunction &MF, outliner::Candidate &C) const override
MachineInstr * foldMemoryOperandImpl(MachineFunction &MF, MachineInstr &MI, ArrayRef< unsigned > Ops, int FrameIndex, MachineInstr *&CopyMI, LiveIntervals *LIS=nullptr, VirtRegMap *VRM=nullptr) const override
bool isBranchOffsetInRange(unsigned BranchOpc, int64_t BrOffset) const override
static RISCVCC::CondCode getCondFromBranchOpc(unsigned Opc)
void buildClearRegister(Register Reg, MachineBasicBlock &MBB, MachineBasicBlock::iterator Iter, DebugLoc &DL, bool AllowSideEffects=true) const override
bool isAssociativeAndCommutative(const MachineInstr &Inst, bool Invert) const override
CombinerObjective getCombinerObjective(unsigned Pattern) const override
bool isHighLatencyDef(int Opc) const override
static bool evaluateCondBranch(RISCVCC::CondCode CC, int64_t C0, int64_t C1)
Return the result of the evaluation of C0 CC C1, where CC is a RISCVCC::CondCode.
bool getMachineCombinerPatterns(MachineInstr &Root, SmallVectorImpl< unsigned > &Patterns, bool DoRegPressureReduce) const override
bool optimizeCondBranch(MachineInstr &MI) const override
std::optional< DestSourcePair > isCopyInstrImpl(const MachineInstr &MI) const override
static bool isFromLoadImm(const MachineRegisterInfo &MRI, const MachineOperand &Op, int64_t &Imm)
Return true if the operand is a load immediate instruction and sets Imm to the immediate value.
bool shouldClusterMemOps(ArrayRef< const MachineOperand * > BaseOps1, int64_t Offset1, bool OffsetIsScalable1, ArrayRef< const MachineOperand * > BaseOps2, int64_t Offset2, bool OffsetIsScalable2, unsigned ClusterSize, unsigned NumBytes) const override
bool areMemAccessesTriviallyDisjoint(const MachineInstr &MIa, const MachineInstr &MIb) const override
RISCVMachineFunctionInfo - This class is derived from MachineFunctionInfo and contains private RISCV-...
const RISCVRegisterInfo * getRegisterInfo() const override
Wrapper class representing virtual and physical registers.
Definition Register.h:20
constexpr bool isValid() const
Definition Register.h:112
constexpr bool isVirtual() const
Return true if the specified register number is in the virtual register namespace.
Definition Register.h:79
SlotIndex - An opaque wrapper around machine indexes.
Definition SlotIndexes.h:66
SlotIndex getRegSlot(bool EC=false) const
Returns the register use/def slot in the current instruction for a normal or early-clobber def.
A templated base class for SmallPtrSet which provides the typesafe interface that is common across al...
bool erase(PtrType Ptr)
Remove pointer from the set.
std::pair< iterator, bool > insert(PtrType Ptr)
Inserts Ptr if and only if there is no element in the container equal to Ptr.
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
MI-level stackmap operands.
Definition StackMaps.h:36
uint32_t getNumPatchBytes() const
Return the number of patchable bytes the given stackmap should emit.
Definition StackMaps.h:51
MI-level Statepoint operands.
Definition StackMaps.h:159
uint32_t getNumPatchBytes() const
Return the number of patchable bytes the given statepoint should emit.
Definition StackMaps.h:208
Represent a constant reference to a string, i.e.
Definition StringRef.h:56
Object returned by analyzeLoopForPipelining.
TargetInstrInfo - Interface to description of machine instruction set.
virtual bool findCommutedOpIndices(const MachineInstr &MI, unsigned &SrcOpIdx1, unsigned &SrcOpIdx2) const
Returns true iff the routine could find two commutable operands in the given machine instruction.
virtual bool hasReassociableOperands(const MachineInstr &Inst, const MachineBasicBlock *MBB) const
Return true when \P Inst has reassociable operands in the same \P MBB.
virtual void genAlternativeCodeSequence(MachineInstr &Root, unsigned Pattern, SmallVectorImpl< MachineInstr * > &InsInstrs, SmallVectorImpl< MachineInstr * > &DelInstrs, DenseMap< Register, unsigned > &InstIdxForVirtReg) const
When getMachineCombinerPatterns() finds patterns, this function generates the instructions that could...
virtual bool getMachineCombinerPatterns(MachineInstr &Root, SmallVectorImpl< unsigned > &Patterns, bool DoRegPressureReduce) const
Return true when there is potentially a faster code sequence for an instruction chain ending in Root.
virtual bool isReMaterializableImpl(const MachineInstr &MI) const
For instructions with opcodes for which the M_REMATERIALIZABLE flag is set, this hook lets the target...
virtual bool isMBBSafeToOutlineFrom(MachineBasicBlock &MBB, unsigned &Flags) const
Optional target hook that returns true if MBB is safe to outline from, and returns any target-specifi...
virtual void getReassociateOperandIndices(const MachineInstr &Root, unsigned Pattern, std::array< unsigned, 5 > &OperandIndices) const
The returned array encodes the operand index for each parameter because the operands may be commuted;...
virtual CombinerObjective getCombinerObjective(unsigned Pattern) const
Return the objective of a combiner pattern.
virtual MachineInstr * commuteInstructionImpl(MachineInstr &MI, bool NewMI, unsigned OpIdx1, unsigned OpIdx2) const
This method commutes the operands of the given machine instruction MI.
virtual bool hasReassociableSibling(const MachineInstr &Inst, bool &Commuted) const
Return true when \P Inst has reassociable sibling.
virtual std::string createMIROperandComment(const MachineInstr &MI, const MachineOperand &Op, unsigned OpIdx, const TargetRegisterInfo *TRI) const
const MCAsmInfo & getMCAsmInfo() const
Return target specific asm information.
TargetRegisterInfo base class - We assume that the target defines a static array of TargetRegisterDes...
TargetSubtargetInfo - Generic base class for all target subtargets.
virtual const TargetInstrInfo * getInstrInfo() const
virtual const TargetRegisterInfo * getRegisterInfo() const =0
Return the target's register information.
Target - Wrapper for Target specific information.
static constexpr TypeSize getFixed(ScalarTy ExactSize)
Definition TypeSize.h:339
static constexpr TypeSize getZero()
Definition TypeSize.h:345
static constexpr TypeSize getScalable(ScalarTy MinimumSize)
Definition TypeSize.h:342
self_iterator getIterator()
Definition ilist_node.h:123
A raw_ostream that writes to an std::string.
#define llvm_unreachable(msg)
Marks that the current location is not supposed to be reachable.
CondCode getInverseBranchCondition(CondCode)
unsigned getInverseBranchOpcode(unsigned BCC)
unsigned getBrCond(CondCode CC, unsigned SelectOpc=0)
static bool isValidRoundingMode(unsigned Mode)
static StringRef roundingModeToString(RoundingMode RndMode)
static unsigned getVecPolicyOpNum(const MCInstrDesc &Desc)
static bool usesMaskPolicy(uint64_t TSFlags)
static bool hasRoundModeOp(uint64_t TSFlags)
static unsigned getVLOpNum(const MCInstrDesc &Desc)
static bool hasVLOp(uint64_t TSFlags)
static MCRegister getTailExpandUseRegNo(const FeatureBitset &FeatureBits)
static int getFRMOpNum(const MCInstrDesc &Desc)
static int getVXRMOpNum(const MCInstrDesc &Desc)
static bool hasVecPolicyOp(uint64_t TSFlags)
static bool usesVXRM(uint64_t TSFlags)
static bool isRVVWideningReduction(uint64_t TSFlags)
static unsigned getSEWOpNum(const MCInstrDesc &Desc)
static bool hasSEWOp(uint64_t TSFlags)
static bool isFirstDefTiedToFirstUse(const MCInstrDesc &Desc)
InstSeq generateInstSeq(int64_t Val, const MCSubtargetInfo &STI)
SmallVector< Inst, 8 > InstSeq
Definition RISCVMatInt.h:43
@ OPERAND_UIMMLOG2XLEN_NONZERO
@ OPERAND_UIMM10_LSB00_NONZERO
@ OPERAND_SIMM10_LSB0000_NONZERO
static unsigned getNF(uint8_t TSFlags)
static RISCVVType::VLMUL getLMul(uint8_t TSFlags)
static bool isTailAgnostic(unsigned VType)
LLVM_ABI void printXSfmmVType(unsigned VType, raw_ostream &OS)
LLVM_ABI std::pair< unsigned, bool > decodeVLMUL(VLMUL VLMul)
static bool isValidSEW(unsigned SEW)
static bool isValidVType(unsigned VType)
LLVM_ABI void printVType(unsigned VType, raw_ostream &OS)
static bool isValidXSfmmVType(unsigned VTypeI)
static unsigned getSEW(unsigned VType)
static VLMUL getVLMUL(unsigned VType)
static bool isValidRoundingMode(unsigned Mode)
static StringRef roundingModeToString(RoundingMode RndMode)
bool hasEqualFRM(const MachineInstr &MI1, const MachineInstr &MI2)
bool isValidYBNDSWImm(int64_t Imm)
unsigned getRVVMCOpcode(unsigned RVVPseudoOpcode)
unsigned getDestLog2EEW(const MCInstrDesc &Desc, unsigned Log2SEW)
std::optional< unsigned > getVectorLowDemandedScalarBits(unsigned Opcode, unsigned Log2SEW)
std::optional< std::pair< unsigned, unsigned > > isRVVSpillForZvlsseg(unsigned Opcode)
static constexpr unsigned RVVBitsPerBlock
bool isRVVSpill(const MachineInstr &MI)
static constexpr unsigned RVVBytesPerBlock
static constexpr int64_t VLMaxSentinel
bool isVLKnownLE(const MachineRegisterInfo &MRI, const MachineOperand &LHS, const MachineOperand &RHS)
Given two VL operands, do we know that LHS <= RHS?
bool isVectorCopy(const TargetRegisterInfo *TRI, const MachineInstr &MI)
Return true if MI is a copy that will be lowered to one or more vmvNr.vs.
static bool isValidSMTVTypeMode(unsigned Mode)
ValuesClass values(OptsTy... Options)
Helper to build a ValuesClass by forwarding a variable number of arguments as an initializer list to ...
initializer< Ty > init(const Ty &Val)
InstrType
Represents how an instruction should be mapped by the outliner.
This is an optimization pass for GlobalISel generic memory operations.
auto drop_begin(T &&RangeOrContainer, size_t N=1)
Return a range covering RangeOrContainer with the first N elements excluded.
Definition STLExtras.h:315
@ Offset
Definition DWP.cpp:577
@ SHXADD_ADD_SLLI_OP2
@ SHXADD_ADD_SLLI_OP1
MachineTraceStrategy
Strategies for selecting traces.
@ TS_MinInstrCount
Select the trace through a block that has the fewest instructions.
@ TS_Local
Select the trace that contains only the current basic block.
bool all_of(R &&range, UnaryPredicate P)
Provide wrappers to std::all_of which take ranges instead of having to pass begin/end explicitly.
Definition STLExtras.h:1739
static const MachineMemOperand::Flags MONontemporalBit1
MachineInstrBuilder BuildMI(MachineFunction &MF, const MIMetadata &MIMD, const MCInstrDesc &MCID)
Builder interface. Specify how to create the initial instruction itself.
constexpr bool isInt(int64_t x)
Checks if an integer fits into the given bit width.
Definition MathExtras.h:166
RegState
Flags to represent properties of register accesses.
@ Implicit
Not emitted register (e.g. carry, or temporary result).
@ Dead
Unused definition.
@ Kill
The last use of a register.
@ Undef
Value of the register doesn't matter.
@ Define
Register definition.
auto enumerate(FirstRange &&First, RestRanges &&...Rest)
Given two or more input ranges, returns a new range whose values are tuples (A, B,...
Definition STLExtras.h:2554
bool isValidAtomicOrdering(Int I)
constexpr RegState getKillRegState(bool B)
static const MachineMemOperand::Flags MONontemporalBit0
constexpr RegState getDeadRegState(bool B)
LLVM_ABI void reportFatalInternalError(Error Err)
Report a fatal error that indicates a bug in LLVM.
Definition Error.cpp:173
Op::Description Desc
unsigned M1(unsigned Val)
Definition VE.h:377
constexpr bool has_single_bit(T Value) noexcept
Definition bit.h:149
unsigned Log2_32(uint32_t Value)
Return the floor log base 2 of the specified value, -1 if the value is zero.
Definition MathExtras.h:326
MachineInstr * getImm(const MachineOperand &MO, const MachineRegisterInfo *MRI)
constexpr RegState getRenamableRegState(bool B)
decltype(auto) get(const PointerIntPair< PointerTy, IntBits, IntType, PtrTraits, Info > &Pair)
LLVM_ABI raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
Definition Debug.cpp:209
LLVM_ABI void report_fatal_error(Error Err, bool gen_crash_diag=true)
Definition Error.cpp:163
constexpr RegState getDefRegState(bool B)
CombinerObjective
The combiner's goal may differ based on which pattern it is attempting to optimize.
constexpr bool isUInt(uint64_t x)
Checks if an unsigned integer fits into the given bit width.
Definition MathExtras.h:190
CodeGenOptLevel
Code generation optimization level.
Definition CodeGen.h:177
int isShifted359(T Value, int &Shift)
bool isa(const From &Val)
isa<X> - Return true if the parameter to the template is an instance of one of the template type argu...
Definition Casting.h:547
uint16_t MCPhysReg
An unsigned integer type large enough to represent all physical registers, but not necessarily virtua...
Definition MCRegister.h:21
DWARFExpression::Operation Op
ArrayRef(const T &OneElt) -> ArrayRef< T >
constexpr bool isShiftedInt(int64_t x)
Checks if a signed integer is an N bit number shifted left by S.
Definition MathExtras.h:183
void erase_if(Container &C, UnaryPredicate P)
Provide a container algorithm similar to C++ Library Fundamentals v2's erase_if which is equivalent t...
Definition STLExtras.h:2192
constexpr int64_t SignExtend64(uint64_t x)
Sign-extend the number in the bottom B bits of X to a 64-bit integer.
Definition MathExtras.h:567
LLVM_ABI const Value * getUnderlyingObject(const Value *V, unsigned MaxLookup=MaxLookupSearchDepth)
This method strips off any GEP address adjustments, pointer casts or llvm.threadlocal....
constexpr bool isShiftedUInt(uint64_t x)
Checks if a unsigned integer is an N bit number shifted left by S.
Definition MathExtras.h:199
MCRegisterClass TargetRegisterClass
Definition FastISel.h:58
void swap(llvm::BitVector &LHS, llvm::BitVector &RHS)
Implement std::swap in terms of BitVector swap.
Definition BitVector.h:880
#define N
This struct is a compact representation of a valid (non-zero power of two) alignment.
Definition Alignment.h:39
Used to describe addressing mode similar to ExtAddrMode in CodeGenPrepare.
This represents a simple continuous liveness interval for a value.
static LLVM_ABI MachinePointerInfo getFixedStack(MachineFunction &MF, int FI, int64_t Offset=0)
Return a MachinePointerInfo record that refers to the specified FrameIndex.
static bool isRVVRegClass(const TargetRegisterClass *RC)
Used to describe a register and immediate addition.
An individual sequence of instructions to be replaced with a call to an outlined function.
MachineFunction * getMF() const
The information necessary to create an outlined function for some class of candidate.