Skip to content

Commit

Permalink
[RISCV] Don't allow X0 to be used for 'r' constraint in inline assembly
Browse files Browse the repository at this point in the history
Some instructions treat x0 as a special encoding rather than as a
value of 0. Since we don't parse the inline assembly to know what
the instruction is, chooser the safest option of never using x0.

Fixes #63747.

Reviewed By: asb

Differential Revision: https://reviews.llvm.org/D154744
  • Loading branch information
topperc committed Jul 10, 2023
1 parent 0b69cc8 commit dbd47c4
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 1 deletion.
2 changes: 1 addition & 1 deletion llvm/lib/Target/RISCV/RISCVISelLowering.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -15845,7 +15845,7 @@ RISCVTargetLowering::getRegForInlineAsmConstraint(const TargetRegisterInfo *TRI,
// TODO: Support fixed vectors up to XLen for P extension?
if (VT.isVector())
break;
return std::make_pair(0U, &RISCV::GPRRegClass);
return std::make_pair(0U, &RISCV::GPRNoX0RegClass);
case 'f':
if (Subtarget.hasStdExtZfhOrZfhmin() && VT == MVT::f16)
return std::make_pair(0U, &RISCV::FPR16RegClass);
Expand Down
27 changes: 27 additions & 0 deletions llvm/test/CodeGen/RISCV/inline-asm.ll
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,33 @@ define i32 @constraint_r(i32 %a) nounwind {
ret i32 %2
}

; Don't allow 'x0' for 'r'. Some instructions have a different behavior when
; x0 is encoded.
define i32 @constraint_r_zero(i32 %a) nounwind {
; RV32I-LABEL: constraint_r_zero:
; RV32I: # %bb.0:
; RV32I-NEXT: lui a0, %hi(gi)
; RV32I-NEXT: lw a0, %lo(gi)(a0)
; RV32I-NEXT: li a1, 0
; RV32I-NEXT: #APP
; RV32I-NEXT: add a0, a1, a0
; RV32I-NEXT: #NO_APP
; RV32I-NEXT: ret
;
; RV64I-LABEL: constraint_r_zero:
; RV64I: # %bb.0:
; RV64I-NEXT: lui a0, %hi(gi)
; RV64I-NEXT: lw a0, %lo(gi)(a0)
; RV64I-NEXT: li a1, 0
; RV64I-NEXT: #APP
; RV64I-NEXT: add a0, a1, a0
; RV64I-NEXT: #NO_APP
; RV64I-NEXT: ret
%1 = load i32, ptr @gi
%2 = tail call i32 asm "add $0, $1, $2", "=r,r,r"(i32 0, i32 %1)
ret i32 %2
}

define i32 @constraint_i(i32 %a) nounwind {
; RV32I-LABEL: constraint_i:
; RV32I: # %bb.0:
Expand Down

0 comments on commit dbd47c4

Please sign in to comment.