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 92declare void @g() 93 94; Check using the comparison result for a branch. 95; CHECK-LABEL: f4 96; CHECK-MAIN-LABEL: f4: 97; CHECK-MAIN: risbg [[RISBG:%r[1-9]+]], %r2, 0, 189, 0{{$}} 98; CHECK-MAIN-DAG: sll %r2, 3 99; CHECK-MAIN-DAG: l [[OLD:%r[0-9]+]], 0([[RISBG]]) 100; CHECK-MAIN: [[LOOP:\.[^ ]*]]: 101; CHECK-MAIN: rll [[TMP:%r[0-9]+]], [[OLD]], 16(%r2) 102; CHECK-MAIN: risbg %r3, [[TMP]], 32, 47, 0 103; CHECK-MAIN: cr [[TMP]], %r3 104; CHECK-MAIN: jlh [[EXIT:\.[^ ]*]] 105; CHECK-MAIN: risbg %r4, [[TMP]], 32, 47, 0 106; CHECK-MAIN: rll [[NEW:%r[0-9]+]], %r4, -16({{%r[1-9]+}}) 107; CHECK-MAIN: cs [[OLD]], [[NEW]], 0([[RISBG]]) 108; CHECK-MAIN: jl [[LOOP]] 109; CHECK-MAIN: [[EXIT]]: 110; CHECK-MAIN-NEXT: jlh [[LABEL:\.[^ ]*]] 111; CHECK-MAIN: jg g 112; CHECK-MAIN: [[LABEL]]: 113; CHECK-MAIN: br %r14 114; 115; CHECK-SHIFT-LABEL: f4: 116; CHECK-SHIFT: sll %r2, 3 117; CHECK-SHIFT: lcr [[NEGSHIFT:%r[1-9]+]], %r2 118; CHECK-SHIFT: rll 119; CHECK-SHIFT: rll {{%r[0-9]+}}, %r4, -16([[NEGSHIFT]]) 120define void @f4(i16 *%src, i16 %cmp, i16 %swap) { 121 %pair = cmpxchg i16 *%src, i16 %cmp, i16 %swap seq_cst seq_cst 122 %cond = extractvalue { i16, i1 } %pair, 1 123 br i1 %cond, label %call, label %exit 124 125call: 126 tail call void @g() 127 br label %exit 128 129exit: 130 ret void 131} 132 133; ... and the same with the inverted direction. 134; CHECK-MAIN-LABEL: f5: 135; CHECK-MAIN: risbg [[RISBG:%r[1-9]+]], %r2, 0, 189, 0{{$}} 136; CHECK-MAIN-DAG: sll %r2, 3 137; CHECK-MAIN-DAG: l [[OLD:%r[0-9]+]], 0([[RISBG]]) 138; CHECK-MAIN: [[LOOP:\.[^ ]*]]: 139; CHECK-MAIN: rll [[TMP:%r[0-9]+]], [[OLD]], 16(%r2) 140; CHECK-MAIN: risbg %r3, [[TMP]], 32, 47, 0 141; CHECK-MAIN: cr [[TMP]], %r3 142; CHECK-MAIN: jlh [[EXIT:\.[^ ]*]] 143; CHECK-MAIN: risbg %r4, [[TMP]], 32, 47, 0 144; CHECK-MAIN: rll [[NEW:%r[0-9]+]], %r4, -16({{%r[1-9]+}}) 145; CHECK-MAIN: cs [[OLD]], [[NEW]], 0([[RISBG]]) 146; CHECK-MAIN: jl [[LOOP]] 147; CHECK-MAIN: [[EXIT]]: 148; CHECK-MAIN-NEXT: jlh [[LABEL:\.[^ ]*]] 149; CHECK-MAIN: br %r14 150; CHECK-MAIN: [[LABEL]]: 151; CHECK-MAIN: jg g 152; 153; CHECK-SHIFT-LABEL: f5: 154; CHECK-SHIFT: sll %r2, 3 155; CHECK-SHIFT: lcr [[NEGSHIFT:%r[1-9]+]], %r2 156; CHECK-SHIFT: rll 157; CHECK-SHIFT: rll {{%r[0-9]+}}, %r4, -16([[NEGSHIFT]]) 158define void @f5(i16 *%src, i16 %cmp, i16 %swap) { 159 %pair = cmpxchg i16 *%src, i16 %cmp, i16 %swap seq_cst seq_cst 160 %cond = extractvalue { i16, i1 } %pair, 1 161 br i1 %cond, label %exit, label %call 162 163call: 164 tail call void @g() 165 br label %exit 166 167exit: 168 ret void 169} 170 171