diff --git a/llvm/lib/Target/LoongArch/LoongArchISelDAGToDAG.cpp b/llvm/lib/Target/LoongArch/LoongArchISelDAGToDAG.cpp --- a/llvm/lib/Target/LoongArch/LoongArchISelDAGToDAG.cpp +++ b/llvm/lib/Target/LoongArch/LoongArchISelDAGToDAG.cpp @@ -12,6 +12,7 @@ #include "LoongArchISelDAGToDAG.h" #include "MCTargetDesc/LoongArchMCTargetDesc.h" +#include "MCTargetDesc/LoongArchMatInt.h" using namespace llvm; @@ -28,11 +29,30 @@ // Instruction Selection not handled by the auto-generated tablegen selection // should be handled here. unsigned Opcode = Node->getOpcode(); + MVT GRLenVT = Subtarget->getGRLenVT(); SDLoc DL(Node); switch (Opcode) { default: break; + case ISD::Constant: { + int64_t Imm = cast(Node)->getSExtValue(); + SDNode *Result = nullptr; + SDValue SrcReg = CurDAG->getRegister(LoongArch::R0, GRLenVT); + + // The instructions in the sequence are handled here. + for (LoongArchMatInt::Inst &Inst : LoongArchMatInt::generateInstSeq(Imm)) { + SDValue SDImm = CurDAG->getTargetConstant(Inst.Imm, DL, GRLenVT); + if (Inst.Opc == LoongArch::LU12I_W) + Result = CurDAG->getMachineNode(LoongArch::LU12I_W, DL, GRLenVT, SDImm); + else + Result = CurDAG->getMachineNode(Inst.Opc, DL, GRLenVT, SrcReg, SDImm); + SrcReg = SDValue(Result, 0); + } + + ReplaceNode(Node, Result); + return; + } // TODO: Add selection nodes needed later. } diff --git a/llvm/lib/Target/LoongArch/MCTargetDesc/CMakeLists.txt b/llvm/lib/Target/LoongArch/MCTargetDesc/CMakeLists.txt --- a/llvm/lib/Target/LoongArch/MCTargetDesc/CMakeLists.txt +++ b/llvm/lib/Target/LoongArch/MCTargetDesc/CMakeLists.txt @@ -6,6 +6,7 @@ LoongArchMCAsmInfo.cpp LoongArchMCTargetDesc.cpp LoongArchMCCodeEmitter.cpp + LoongArchMatInt.cpp LINK_COMPONENTS MC diff --git a/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchMatInt.h b/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchMatInt.h new file mode 100644 --- /dev/null +++ b/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchMatInt.h @@ -0,0 +1,30 @@ +//===- LoongArchMatInt.h - Immediate materialisation - --------*- C++ -*--===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_LIB_TARGET_LOONGARCH_MCTARGETDESC_MATINT_H +#define LLVM_LIB_TARGET_LOONGARCH_MCTARGETDESC_MATINT_H + +#include "llvm/ADT/SmallVector.h" +#include + +namespace llvm { +namespace LoongArchMatInt { +struct Inst { + unsigned Opc; + int64_t Imm; + Inst(unsigned Opc, int64_t Imm) : Opc(Opc), Imm(Imm) {} +}; +using InstSeq = SmallVector; + +// Helper to generate an instruction sequence that will materialise the given +// immediate value into a register. +InstSeq generateInstSeq(int64_t Val); +} // namespace LoongArchMatInt +} // namespace llvm + +#endif diff --git a/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchMatInt.cpp b/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchMatInt.cpp new file mode 100644 --- /dev/null +++ b/llvm/lib/Target/LoongArch/MCTargetDesc/LoongArchMatInt.cpp @@ -0,0 +1,51 @@ +//===- LoongArchMatInt.cpp - Immediate materialisation ---------*- C++ -*--===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#include "LoongArchMatInt.h" +#include "MCTargetDesc/LoongArchMCTargetDesc.h" +#include "llvm/Support/MathExtras.h" + +using namespace llvm; + +LoongArchMatInt::InstSeq LoongArchMatInt::generateInstSeq(int64_t Val) { + // Val: + // | hi32 | lo32 | + // +-----------+------------------+------------------+-----------+ + // | Highest12 | Higher20 | Hi20 | Lo12 | + // +-----------+------------------+------------------+-----------+ + // 63 52 51 32 31 12 11 0 + // + const int64_t Highest12 = Val >> 52 & 0xFFF; + const int64_t Higher20 = Val >> 32 & 0xFFFFF; + const int64_t Hi20 = Val >> 12 & 0xFFFFF; + const int64_t Lo12 = Val & 0xFFF; + InstSeq Insts; + + if (Highest12 != 0 && SignExtend64<52>(Val) == 0) { + Insts.push_back(Inst(LoongArch::LU52I_D, SignExtend64<12>(Highest12))); + return Insts; + } + + if (Hi20 == 0) + Insts.push_back(Inst(LoongArch::ORI, Lo12)); + else if (SignExtend32<1>(Lo12 >> 11) == SignExtend32<20>(Hi20)) + Insts.push_back(Inst(LoongArch::ADDI_W, SignExtend64<12>(Lo12))); + else { + Insts.push_back(Inst(LoongArch::LU12I_W, SignExtend64<20>(Hi20))); + if (Lo12 != 0) + Insts.push_back(Inst(LoongArch::ORI, Lo12)); + } + + if (SignExtend32<1>(Hi20 >> 19) != SignExtend32<20>(Higher20)) + Insts.push_back(Inst(LoongArch::LU32I_D, SignExtend64<20>(Higher20))); + + if (SignExtend32<1>(Higher20 >> 19) != SignExtend32<12>(Highest12)) + Insts.push_back(Inst(LoongArch::LU52I_D, SignExtend64<12>(Highest12))); + + return Insts; +} diff --git a/llvm/test/CodeGen/LoongArch/imm.ll b/llvm/test/CodeGen/LoongArch/imm.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/LoongArch/imm.ll @@ -0,0 +1,157 @@ +; RUN: llc --mtriple=loongarch64 < %s | FileCheck %s + +define i64 @imm7ff0000000000000() { +; CHECK-LABEL: imm7ff0000000000000: +; CHECK: # %bb.0: +; CHECK-NEXT: lu52i.d $a0, $zero, 2047 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 9218868437227405312 +} + +define i64 @imm0000000000000fff() { +; CHECK-LABEL: imm0000000000000fff: +; CHECK: # %bb.0: +; CHECK-NEXT: ori $a0, $zero, 4095 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 4095 +} + +define i64 @imm0007ffff00000800() { +; CHECK-LABEL: imm0007ffff00000800: +; CHECK: # %bb.0: +; CHECK-NEXT: ori $a0, $zero, 2048 +; CHECK-NEXT: lu32i.d $a0, 524287 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 2251795518720000 +} + +define i64 @immfff0000000000fff() { +; CHECK-LABEL: immfff0000000000fff: +; CHECK: # %bb.0: +; CHECK-NEXT: ori $a0, $zero, 4095 +; CHECK-NEXT: lu52i.d $a0, $a0, -1 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 -4503599627366401 +} + +define i64 @imm0008000000000fff() { +; CHECK-LABEL: imm0008000000000fff: +; CHECK: # %bb.0: +; CHECK-NEXT: ori $a0, $zero, 4095 +; CHECK-NEXT: lu32i.d $a0, -524288 +; CHECK-NEXT: lu52i.d $a0, $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 2251799813689343 +} + +define i64 @immfffffffffffff800() { +; CHECK-LABEL: immfffffffffffff800: +; CHECK: # %bb.0: +; CHECK-NEXT: addi.w $a0, $zero, -2048 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 -2048 +} + +define i64 @imm00000000fffff800() { +; CHECK-LABEL: imm00000000fffff800: +; CHECK: # %bb.0: +; CHECK-NEXT: addi.w $a0, $zero, -2048 +; CHECK-NEXT: lu32i.d $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 4294965248 +} + +define i64 @imm000ffffffffff800() { +; CHECK-LABEL: imm000ffffffffff800: +; CHECK: # %bb.0: +; CHECK-NEXT: addi.w $a0, $zero, -2048 +; CHECK-NEXT: lu52i.d $a0, $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 4503599627368448 +} + +define i64 @imm00080000fffff800() { +; CHECK-LABEL: imm00080000fffff800: +; CHECK: # %bb.0: +; CHECK-NEXT: addi.w $a0, $zero, -2048 +; CHECK-NEXT: lu32i.d $a0, -524288 +; CHECK-NEXT: lu52i.d $a0, $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 2251804108650496 +} + +define i64 @imm000000007ffff000() { +; CHECK-LABEL: imm000000007ffff000: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, 524287 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 2147479552 +} + +define i64 @imm0000000080000000() { +; CHECK-LABEL: imm0000000080000000: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, -524288 +; CHECK-NEXT: lu32i.d $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 2147483648 +} + +define i64 @imm000ffffffffff000() { +; CHECK-LABEL: imm000ffffffffff000: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, -1 +; CHECK-NEXT: lu52i.d $a0, $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 4503599627366400 +} + +define i64 @imm7ff0000080000000() { +; CHECK-LABEL: imm7ff0000080000000: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, -524288 +; CHECK-NEXT: lu32i.d $a0, 0 +; CHECK-NEXT: lu52i.d $a0, $a0, 2047 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 9218868439374888960 +} + +define i64 @immffffffff80000800() { +; CHECK-LABEL: immffffffff80000800: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, -524288 +; CHECK-NEXT: ori $a0, $a0, 2048 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 -2147481600 +} + +define i64 @immffffffff7ffff800() { +; CHECK-LABEL: immffffffff7ffff800: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, 524287 +; CHECK-NEXT: ori $a0, $a0, 2048 +; CHECK-NEXT: lu32i.d $a0, -1 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 -2147485696 +} + +define i64 @imm7fffffff800007ff() { +; CHECK-LABEL: imm7fffffff800007ff: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, -524288 +; CHECK-NEXT: ori $a0, $a0, 2047 +; CHECK-NEXT: lu52i.d $a0, $a0, 2047 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 9223372034707294207 +} + +define i64 @imm0008000080000800() { +; CHECK-LABEL: imm0008000080000800: +; CHECK: # %bb.0: +; CHECK-NEXT: lu12i.w $a0, -524288 +; CHECK-NEXT: ori $a0, $a0, 2048 +; CHECK-NEXT: lu32i.d $a0, -524288 +; CHECK-NEXT: lu52i.d $a0, $a0, 0 +; CHECK-NEXT: jirl $zero, $ra, 0 + ret i64 2251801961170944 +}