1; Test 16-bit atomic additions. 2; 3; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s 4; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s -check-prefix=CHECK-SHIFT1 5; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s -check-prefix=CHECK-SHIFT2 6 7; Check addition of a variable. 8; - CHECK is for the main loop. 9; - CHECK-SHIFT1 makes sure that the negated shift count used by the second 10; RLL is set up correctly. The negation is independent of the NILL and L 11; tested in CHECK. 12; - CHECK-SHIFT2 makes sure that %b is shifted into the high part of the word 13; before being used. This shift is independent of the other loop prologue 14; instructions. 15define i16 @f1(i16 *%src, i16 %b) { 16; CHECK-LABEL: f1: 17; CHECK: risbg [[RISBG:%r[1-9]+]], %r2, 0, 189, 0{{$}} 18; CHECK-DAG: sll %r2, 3 19; CHECK-DAG: l [[OLD:%r[0-9]+]], 0([[RISBG]]) 20; CHECK: [[LABEL:\.[^:]*]]: 21; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0(%r2) 22; CHECK: ar [[ROT]], %r3 23; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0({{%r[1-9]+}}) 24; CHECK: cs [[OLD]], [[NEW]], 0([[RISBG]]) 25; CHECK: jl [[LABEL]] 26; CHECK: rll %r2, [[OLD]], 16(%r2) 27; CHECK: br %r14 28; 29; CHECK-SHIFT1-LABEL: f1: 30; CHECK-SHIFT1: sll [[SHIFT:%r[1-9]+]], 3 31; CHECK-SHIFT1: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]] 32; CHECK-SHIFT1: rll 33; CHECK-SHIFT1: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]]) 34; CHECK-SHIFT1: rll 35; CHECK-SHIFT1: br %r14 36; 37; CHECK-SHIFT2-LABEL: f1: 38; CHECK-SHIFT2: sll %r3, 16 39; CHECK-SHIFT2: rll 40; CHECK-SHIFT2: ar {{%r[0-9]+}}, %r3 41; CHECK-SHIFT2: rll 42; CHECK-SHIFT2: rll 43; CHECK-SHIFT2: br %r14 44 %res = atomicrmw add i16 *%src, i16 %b seq_cst 45 ret i16 %res 46} 47 48; Check the minimum signed value. We add 0x80000000 to the rotated word. 49define i16 @f2(i16 *%src) { 50; CHECK-LABEL: f2: 51; CHECK: risbg [[RISBG:%r[1-9]+]], %r2, 0, 189, 0 52; CHECK-DAG: sll %r2, 3 53; CHECK-DAG: l [[OLD:%r[0-9]+]], 0([[RISBG]]) 54; CHECK: [[LABEL:\.[^:]*]]: 55; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0(%r2) 56; CHECK: afi [[ROT]], -2147483648 57; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0([[NEGSHIFT:%r[1-9]+]]) 58; CHECK: cs [[OLD]], [[NEW]], 0([[RISBG]]) 59; CHECK: jl [[LABEL]] 60; CHECK: rll %r2, [[OLD]], 16(%r2) 61; CHECK: br %r14 62; 63; CHECK-SHIFT1-LABEL: f2: 64; CHECK-SHIFT1: sll [[SHIFT:%r[1-9]+]], 3 65; CHECK-SHIFT1: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]] 66; CHECK-SHIFT1: rll 67; CHECK-SHIFT1: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]]) 68; CHECK-SHIFT1: rll 69; CHECK-SHIFT1: br %r14 70; 71; CHECK-SHIFT2-LABEL: f2: 72; CHECK-SHIFT2: br %r14 73 %res = atomicrmw add i16 *%src, i16 -32768 seq_cst 74 ret i16 %res 75} 76 77; Check addition of -1. We add 0xffff0000 to the rotated word. 78define i16 @f3(i16 *%src) { 79; CHECK-LABEL: f3: 80; CHECK: afi [[ROT]], -65536 81; CHECK: br %r14 82; 83; CHECK-SHIFT1-LABEL: f3: 84; CHECK-SHIFT1: br %r14 85; CHECK-SHIFT2-LABEL: f3: 86; CHECK-SHIFT2: br %r14 87 %res = atomicrmw add i16 *%src, i16 -1 seq_cst 88 ret i16 %res 89} 90 91; Check addition of 1. We add 0x00010000 to the rotated word. 92define i16 @f4(i16 *%src) { 93; CHECK-LABEL: f4: 94; CHECK: afi [[ROT]], 65536 95; CHECK: br %r14 96; 97; CHECK-SHIFT1-LABEL: f4: 98; CHECK-SHIFT1: br %r14 99; CHECK-SHIFT2-LABEL: f4: 100; CHECK-SHIFT2: br %r14 101 %res = atomicrmw add i16 *%src, i16 1 seq_cst 102 ret i16 %res 103} 104 105; Check the maximum signed value. We add 0x7fff0000 to the rotated word. 106define i16 @f5(i16 *%src) { 107; CHECK-LABEL: f5: 108; CHECK: afi [[ROT]], 2147418112 109; CHECK: br %r14 110; 111; CHECK-SHIFT1-LABEL: f5: 112; CHECK-SHIFT1: br %r14 113; CHECK-SHIFT2-LABEL: f5: 114; CHECK-SHIFT2: br %r14 115 %res = atomicrmw add i16 *%src, i16 32767 seq_cst 116 ret i16 %res 117} 118 119; Check addition of a large unsigned value. We add 0xfffe0000 to the 120; rotated word, expressed as a negative AFI operand. 121define i16 @f6(i16 *%src) { 122; CHECK-LABEL: f6: 123; CHECK: afi [[ROT]], -131072 124; CHECK: br %r14 125; 126; CHECK-SHIFT1-LABEL: f6: 127; CHECK-SHIFT1: br %r14 128; CHECK-SHIFT2-LABEL: f6: 129; CHECK-SHIFT2: br %r14 130 %res = atomicrmw add i16 *%src, i16 65534 seq_cst 131 ret i16 %res 132} 133