xref: /aosp_15_r20/external/llvm/test/CodeGen/SystemZ/atomicrmw-xchg-01.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; Test 8-bit atomic exchange.
2*9880d681SAndroid Build Coastguard Worker;
3*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
4*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s -check-prefix=CHECK-SHIFT
5*9880d681SAndroid Build Coastguard Worker
6*9880d681SAndroid Build Coastguard Worker; Check exchange with a variable.
7*9880d681SAndroid Build Coastguard Worker; - CHECK is for the main loop.
8*9880d681SAndroid Build Coastguard Worker; - CHECK-SHIFT makes sure that the negated shift count used by the second
9*9880d681SAndroid Build Coastguard Worker;   RLL is set up correctly.  The negation is independent of the NILL and L
10*9880d681SAndroid Build Coastguard Worker;   tested in CHECK.  CHECK-SHIFT also checks that %r3 is not modified before
11*9880d681SAndroid Build Coastguard Worker;   being used in the RISBG (in contrast to things like atomic addition,
12*9880d681SAndroid Build Coastguard Worker;   which shift %r3 left so that %b is at the high end of the word).
13*9880d681SAndroid Build Coastguard Workerdefine i8 @f1(i8 *%src, i8 %b) {
14*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: f1:
15*9880d681SAndroid Build Coastguard Worker; CHECK: risbg [[RISBG:%r[1-9]+]], %r2, 0, 189, 0{{$}}
16*9880d681SAndroid Build Coastguard Worker; CHECK: sll %r2, 3
17*9880d681SAndroid Build Coastguard Worker; CHECK: l [[OLD:%r[0-9]+]], 0([[RISBG]])
18*9880d681SAndroid Build Coastguard Worker; CHECK: [[LABEL:\.[^:]*]]:
19*9880d681SAndroid Build Coastguard Worker; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0(%r2)
20*9880d681SAndroid Build Coastguard Worker; CHECK: risbg [[ROT]], %r3, 32, 39, 24
21*9880d681SAndroid Build Coastguard Worker; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0({{%r[1-9]+}})
22*9880d681SAndroid Build Coastguard Worker; CHECK: cs [[OLD]], [[NEW]], 0([[RISBG]])
23*9880d681SAndroid Build Coastguard Worker; CHECK: jl [[LABEL]]
24*9880d681SAndroid Build Coastguard Worker; CHECK: rll %r2, [[OLD]], 8(%r2)
25*9880d681SAndroid Build Coastguard Worker; CHECK: br %r14
26*9880d681SAndroid Build Coastguard Worker;
27*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT-LABEL: f1:
28*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT-NOT: %r3
29*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: sll %r2, 3
30*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT-NOT: %r3
31*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: lcr [[NEGSHIFT:%r[1-9]+]], %r2
32*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT-NOT: %r3
33*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: rll
34*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT-NOT: %r3
35*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: risbg {{%r[0-9]+}}, %r3, 32, 39, 24
36*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]])
37*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: rll
38*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: br %r14
39*9880d681SAndroid Build Coastguard Worker  %res = atomicrmw xchg i8 *%src, i8 %b seq_cst
40*9880d681SAndroid Build Coastguard Worker  ret i8 %res
41*9880d681SAndroid Build Coastguard Worker}
42*9880d681SAndroid Build Coastguard Worker
43*9880d681SAndroid Build Coastguard Worker; Check exchange with a constant.  We should force the constant into
44*9880d681SAndroid Build Coastguard Worker; a register and use the sequence above.
45*9880d681SAndroid Build Coastguard Workerdefine i8 @f2(i8 *%src) {
46*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: f2:
47*9880d681SAndroid Build Coastguard Worker; CHECK: lhi [[VALUE:%r[0-9]+]], 88
48*9880d681SAndroid Build Coastguard Worker; CHECK: risbg {{%r[0-9]+}}, [[VALUE]], 32, 39, 24
49*9880d681SAndroid Build Coastguard Worker; CHECK: br %r14
50*9880d681SAndroid Build Coastguard Worker;
51*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT-LABEL: f2:
52*9880d681SAndroid Build Coastguard Worker; CHECK-SHIFT: br %r14
53*9880d681SAndroid Build Coastguard Worker  %res = atomicrmw xchg i8 *%src, i8 88 seq_cst
54*9880d681SAndroid Build Coastguard Worker  ret i8 %res
55*9880d681SAndroid Build Coastguard Worker}
56