test/CodeGen/SystemZ/atomicrmw-xchg-04.ll

   1 ; Test 64-bit atomic exchange.
   2 ;
   3 ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
   4
   5 ; Check register exchange.
   6 define i64 @f1(i64 %dummy, i64 *%src, i64 %b) {
   7 ; CHECK-LABEL: f1:
   8 ; CHECK: lg %r2, 0(%r3)
   9 ; CHECK: [[LABEL:\.[^:]*]]:
  10 ; CHECK: csg %r2, %r4, 0(%r3)
  11 ; CHECK: jl [[LABEL]]
  12 ; CHECK: br %r14
  13   %res = atomicrmw xchg i64 *%src, i64 %b seq_cst
  14   ret i64 %res
  15 }
  16
  17 ; Check the high end of the aligned CSG range.
  18 define i64 @f2(i64 %dummy, i64 *%src, i64 %b) {
  19 ; CHECK-LABEL: f2:
  20 ; CHECK: lg %r2, 524280(%r3)
  21 ; CHECK: csg %r2, {{%r[0-9]+}}, 524280(%r3)
  22 ; CHECK: br %r14
  23   %ptr = getelementptr i64 *%src, i64 65535
  24   %res = atomicrmw xchg i64 *%ptr, i64 %b seq_cst
  25   ret i64 %res
  26 }
  27
  28 ; Check the next doubleword up, which requires separate address logic.
  29 define i64 @f3(i64 %dummy, i64 *%src, i64 %b) {
  30 ; CHECK-LABEL: f3:
  31 ; CHECK: agfi %r3, 524288
  32 ; CHECK: lg %r2, 0(%r3)
  33 ; CHECK: csg %r2, {{%r[0-9]+}}, 0(%r3)
  34 ; CHECK: br %r14
  35   %ptr = getelementptr i64 *%src, i64 65536
  36   %res = atomicrmw xchg i64 *%ptr, i64 %b seq_cst
  37   ret i64 %res
  38 }
  39
  40 ; Check the low end of the CSG range.
  41 define i64 @f4(i64 %dummy, i64 *%src, i64 %b) {
  42 ; CHECK-LABEL: f4:
  43 ; CHECK: lg %r2, -524288(%r3)
  44 ; CHECK: csg %r2, {{%r[0-9]+}}, -524288(%r3)
  45 ; CHECK: br %r14
  46   %ptr = getelementptr i64 *%src, i64 -65536
  47   %res = atomicrmw xchg i64 *%ptr, i64 %b seq_cst
  48   ret i64 %res
  49 }
  50
  51 ; Check the next doubleword down, which requires separate address logic.
  52 define i64 @f5(i64 %dummy, i64 *%src, i64 %b) {
  53 ; CHECK-LABEL: f5:
  54 ; CHECK: agfi %r3, -524296
  55 ; CHECK: lg %r2, 0(%r3)
  56 ; CHECK: csg %r2, {{%r[0-9]+}}, 0(%r3)
  57 ; CHECK: br %r14
  58   %ptr = getelementptr i64 *%src, i64 -65537
  59   %res = atomicrmw xchg i64 *%ptr, i64 %b seq_cst
  60   ret i64 %res
  61 }
  62
  63 ; Check that indexed addresses are not allowed.
  64 define i64 @f6(i64 %dummy, i64 %base, i64 %index, i64 %b) {
  65 ; CHECK-LABEL: f6:
  66 ; CHECK: agr %r3, %r4
  67 ; CHECK: lg %r2, 0(%r3)
  68 ; CHECK: csg %r2, {{%r[0-9]+}}, 0(%r3)
  69 ; CHECK: br %r14
  70   %add = add i64 %base, %index
  71   %ptr = inttoptr i64 %add to i64 *
  72   %res = atomicrmw xchg i64 *%ptr, i64 %b seq_cst
  73   ret i64 %res
  74 }
  75
  76 ; Check exchange of a constant.  We should force it into a register and
  77 ; use the sequence above.
  78 define i64 @f7(i64 %dummy, i64 *%ptr) {
  79 ; CHECK-LABEL: f7:
  80 ; CHECK: llilf [[VALUE:%r[0-9+]]], 3000000000
  81 ; CHECK: lg %r2, 0(%r3)
  82 ; CHECK: [[LABEL:\.[^:]*]]:
  83 ; CHECK: csg %r2, [[VALUE]], 0(%r3)
  84 ; CHECK: jl [[LABEL]]
  85 ; CHECK: br %r14
  86   %res = atomicrmw xchg i64 *%ptr, i64 3000000000 seq_cst
  87   ret i64 %res
  88 }