diff --git a/llvm/lib/Target/PowerPC/PPCISelLowering.cpp b/llvm/lib/Target/PowerPC/PPCISelLowering.cpp --- a/llvm/lib/Target/PowerPC/PPCISelLowering.cpp +++ b/llvm/lib/Target/PowerPC/PPCISelLowering.cpp @@ -14388,6 +14388,17 @@ return std::make_pair(0U, &PPC::VSFRCRegClass); } + // If we name a VSX register, we can't defer to the base class because it + // will not recognize the correct register (their names will be VSL{0-31} + // and V{0-31} so they won't match). So we match them here. + if (Constraint.size() > 3 && Constraint[1] == 'v' && Constraint[2] == 's') { + int VSNum = atoi(Constraint.data() + 3); + assert(VSNum >= 0 && VSNum <= 63 && + "Attempted to access a vsr out of range"); + if (VSNum < 32) + return std::make_pair(PPC::VSL0 + VSNum, &PPC::VSRCRegClass); + return std::make_pair(PPC::V0 + VSNum - 32, &PPC::VSRCRegClass); + } std::pair R = TargetLowering::getRegForInlineAsmConstraint(TRI, Constraint, VT); diff --git a/llvm/test/CodeGen/PowerPC/CSR-fit.ll b/llvm/test/CodeGen/PowerPC/CSR-fit.ll --- a/llvm/test/CodeGen/PowerPC/CSR-fit.ll +++ b/llvm/test/CodeGen/PowerPC/CSR-fit.ll @@ -178,16 +178,26 @@ ; CHECK-PWR8: # %bb.0: # %entry ; CHECK-PWR8-NEXT: mflr r0 ; CHECK-PWR8-NEXT: std r0, 16(r1) -; CHECK-PWR8-NEXT: stdu r1, -32(r1) -; CHECK-PWR8-NEXT: .cfi_def_cfa_offset 32 +; CHECK-PWR8-NEXT: stdu r1, -240(r1) +; CHECK-PWR8-NEXT: .cfi_def_cfa_offset 240 ; CHECK-PWR8-NEXT: .cfi_offset lr, 16 +; CHECK-PWR8-NEXT: .cfi_offset v20, -192 +; CHECK-PWR8-NEXT: .cfi_offset v21, -176 +; CHECK-PWR8-NEXT: li r5, 48 +; CHECK-PWR8-NEXT: stxvd2x v20, r1, r5 # 16-byte Folded Spill +; CHECK-PWR8-NEXT: li r5, 64 +; CHECK-PWR8-NEXT: stxvd2x v21, r1, r5 # 16-byte Folded Spill ; CHECK-PWR8-NEXT: #APP ; CHECK-PWR8-NEXT: add r3, r3, r4 ; CHECK-PWR8-NEXT: #NO_APP ; CHECK-PWR8-NEXT: extsw r3, r3 ; CHECK-PWR8-NEXT: bl callee ; CHECK-PWR8-NEXT: nop -; CHECK-PWR8-NEXT: addi r1, r1, 32 +; CHECK-PWR8-NEXT: li r4, 64 +; CHECK-PWR8-NEXT: lxvd2x v21, r1, r4 # 16-byte Folded Reload +; CHECK-PWR8-NEXT: li r4, 48 +; CHECK-PWR8-NEXT: lxvd2x v20, r1, r4 # 16-byte Folded Reload +; CHECK-PWR8-NEXT: addi r1, r1, 240 ; CHECK-PWR8-NEXT: ld r0, 16(r1) ; CHECK-PWR8-NEXT: mtlr r0 ; CHECK-PWR8-NEXT: blr @@ -196,16 +206,22 @@ ; CHECK-PWR9: # %bb.0: # %entry ; CHECK-PWR9-NEXT: mflr r0 ; CHECK-PWR9-NEXT: std r0, 16(r1) -; CHECK-PWR9-NEXT: stdu r1, -32(r1) -; CHECK-PWR9-NEXT: .cfi_def_cfa_offset 32 +; CHECK-PWR9-NEXT: stdu r1, -224(r1) +; CHECK-PWR9-NEXT: .cfi_def_cfa_offset 224 ; CHECK-PWR9-NEXT: .cfi_offset lr, 16 +; CHECK-PWR9-NEXT: .cfi_offset v20, -192 +; CHECK-PWR9-NEXT: .cfi_offset v21, -176 +; CHECK-PWR9-NEXT: stxv v20, 32(r1) # 16-byte Folded Spill +; CHECK-PWR9-NEXT: stxv v21, 48(r1) # 16-byte Folded Spill ; CHECK-PWR9-NEXT: #APP ; CHECK-PWR9-NEXT: add r3, r3, r4 ; CHECK-PWR9-NEXT: #NO_APP ; CHECK-PWR9-NEXT: extsw r3, r3 ; CHECK-PWR9-NEXT: bl callee ; CHECK-PWR9-NEXT: nop -; CHECK-PWR9-NEXT: addi r1, r1, 32 +; CHECK-PWR9-NEXT: lxv v21, 48(r1) # 16-byte Folded Reload +; CHECK-PWR9-NEXT: lxv v20, 32(r1) # 16-byte Folded Reload +; CHECK-PWR9-NEXT: addi r1, r1, 224 ; CHECK-PWR9-NEXT: ld r0, 16(r1) ; CHECK-PWR9-NEXT: mtlr r0 ; CHECK-PWR9-NEXT: blr @@ -226,19 +242,24 @@ ; CHECK-PWR8-NEXT: .cfi_offset r14, -288 ; CHECK-PWR8-NEXT: .cfi_offset f14, -144 ; CHECK-PWR8-NEXT: .cfi_offset v20, -480 +; CHECK-PWR8-NEXT: .cfi_offset v21, -464 ; CHECK-PWR8-NEXT: li r5, 48 ; CHECK-PWR8-NEXT: std r14, 240(r1) # 8-byte Folded Spill ; CHECK-PWR8-NEXT: stfd f14, 384(r1) # 8-byte Folded Spill ; CHECK-PWR8-NEXT: stxvd2x v20, r1, r5 # 16-byte Folded Spill +; CHECK-PWR8-NEXT: li r5, 64 +; CHECK-PWR8-NEXT: stxvd2x v21, r1, r5 # 16-byte Folded Spill ; CHECK-PWR8-NEXT: #APP ; CHECK-PWR8-NEXT: add r3, r3, r4 ; CHECK-PWR8-NEXT: #NO_APP ; CHECK-PWR8-NEXT: extsw r3, r3 ; CHECK-PWR8-NEXT: bl callee ; CHECK-PWR8-NEXT: nop -; CHECK-PWR8-NEXT: li r4, 48 +; CHECK-PWR8-NEXT: li r4, 64 ; CHECK-PWR8-NEXT: lfd f14, 384(r1) # 8-byte Folded Reload ; CHECK-PWR8-NEXT: ld r14, 240(r1) # 8-byte Folded Reload +; CHECK-PWR8-NEXT: lxvd2x v21, r1, r4 # 16-byte Folded Reload +; CHECK-PWR8-NEXT: li r4, 48 ; CHECK-PWR8-NEXT: lxvd2x v20, r1, r4 # 16-byte Folded Reload ; CHECK-PWR8-NEXT: addi r1, r1, 528 ; CHECK-PWR8-NEXT: ld r0, 16(r1) @@ -255,15 +276,18 @@ ; CHECK-PWR9-NEXT: .cfi_offset r14, -288 ; CHECK-PWR9-NEXT: .cfi_offset f14, -144 ; CHECK-PWR9-NEXT: .cfi_offset v20, -480 +; CHECK-PWR9-NEXT: .cfi_offset v21, -464 ; CHECK-PWR9-NEXT: std r14, 224(r1) # 8-byte Folded Spill ; CHECK-PWR9-NEXT: stfd f14, 368(r1) # 8-byte Folded Spill ; CHECK-PWR9-NEXT: stxv v20, 32(r1) # 16-byte Folded Spill +; CHECK-PWR9-NEXT: stxv v21, 48(r1) # 16-byte Folded Spill ; CHECK-PWR9-NEXT: #APP ; CHECK-PWR9-NEXT: add r3, r3, r4 ; CHECK-PWR9-NEXT: #NO_APP ; CHECK-PWR9-NEXT: extsw r3, r3 ; CHECK-PWR9-NEXT: bl callee ; CHECK-PWR9-NEXT: nop +; CHECK-PWR9-NEXT: lxv v21, 48(r1) # 16-byte Folded Reload ; CHECK-PWR9-NEXT: lxv v20, 32(r1) # 16-byte Folded Reload ; CHECK-PWR9-NEXT: lfd f14, 368(r1) # 8-byte Folded Reload ; CHECK-PWR9-NEXT: ld r14, 224(r1) # 8-byte Folded Reload diff --git a/llvm/test/CodeGen/PowerPC/inline-asm-vsx-clobbers.ll b/llvm/test/CodeGen/PowerPC/inline-asm-vsx-clobbers.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/PowerPC/inline-asm-vsx-clobbers.ll @@ -0,0 +1,32 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py +; RUN: llc -mcpu=pwr9 -mtriple=powerpc64le-unknown-unknown \ +; RUN: -verify-machineinstrs -ppc-vsr-nums-as-vr \ +; RUN: -ppc-asm-full-reg-names < %s | FileCheck %s + +define dso_local void @clobberVR(<4 x i32> %a, <4 x i32> %b) local_unnamed_addr { +; CHECK-LABEL: clobberVR: +; CHECK: # %bb.0: # %entry +; CHECK-NEXT: stxv v22, -160(r1) # 16-byte Folded Spill +; CHECK-NEXT: #APP +; CHECK-NEXT: nop +; CHECK-NEXT: #NO_APP +; CHECK-NEXT: lxv v22, -160(r1) # 16-byte Folded Reload +; CHECK-NEXT: blr +entry: + tail call void asm sideeffect "nop", "~{vs54}"() + ret void +} + +define dso_local void @clobberFPR(<4 x i32> %a, <4 x i32> %b) local_unnamed_addr { +; CHECK-LABEL: clobberFPR: +; CHECK: # %bb.0: # %entry +; CHECK-NEXT: stfd f14, -144(r1) # 8-byte Folded Spill +; CHECK-NEXT: #APP +; CHECK-NEXT: nop +; CHECK-NEXT: #NO_APP +; CHECK-NEXT: lfd f14, -144(r1) # 8-byte Folded Reload +; CHECK-NEXT: blr +entry: + tail call void asm sideeffect "nop", "~{vs14}"() + ret void +}