llvm/test/CodeGen/SystemZ/int-mul-10.ll

   1 ; Test signed high-part i64->i128 multiplications on z14.
   2 ;
   3 ; RUN: llc < %s -mtriple=s390x-linux-gnu -mcpu=z14 -asm-verbose=0 | FileCheck %s
   4
   5 declare i64 @foo()
   6
   7 ; Check sign-extended multiplication in which only the high part is used.
   8 define i64 @f1(i64 %dummy, i64 %a, i64 %b) {
   9 ; CHECK-LABEL: f1:
  10 ; CHECK-NOT: {{%r[234]}}
  11 ; CHECK: mgrk %r2, %r3, %r4
  12 ; CHECK: br %r14
  13   %ax = sext i64 %a to i128
  14   %bx = sext i64 %b to i128
  15   %mulx = mul i128 %ax, %bx
  16   %highx = lshr i128 %mulx, 64
  17   %high = trunc i128 %highx to i64
  18   ret i64 %high
  19 }
  20
  21 ; Check sign-extended multiplication in which only part of the high half
  22 ; is used.
  23 define i64 @f2(i64 %dummy, i64 %a, i64 %b) {
  24 ; CHECK-LABEL: f2:
  25 ; CHECK-NOT: {{%r[234]}}
  26 ; CHECK: mgrk [[REG:%r[0-9]+]], %r3, %r4
  27 ; CHECK: srlg %r2, [[REG]], 3
  28 ; CHECK: br %r14
  29   %ax = sext i64 %a to i128
  30   %bx = sext i64 %b to i128
  31   %mulx = mul i128 %ax, %bx
  32   %highx = lshr i128 %mulx, 67
  33   %high = trunc i128 %highx to i64
  34   ret i64 %high
  35 }
  36
  37 ; Check sign-extended multiplication in which the result is split into
  38 ; high and low halves.
  39 define i64 @f3(i64 %dummy, i64 %a, i64 %b) {
  40 ; CHECK-LABEL: f3:
  41 ; CHECK-NOT: {{%r[234]}}
  42 ; CHECK: mgrk %r2, %r3, %r4
  43 ; CHECK: ogr %r2, %r3
  44 ; CHECK: br %r14
  45   %ax = sext i64 %a to i128
  46   %bx = sext i64 %b to i128
  47   %mulx = mul i128 %ax, %bx
  48   %highx = lshr i128 %mulx, 64
  49   %high = trunc i128 %highx to i64
  50   %low = trunc i128 %mulx to i64
  51   %or = or i64 %high, %low
  52   ret i64 %or
  53 }
  54
  55 ; Check MG with no displacement.
  56 define i64 @f4(i64 %dummy, i64 %a, i64 *%src) {
  57 ; CHECK-LABEL: f4:
  58 ; CHECK-NOT: {{%r[234]}}
  59 ; CHECK: mg %r2, 0(%r4)
  60 ; CHECK: br %r14
  61   %b = load i64, i64 *%src
  62   %ax = sext i64 %a to i128
  63   %bx = sext i64 %b to i128
  64   %mulx = mul i128 %ax, %bx
  65   %highx = lshr i128 %mulx, 64
  66   %high = trunc i128 %highx to i64
  67   ret i64 %high
  68 }
  69
  70 ; Check the high end of the aligned MG range.
  71 define i64 @f5(i64 %dummy, i64 %a, i64 *%src) {
  72 ; CHECK-LABEL: f5:
  73 ; CHECK: mg %r2, 524280(%r4)
  74 ; CHECK: br %r14
  75   %ptr = getelementptr i64, i64 *%src, i64 65535
  76   %b = load i64, i64 *%ptr
  77   %ax = sext i64 %a to i128
  78   %bx = sext i64 %b to i128
  79   %mulx = mul i128 %ax, %bx
  80   %highx = lshr i128 %mulx, 64
  81   %high = trunc i128 %highx to i64
  82   ret i64 %high
  83 }
  84
  85 ; Check the next doubleword up, which requires separate address logic.
  86 ; Other sequences besides this one would be OK.
  87 define i64 @f6(i64 %dummy, i64 %a, i64 *%src) {
  88 ; CHECK-LABEL: f6:
  89 ; CHECK: agfi %r4, 524288
  90 ; CHECK: mg %r2, 0(%r4)
  91 ; CHECK: br %r14
  92   %ptr = getelementptr i64, i64 *%src, i64 65536
  93   %b = load i64, i64 *%ptr
  94   %ax = sext i64 %a to i128
  95   %bx = sext i64 %b to i128
  96   %mulx = mul i128 %ax, %bx
  97   %highx = lshr i128 %mulx, 64
  98   %high = trunc i128 %highx to i64
  99   ret i64 %high
 100 }
 101
 102 ; Check the high end of the negative aligned MG range.
 103 define i64 @f7(i64 %dummy, i64 %a, i64 *%src) {
 104 ; CHECK-LABEL: f7:
 105 ; CHECK: mg %r2, -8(%r4)
 106 ; CHECK: br %r14
 107   %ptr = getelementptr i64, i64 *%src, i64 -1
 108   %b = load i64, i64 *%ptr
 109   %ax = sext i64 %a to i128
 110   %bx = sext i64 %b to i128
 111   %mulx = mul i128 %ax, %bx
 112   %highx = lshr i128 %mulx, 64
 113   %high = trunc i128 %highx to i64
 114   ret i64 %high
 115 }
 116
 117 ; Check the low end of the MG range.
 118 define i64 @f8(i64 %dummy, i64 %a, i64 *%src) {
 119 ; CHECK-LABEL: f8:
 120 ; CHECK: mg %r2, -524288(%r4)
 121 ; CHECK: br %r14
 122   %ptr = getelementptr i64, i64 *%src, i64 -65536
 123   %b = load i64, i64 *%ptr
 124   %ax = sext i64 %a to i128
 125   %bx = sext i64 %b to i128
 126   %mulx = mul i128 %ax, %bx
 127   %highx = lshr i128 %mulx, 64
 128   %high = trunc i128 %highx to i64
 129   ret i64 %high
 130 }
 131
 132 ; Check the next doubleword down, which needs separate address logic.
 133 ; Other sequences besides this one would be OK.
 134 define i64 @f9(i64 *%dest, i64 %a, i64 *%src) {
 135 ; CHECK-LABEL: f9:
 136 ; CHECK: agfi %r4, -524296
 137 ; CHECK: mg %r2, 0(%r4)
 138 ; CHECK: br %r14
 139   %ptr = getelementptr i64, i64 *%src, i64 -65537
 140   %b = load i64, i64 *%ptr
 141   %ax = sext i64 %a to i128
 142   %bx = sext i64 %b to i128
 143   %mulx = mul i128 %ax, %bx
 144   %highx = lshr i128 %mulx, 64
 145   %high = trunc i128 %highx to i64
 146   ret i64 %high
 147 }
 148
 149 ; Check that MG allows an index.
 150 define i64 @f10(i64 *%dest, i64 %a, i64 %src, i64 %index) {
 151 ; CHECK-LABEL: f10:
 152 ; CHECK: mg %r2, 524287(%r5,%r4)
 153 ; CHECK: br %r14
 154   %add1 = add i64 %src, %index
 155   %add2 = add i64 %add1, 524287
 156   %ptr = inttoptr i64 %add2 to i64 *
 157   %b = load i64, i64 *%ptr
 158   %ax = sext i64 %a to i128
 159   %bx = sext i64 %b to i128
 160   %mulx = mul i128 %ax, %bx
 161   %highx = lshr i128 %mulx, 64
 162   %high = trunc i128 %highx to i64
 163   ret i64 %high
 164 }
 165