Home | History | Annotate | Download | only in SystemZ
      1 ; Test 8-bit atomic min/max operations.
      2 ;
      3 ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s -check-prefix=CHECK
      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 signed minimum.
      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, and that the low bits are set to 1.  This sequence is
     14 ;   independent of the other loop prologue instructions.
     15 define i8 @f1(i8 *%src, i8 %b) {
     16 ; CHECK-LABEL: f1:
     17 ; CHECK: sllg [[SHIFT:%r[1-9]+]], %r2, 3
     18 ; CHECK: nill %r2, 65532
     19 ; CHECK: l [[OLD:%r[0-9]+]], 0(%r2)
     20 ; CHECK: [[LOOP:\.[^:]*]]:
     21 ; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0([[SHIFT]])
     22 ; CHECK: crjle [[ROT]], %r3, [[KEEP:\..*]]
     23 ; CHECK: risbg [[ROT]], %r3, 32, 39, 0
     24 ; CHECK: [[KEEP]]:
     25 ; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0({{%r[1-9]+}})
     26 ; CHECK: cs [[OLD]], [[NEW]], 0(%r2)
     27 ; CHECK: jl [[LOOP]]
     28 ; CHECK: rll %r2, [[OLD]], 8([[SHIFT]])
     29 ; CHECK: br %r14
     30 ;
     31 ; CHECK-SHIFT1-LABEL: f1:
     32 ; CHECK-SHIFT1: sllg [[SHIFT:%r[1-9]+]], %r2, 3
     33 ; CHECK-SHIFT1: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]]
     34 ; CHECK-SHIFT1: rll
     35 ; CHECK-SHIFT1: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]])
     36 ; CHECK-SHIFT1: rll
     37 ; CHECK-SHIFT1: br %r14
     38 ;
     39 ; CHECK-SHIFT2-LABEL: f1:
     40 ; CHECK-SHIFT2: sll %r3, 24
     41 ; CHECK-SHIFT2: rll
     42 ; CHECK-SHIFT2: crjle {{%r[0-9]+}}, %r3
     43 ; CHECK-SHIFT2: rll
     44 ; CHECK-SHIFT2: rll
     45 ; CHECK-SHIFT2: br %r14
     46   %res = atomicrmw min i8 *%src, i8 %b seq_cst
     47   ret i8 %res
     48 }
     49 
     50 ; Check signed maximum.
     51 define i8 @f2(i8 *%src, i8 %b) {
     52 ; CHECK-LABEL: f2:
     53 ; CHECK: sllg [[SHIFT:%r[1-9]+]], %r2, 3
     54 ; CHECK: nill %r2, 65532
     55 ; CHECK: l [[OLD:%r[0-9]+]], 0(%r2)
     56 ; CHECK: [[LOOP:\.[^:]*]]:
     57 ; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0([[SHIFT]])
     58 ; CHECK: crjhe [[ROT]], %r3, [[KEEP:\..*]]
     59 ; CHECK: risbg [[ROT]], %r3, 32, 39, 0
     60 ; CHECK: [[KEEP]]:
     61 ; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0({{%r[1-9]+}})
     62 ; CHECK: cs [[OLD]], [[NEW]], 0(%r2)
     63 ; CHECK: jl [[LOOP]]
     64 ; CHECK: rll %r2, [[OLD]], 8([[SHIFT]])
     65 ; CHECK: br %r14
     66 ;
     67 ; CHECK-SHIFT1-LABEL: f2:
     68 ; CHECK-SHIFT1: sllg [[SHIFT:%r[1-9]+]], %r2, 3
     69 ; CHECK-SHIFT1: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]]
     70 ; CHECK-SHIFT1: rll
     71 ; CHECK-SHIFT1: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]])
     72 ; CHECK-SHIFT1: rll
     73 ; CHECK-SHIFT1: br %r14
     74 ;
     75 ; CHECK-SHIFT2-LABEL: f2:
     76 ; CHECK-SHIFT2: sll %r3, 24
     77 ; CHECK-SHIFT2: rll
     78 ; CHECK-SHIFT2: crjhe {{%r[0-9]+}}, %r3
     79 ; CHECK-SHIFT2: rll
     80 ; CHECK-SHIFT2: rll
     81 ; CHECK-SHIFT2: br %r14
     82   %res = atomicrmw max i8 *%src, i8 %b seq_cst
     83   ret i8 %res
     84 }
     85 
     86 ; Check unsigned minimum.
     87 define i8 @f3(i8 *%src, i8 %b) {
     88 ; CHECK-LABEL: f3:
     89 ; CHECK: sllg [[SHIFT:%r[1-9]+]], %r2, 3
     90 ; CHECK: nill %r2, 65532
     91 ; CHECK: l [[OLD:%r[0-9]+]], 0(%r2)
     92 ; CHECK: [[LOOP:\.[^:]*]]:
     93 ; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0([[SHIFT]])
     94 ; CHECK: clr [[ROT]], %r3
     95 ; CHECK: jle [[KEEP:\..*]]
     96 ; CHECK: risbg [[ROT]], %r3, 32, 39, 0
     97 ; CHECK: [[KEEP]]:
     98 ; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0({{%r[1-9]+}})
     99 ; CHECK: cs [[OLD]], [[NEW]], 0(%r2)
    100 ; CHECK: jl [[LOOP]]
    101 ; CHECK: rll %r2, [[OLD]], 8([[SHIFT]])
    102 ; CHECK: br %r14
    103 ;
    104 ; CHECK-SHIFT1-LABEL: f3:
    105 ; CHECK-SHIFT1: sllg [[SHIFT:%r[1-9]+]], %r2, 3
    106 ; CHECK-SHIFT1: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]]
    107 ; CHECK-SHIFT1: rll
    108 ; CHECK-SHIFT1: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]])
    109 ; CHECK-SHIFT1: rll
    110 ; CHECK-SHIFT1: br %r14
    111 ;
    112 ; CHECK-SHIFT2-LABEL: f3:
    113 ; CHECK-SHIFT2: sll %r3, 24
    114 ; CHECK-SHIFT2: rll
    115 ; CHECK-SHIFT2: clr {{%r[0-9]+}}, %r3
    116 ; CHECK-SHIFT2: rll
    117 ; CHECK-SHIFT2: rll
    118 ; CHECK-SHIFT2: br %r14
    119   %res = atomicrmw umin i8 *%src, i8 %b seq_cst
    120   ret i8 %res
    121 }
    122 
    123 ; Check unsigned maximum.
    124 define i8 @f4(i8 *%src, i8 %b) {
    125 ; CHECK-LABEL: f4:
    126 ; CHECK: sllg [[SHIFT:%r[1-9]+]], %r2, 3
    127 ; CHECK: nill %r2, 65532
    128 ; CHECK: l [[OLD:%r[0-9]+]], 0(%r2)
    129 ; CHECK: [[LOOP:\.[^:]*]]:
    130 ; CHECK: rll [[ROT:%r[0-9]+]], [[OLD]], 0([[SHIFT]])
    131 ; CHECK: clr [[ROT]], %r3
    132 ; CHECK: jhe [[KEEP:\..*]]
    133 ; CHECK: risbg [[ROT]], %r3, 32, 39, 0
    134 ; CHECK: [[KEEP]]:
    135 ; CHECK: rll [[NEW:%r[0-9]+]], [[ROT]], 0({{%r[1-9]+}})
    136 ; CHECK: cs [[OLD]], [[NEW]], 0(%r2)
    137 ; CHECK: jl [[LOOP]]
    138 ; CHECK: rll %r2, [[OLD]], 8([[SHIFT]])
    139 ; CHECK: br %r14
    140 ;
    141 ; CHECK-SHIFT1-LABEL: f4:
    142 ; CHECK-SHIFT1: sllg [[SHIFT:%r[1-9]+]], %r2, 3
    143 ; CHECK-SHIFT1: lcr [[NEGSHIFT:%r[1-9]+]], [[SHIFT]]
    144 ; CHECK-SHIFT1: rll
    145 ; CHECK-SHIFT1: rll {{%r[0-9]+}}, {{%r[0-9]+}}, 0([[NEGSHIFT]])
    146 ; CHECK-SHIFT1: rll
    147 ; CHECK-SHIFT1: br %r14
    148 ;
    149 ; CHECK-SHIFT2-LABEL: f4:
    150 ; CHECK-SHIFT2: sll %r3, 24
    151 ; CHECK-SHIFT2: rll
    152 ; CHECK-SHIFT2: clr {{%r[0-9]+}}, %r3
    153 ; CHECK-SHIFT2: rll
    154 ; CHECK-SHIFT2: rll
    155 ; CHECK-SHIFT2: br %r14
    156   %res = atomicrmw umax i8 *%src, i8 %b seq_cst
    157   ret i8 %res
    158 }
    159 
    160 ; Check the lowest useful signed minimum value.  We need to load 0x81000000
    161 ; into the source register.
    162 define i8 @f5(i8 *%src) {
    163 ; CHECK-LABEL: f5:
    164 ; CHECK: llilh [[SRC2:%r[0-9]+]], 33024
    165 ; CHECK: crjle [[ROT:%r[0-9]+]], [[SRC2]]
    166 ; CHECK: risbg [[ROT]], [[SRC2]], 32, 39, 0
    167 ; CHECK: br %r14
    168 ;
    169 ; CHECK-SHIFT1-LABEL: f5:
    170 ; CHECK-SHIFT1: br %r14
    171 ; CHECK-SHIFT2-LABEL: f5:
    172 ; CHECK-SHIFT2: br %r14
    173   %res = atomicrmw min i8 *%src, i8 -127 seq_cst
    174   ret i8 %res
    175 }
    176 
    177 ; Check the highest useful signed maximum value.  We need to load 0x7e000000
    178 ; into the source register.
    179 define i8 @f6(i8 *%src) {
    180 ; CHECK-LABEL: f6:
    181 ; CHECK: llilh [[SRC2:%r[0-9]+]], 32256
    182 ; CHECK: crjhe [[ROT:%r[0-9]+]], [[SRC2]]
    183 ; CHECK: risbg [[ROT]], [[SRC2]], 32, 39, 0
    184 ; CHECK: br %r14
    185 ;
    186 ; CHECK-SHIFT1-LABEL: f6:
    187 ; CHECK-SHIFT1: br %r14
    188 ; CHECK-SHIFT2-LABEL: f6:
    189 ; CHECK-SHIFT2: br %r14
    190   %res = atomicrmw max i8 *%src, i8 126 seq_cst
    191   ret i8 %res
    192 }
    193 
    194 ; Check the lowest useful unsigned minimum value.  We need to load 0x01000000
    195 ; into the source register.
    196 define i8 @f7(i8 *%src) {
    197 ; CHECK-LABEL: f7:
    198 ; CHECK: llilh [[SRC2:%r[0-9]+]], 256
    199 ; CHECK: clr [[ROT:%r[0-9]+]], [[SRC2]]
    200 ; CHECK: risbg [[ROT]], [[SRC2]], 32, 39, 0
    201 ; CHECK: br %r14
    202 ;
    203 ; CHECK-SHIFT1-LABEL: f7:
    204 ; CHECK-SHIFT1: br %r14
    205 ; CHECK-SHIFT2-LABEL: f7:
    206 ; CHECK-SHIFT2: br %r14
    207   %res = atomicrmw umin i8 *%src, i8 1 seq_cst
    208   ret i8 %res
    209 }
    210 
    211 ; Check the highest useful unsigned maximum value.  We need to load 0xfe000000
    212 ; into the source register.
    213 define i8 @f8(i8 *%src) {
    214 ; CHECK-LABEL: f8:
    215 ; CHECK: llilh [[SRC2:%r[0-9]+]], 65024
    216 ; CHECK: clr [[ROT:%r[0-9]+]], [[SRC2]]
    217 ; CHECK: risbg [[ROT]], [[SRC2]], 32, 39, 0
    218 ; CHECK: br %r14
    219 ;
    220 ; CHECK-SHIFT1-LABEL: f8:
    221 ; CHECK-SHIFT1: br %r14
    222 ; CHECK-SHIFT2-LABEL: f8:
    223 ; CHECK-SHIFT2: br %r14
    224   %res = atomicrmw umax i8 *%src, i8 254 seq_cst
    225   ret i8 %res
    226 }
    227