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). 13 define i16 @f1(i16 %dummy, i16 *%src, i16 %cmp, i16 %swap) { 14 ; CHECK-MAIN-LABEL: f1: 15 ; CHECK-MAIN: sllg [[SHIFT:%r[1-9]+]], %r3, 3 16 ; CHECK-MAIN: nill %r3, 65532 17 ; CHECK-MAIN: l [[OLD:%r[0-9]+]], 0(%r3) 18 ; CHECK-MAIN: [[LOOP:\.[^ ]*]]: 19 ; CHECK-MAIN: rll %r2, [[OLD]], 16([[SHIFT]]) 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(%r3) 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: sllg [[SHIFT:%r[1-9]+]], %r3, 3 32 ; CHECK-SHIFT: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]] 33 ; CHECK-SHIFT: rll 34 ; CHECK-SHIFT: rll {{%r[0-9]+}}, %r5, -16([[NEGSHIFT]]) 35 %res = cmpxchg i16 *%src, i16 %cmp, i16 %swap seq_cst 36 ret i16 %res 37 } 38 39 ; Check compare and swap with constants. We should force the constants into 40 ; registers and use the sequence above. 41 define i16 @f2(i16 *%src) { 42 ; CHECK-LABEL: f2: 43 ; CHECK: lhi [[CMP:%r[0-9]+]], 42 44 ; CHECK: risbg [[CMP]], {{%r[0-9]+}}, 32, 47, 0 45 ; CHECK: risbg 46 ; CHECK: br %r14 47 ; 48 ; CHECK-SHIFT-LABEL: f2: 49 ; CHECK-SHIFT: lhi [[SWAP:%r[0-9]+]], 88 50 ; CHECK-SHIFT: risbg 51 ; CHECK-SHIFT: risbg [[SWAP]], {{%r[0-9]+}}, 32, 47, 0 52 ; CHECK-SHIFT: br %r14 53 %res = cmpxchg i16 *%src, i16 42, i16 88 seq_cst 54 ret i16 %res 55 } 56