llvm/test/CodeGen/SystemZ/int-uadd-06.ll

   1 ; Test the three-operand form of 32-bit addition.
   2 ;
   3 ; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z196 | FileCheck %s
   4
   5 declare i32 @foo(i32, i32, i32)
   6
   7 ; Check ALRK.
   8 define i32 @f1(i32 %dummy, i32 %a, i32 %b, i32 *%flag) {
   9 ; CHECK-LABEL: f1:
  10 ; CHECK: alrk %r2, %r3, %r4
  11 ; CHECK: ipm [[REG1:%r[0-5]]]
  12 ; CHECK: risblg [[REG2:%r[0-5]]], [[REG1]], 31, 159, 35
  13 ; CHECK: st [[REG2]], 0(%r5)
  14 ; CHECK: br %r14
  15   %t = call {i32, i1} @llvm.uadd.with.overflow.i32(i32 %a, i32 %b)
  16   %val = extractvalue {i32, i1} %t, 0
  17   %obit = extractvalue {i32, i1} %t, 1
  18   %ext = zext i1 %obit to i32
  19   store i32 %ext, i32 *%flag
  20   ret i32 %val
  21 }
  22
  23 ; Check using the overflow result for a branch.
  24 define i32 @f2(i32 %dummy, i32 %a, i32 %b) {
  25 ; CHECK-LABEL: f2:
  26 ; CHECK: alrk %r2, %r3, %r4
  27 ; CHECK-NEXT: bler %r14
  28 ; CHECK: lhi %r2, 0
  29 ; CHECK: jg foo@PLT
  30   %t = call {i32, i1} @llvm.uadd.with.overflow.i32(i32 %a, i32 %b)
  31   %val = extractvalue {i32, i1} %t, 0
  32   %obit = extractvalue {i32, i1} %t, 1
  33   br i1 %obit, label %call, label %exit
  34
  35 call:
  36   %res = tail call i32 @foo(i32 0, i32 %a, i32 %b)
  37   ret i32 %res
  38
  39 exit:
  40   ret i32 %val
  41 }
  42
  43 ; ... and the same with the inverted direction.
  44 define i32 @f3(i32 %dummy, i32 %a, i32 %b) {
  45 ; CHECK-LABEL: f3:
  46 ; CHECK: alrk %r2, %r3, %r4
  47 ; CHECK-NEXT: bnler %r14
  48 ; CHECK: lhi %r2, 0
  49 ; CHECK: jg foo@PLT
  50   %t = call {i32, i1} @llvm.uadd.with.overflow.i32(i32 %a, i32 %b)
  51   %val = extractvalue {i32, i1} %t, 0
  52   %obit = extractvalue {i32, i1} %t, 1
  53   br i1 %obit, label %exit, label %call
  54
  55 call:
  56   %res = tail call i32 @foo(i32 0, i32 %a, i32 %b)
  57   ret i32 %res
  58
  59 exit:
  60   ret i32 %val
  61 }
  62
  63 ; Check that we can still use ALR in obvious cases.
  64 define i32 @f4(i32 %a, i32 %b, i32 *%flag) {
  65 ; CHECK-LABEL: f4:
  66 ; CHECK: alr %r2, %r3
  67 ; CHECK: ipm [[REG1:%r[0-5]]]
  68 ; CHECK: risblg [[REG2:%r[0-5]]], [[REG1]], 31, 159, 35
  69 ; CHECK: st [[REG2]], 0(%r4)
  70 ; CHECK: br %r14
  71   %t = call {i32, i1} @llvm.uadd.with.overflow.i32(i32 %a, i32 %b)
  72   %val = extractvalue {i32, i1} %t, 0
  73   %obit = extractvalue {i32, i1} %t, 1
  74   %ext = zext i1 %obit to i32
  75   store i32 %ext, i32 *%flag
  76   ret i32 %val
  77 }
  78
  79 declare {i32, i1} @llvm.uadd.with.overflow.i32(i32, i32) nounwind readnone
  80