[llvm] r355869 - [RISCV] Do a sign-extension in a compare-and-swap of 32 bit in RV64A
Alex Bradbury via llvm-commits
llvm-commits at lists.llvm.org
Mon Mar 11 14:41:22 PDT 2019
Author: asb
Date: Mon Mar 11 14:41:22 2019
New Revision: 355869
URL: http://llvm.org/viewvc/llvm-project?rev=355869&view=rev
Log:
[RISCV] Do a sign-extension in a compare-and-swap of 32 bit in RV64A
AtomicCmpSwapWithSuccess is legalised into an AtomicCmpSwap plus a comparison.
This requires an extension of the value which, by default, is a
zero-extension. When we later lower AtomicCmpSwap into a PseudoCmpXchg32 and then expanded in
RISCVExpandPseudoInsts.cpp, the lr.w instruction does a sign-extension.
This mismatch of extensions causes the comparison to fail when the compared
value is negative. This change overrides TargetLowering::getExtendForAtomicOps
for RISC-V so it does a sign-extension instead.
Differential Revision: https://reviews.llvm.org/D58829
Patch by Ferran Pallarès Roca.
Added:
llvm/trunk/test/CodeGen/RISCV/atomic-cmpxchg-flag.ll
Modified:
llvm/trunk/lib/Target/RISCV/RISCVISelLowering.h
Modified: llvm/trunk/lib/Target/RISCV/RISCVISelLowering.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/RISCV/RISCVISelLowering.h?rev=355869&r1=355868&r2=355869&view=diff
==============================================================================
--- llvm/trunk/lib/Target/RISCV/RISCVISelLowering.h (original)
+++ llvm/trunk/lib/Target/RISCV/RISCVISelLowering.h Mon Mar 11 14:41:22 2019
@@ -106,6 +106,10 @@ public:
Instruction *emitTrailingFence(IRBuilder<> &Builder, Instruction *Inst,
AtomicOrdering Ord) const override;
+ ISD::NodeType getExtendForAtomicOps() const override {
+ return ISD::SIGN_EXTEND;
+ }
+
private:
void analyzeInputArgs(MachineFunction &MF, CCState &CCInfo,
const SmallVectorImpl<ISD::InputArg> &Ins,
Added: llvm/trunk/test/CodeGen/RISCV/atomic-cmpxchg-flag.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/RISCV/atomic-cmpxchg-flag.ll?rev=355869&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/RISCV/atomic-cmpxchg-flag.ll (added)
+++ llvm/trunk/test/CodeGen/RISCV/atomic-cmpxchg-flag.ll Mon Mar 11 14:41:22 2019
@@ -0,0 +1,29 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
+; RUN: llc -mtriple=riscv64 -mattr=+a -verify-machineinstrs < %s \
+; RUN: | FileCheck -check-prefix=RV64IA %s
+
+; This test ensures that the output of the 'lr.w' instruction is sign-extended.
+; Previously, the default zero-extension was being used and 'cmp' parameter
+; higher bits were masked to zero for the comparison.
+
+define i1 @cmpxchg_i32_seq_cst_seq_cst(i32* %ptr, i32 signext %cmp,
+ i32 signext %val) {
+; RV64IA-LABEL: cmpxchg_i32_seq_cst_seq_cst:
+; RV64IA: # %bb.0: # %entry
+; RV64IA-NEXT: .LBB0_1: # %entry
+; RV64IA-NEXT: # =>This Inner Loop Header: Depth=1
+; RV64IA-NEXT: lr.w.aqrl a3, (a0)
+; RV64IA-NEXT: bne a3, a1, .LBB0_3
+; RV64IA-NEXT: # %bb.2: # %entry
+; RV64IA-NEXT: # in Loop: Header=BB0_1 Depth=1
+; RV64IA-NEXT: sc.w.aqrl a4, a2, (a0)
+; RV64IA-NEXT: bnez a4, .LBB0_1
+; RV64IA-NEXT: .LBB0_3: # %entry
+; RV64IA-NEXT: xor a0, a3, a1
+; RV64IA-NEXT: seqz a0, a0
+; RV64IA-NEXT: ret
+entry:
+ %0 = cmpxchg i32* %ptr, i32 %cmp, i32 %val seq_cst seq_cst
+ %1 = extractvalue { i32, i1 } %0, 1
+ ret i1 %1
+}
More information about the llvm-commits
mailing list