1; Test 16-bit compare and swap.
2;
3; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s -check-prefix=CHECK-MAIN
4; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s -check-prefix=CHECK-SHIFT
5
6; Check compare and swap with a variable.
7; - CHECK is for the main loop.
8; - CHECK-SHIFT makes sure that the negated shift count used by the second
9;   RLL is set up correctly.  The negation is independent of the NILL and L
10;   tested in CHECK.  CHECK-SHIFT also checks that %r3 is not modified before
11;   being used in the RISBG (in contrast to things like atomic addition,
12;   which shift %r3 left so that %b is at the high end of the word).
13define i16 @f1(i16 %dummy, i16 *%src, i16 %cmp, i16 %swap) {
14; CHECK-MAIN-LABEL: f1:
15; CHECK-MAIN: risbg [[RISBG:%r[1-9]+]], %r3, 0, 189, 0{{$}}
16; CHECK-MAIN-DAG: sll %r3, 3
17; CHECK-MAIN-DAG: l [[OLD:%r[0-9]+]], 0([[RISBG]])
18; CHECK-MAIN: [[LOOP:\.[^ ]*]]:
19; CHECK-MAIN: rll %r2, [[OLD]], 16(%r3)
20; CHECK-MAIN: risbg %r4, %r2, 32, 47, 0
21; CHECK-MAIN: crjlh %r2, %r4, [[EXIT:\.[^ ]*]]
22; CHECK-MAIN: risbg %r5, %r2, 32, 47, 0
23; CHECK-MAIN: rll [[NEW:%r[0-9]+]], %r5, -16({{%r[1-9]+}})
24; CHECK-MAIN: cs [[OLD]], [[NEW]], 0([[RISBG]])
25; CHECK-MAIN: jl [[LOOP]]
26; CHECK-MAIN: [[EXIT]]:
27; CHECK-MAIN-NOT: %r2
28; CHECK-MAIN: br %r14
29;
30; CHECK-SHIFT-LABEL: f1:
31; CHECK-SHIFT: sll %r3, 3
32; CHECK-SHIFT: lcr [[NEGSHIFT:%r[1-9]+]], %r3
33; CHECK-SHIFT: rll
34; CHECK-SHIFT: rll {{%r[0-9]+}}, %r5, -16([[NEGSHIFT]])
35  %pair = cmpxchg i16 *%src, i16 %cmp, i16 %swap seq_cst seq_cst
36  %res = extractvalue { i16, i1 } %pair, 0
37  ret i16 %res
38}
39
40; Check compare and swap with constants.  We should force the constants into
41; registers and use the sequence above.
42define i16 @f2(i16 *%src) {
43; CHECK-LABEL: f2:
44; CHECK: lhi [[CMP:%r[0-9]+]], 42
45; CHECK: risbg [[CMP]], {{%r[0-9]+}}, 32, 47, 0
46; CHECK: risbg
47; CHECK: br %r14
48;
49; CHECK-SHIFT-LABEL: f2:
50; CHECK-SHIFT: lhi [[SWAP:%r[0-9]+]], 88
51; CHECK-SHIFT: risbg
52; CHECK-SHIFT: risbg [[SWAP]], {{%r[0-9]+}}, 32, 47, 0
53; CHECK-SHIFT: br %r14
54  %pair = cmpxchg i16 *%src, i16 42, i16 88 seq_cst seq_cst
55  %res = extractvalue { i16, i1 } %pair, 0
56  ret i16 %res
57}
58
59; Check generating the comparison result.
60define i32 @f3(i16 %dummy, i16 *%src, i16 %cmp, i16 %swap) {
61; CHECK-MAIN-LABEL: f3:
62; CHECK-MAIN: risbg [[RISBG:%r[1-9]+]], %r3, 0, 189, 0{{$}}
63; CHECK-MAIN-DAG: sll %r3, 3
64; CHECK-MAIN-DAG: l [[OLD:%r[0-9]+]], 0([[RISBG]])
65; CHECK-MAIN: [[LOOP:\.[^ ]*]]:
66; CHECK-MAIN: rll [[TMP:%r[0-9]+]], [[OLD]], 16(%r3)
67; CHECK-MAIN: risbg %r4, [[TMP]], 32, 47, 0
68; CHECK-MAIN: cr [[TMP]], %r4
69; CHECK-MAIN: jlh [[EXIT:\.[^ ]*]]
70; CHECK-MAIN: risbg %r5, [[TMP]], 32, 47, 0
71; CHECK-MAIN: rll [[NEW:%r[0-9]+]], %r5, -16({{%r[1-9]+}})
72; CHECK-MAIN: cs [[OLD]], [[NEW]], 0([[RISBG]])
73; CHECK-MAIN: jl [[LOOP]]
74; CHECK-MAIN: [[EXIT]]:
75; CHECK-MAIN-NEXT: ipm %r2
76; CHECK-MAIN-NEXT: afi %r2, -268435456
77; CHECK-MAIN-NEXT: srl %r2, 31
78; CHECK-MAIN-NOT: %r2
79; CHECK-MAIN: br %r14
80;
81; CHECK-SHIFT-LABEL: f3:
82; CHECK-SHIFT: sll %r3, 3
83; CHECK-SHIFT: lcr [[NEGSHIFT:%r[1-9]+]], %r3
84; CHECK-SHIFT: rll
85; CHECK-SHIFT: rll {{%r[0-9]+}}, %r5, -16([[NEGSHIFT]])
86  %pair = cmpxchg i16 *%src, i16 %cmp, i16 %swap seq_cst seq_cst
87  %val = extractvalue { i16, i1 } %pair, 1
88  %res = zext i1 %val to i32
89  ret i32 %res
90}
91
92