test/Transforms/NaryReassociate/NVPTX/nary-gep.ll

   1 ; RUN: opt < %s -nary-reassociate -early-cse -S | FileCheck %s
   2
   3 target datalayout = "e-i64:64-v16:16-v32:32-n16:32:64"
   4 target triple = "nvptx64-unknown-unknown"
   5
   6 declare void @foo(float*)
   7
   8 ; foo(&a[i]);
   9 ; foo(&a[i + j]);
  10 ;   =>
  11 ; t = &a[i];
  12 ; foo(t);
  13 ; foo(t + j);
  14 define void @reassociate_gep(float* %a, i64 %i, i64 %j) {
  15 ; CHECK-LABEL: @reassociate_gep(
  16   %1 = add i64 %i, %j
  17   %2 = getelementptr float, float* %a, i64 %i
  18 ; CHECK: [[t1:[^ ]+]] = getelementptr float, float* %a, i64 %i
  19   call void @foo(float* %2)
  20 ; CHECK: call void @foo(float* [[t1]])
  21   %3 = getelementptr float, float* %a, i64 %1
  22 ; CHECK: [[t2:[^ ]+]] = getelementptr float, float* [[t1]], i64 %j
  23   call void @foo(float* %3)
  24 ; CHECK: call void @foo(float* [[t2]])
  25   ret void
  26 }
  27
  28 ; foo(&a[sext(j)]);
  29 ; foo(&a[sext(i +nsw j)]);
  30 ; foo(&a[sext((i +nsw j) +nsw i)]);
  31 ;   =>
  32 ; t1 = &a[sext(j)];
  33 ; foo(t1);
  34 ; t2 = t1 + sext(i);
  35 ; foo(t2);
  36 ; t3 = t2 + sext(i); // sext(i) should be GVN'ed.
  37 ; foo(t3);
  38 define void @reassociate_gep_nsw(float* %a, i32 %i, i32 %j) {
  39 ; CHECK-LABEL: @reassociate_gep_nsw(
  40   %idxprom.j = sext i32 %j to i64
  41   %1 = getelementptr float, float* %a, i64 %idxprom.j
  42 ; CHECK: [[t1:[^ ]+]] = getelementptr float, float* %a, i64 %idxprom.j
  43   call void @foo(float* %1)
  44 ; CHECK: call void @foo(float* [[t1]])
  45
  46   %2 = add nsw i32 %i, %j
  47   %idxprom.2 = sext i32 %2 to i64
  48   %3 = getelementptr float, float* %a, i64 %idxprom.2
  49 ; CHECK: [[sexti:[^ ]+]] = sext i32 %i to i64
  50 ; CHECK: [[t2:[^ ]+]] = getelementptr float, float* [[t1]], i64 [[sexti]]
  51   call void @foo(float* %3)
  52 ; CHECK: call void @foo(float* [[t2]])
  53
  54   %4 = add nsw i32 %2, %i
  55   %idxprom.4 = sext i32 %4 to i64
  56   %5 = getelementptr float, float* %a, i64 %idxprom.4
  57 ; CHECK: [[t3:[^ ]+]] = getelementptr float, float* [[t2]], i64 [[sexti]]
  58   call void @foo(float* %5)
  59 ; CHECK: call void @foo(float* [[t3]])
  60
  61   ret void
  62 }
  63
  64 ; assume(j >= 0);
  65 ; foo(&a[zext(j)]);
  66 ; assume(i + j >= 0);
  67 ; foo(&a[zext(i + j)]);
  68 ;   =>
  69 ; t1 = &a[zext(j)];
  70 ; foo(t1);
  71 ; t2 = t1 + sext(i);
  72 ; foo(t2);
  73 define void @reassociate_gep_assume(float* %a, i32 %i, i32 %j) {
  74 ; CHECK-LABEL: @reassociate_gep_assume(
  75   ; assume(j >= 0)
  76   %cmp = icmp sgt i32 %j, -1
  77   call void @llvm.assume(i1 %cmp)
  78   %1 = add i32 %i, %j
  79   %cmp2 = icmp sgt i32 %1, -1
  80   call void @llvm.assume(i1 %cmp2)
  81
  82   %idxprom.j = zext i32 %j to i64
  83   %2 = getelementptr float, float* %a, i64 %idxprom.j
  84 ; CHECK: [[t1:[^ ]+]] = getelementptr float, float* %a, i64 %idxprom.j
  85   call void @foo(float* %2)
  86 ; CHECK: call void @foo(float* [[t1]])
  87
  88   %idxprom.1 = zext i32 %1 to i64
  89   %3 = getelementptr float, float* %a, i64 %idxprom.1
  90 ; CHECK: [[sexti:[^ ]+]] = sext i32 %i to i64
  91 ; CHECK: [[t2:[^ ]+]] = getelementptr float, float* [[t1]], i64 [[sexti]]
  92   call void @foo(float* %3)
  93 ; CHECK: call void @foo(float* [[t2]])
  94
  95   ret void
  96 }
  97
  98 ; Do not split the second GEP because sext(i + j) != sext(i) + sext(j).
  99 define void @reassociate_gep_no_nsw(float* %a, i32 %i, i32 %j) {
 100 ; CHECK-LABEL: @reassociate_gep_no_nsw(
 101   %1 = add i32 %i, %j
 102   %2 = getelementptr float, float* %a, i32 %j
 103 ; CHECK: getelementptr float, float* %a, i32 %j
 104   call void @foo(float* %2)
 105   %3 = getelementptr float, float* %a, i32 %1
 106 ; CHECK: getelementptr float, float* %a, i32 %1
 107   call void @foo(float* %3)
 108   ret void
 109 }
 110
 111 define void @reassociate_gep_128(float* %a, i128 %i, i128 %j) {
 112 ; CHECK-LABEL: @reassociate_gep_128(
 113   %1 = add i128 %i, %j
 114   %2 = getelementptr float, float* %a, i128 %i
 115 ; CHECK: [[t1:[^ ]+]] = getelementptr float, float* %a, i128 %i
 116   call void @foo(float* %2)
 117 ; CHECK: call void @foo(float* [[t1]])
 118   %3 = getelementptr float, float* %a, i128 %1
 119 ; CHECK: [[truncj:[^ ]+]] = trunc i128 %j to i64
 120 ; CHECK: [[t2:[^ ]+]] = getelementptr float, float* [[t1]], i64 [[truncj]]
 121   call void @foo(float* %3)
 122 ; CHECK: call void @foo(float* [[t2]])
 123   ret void
 124 }
 125
 126 %struct.complex = type { float, float }
 127
 128 declare void @bar(%struct.complex*)
 129
 130 define void @different_types(%struct.complex* %input, i64 %i) {
 131 ; CHECK-LABEL: @different_types(
 132   %t1 = getelementptr %struct.complex, %struct.complex* %input, i64 %i
 133   call void @bar(%struct.complex* %t1)
 134   %j = add i64 %i, 5
 135   %t2 = getelementptr %struct.complex, %struct.complex* %input, i64 %j, i32 0
 136 ; CHECK: [[cast:[^ ]+]] = bitcast %struct.complex* %t1 to float*
 137 ; CHECK-NEXT: %t2 = getelementptr float, float* [[cast]], i64 10
 138 ; CHECK-NEXT: call void @foo(float* %t2)
 139   call void @foo(float* %t2)
 140   ret void
 141 }
 142
 143 declare void @llvm.assume(i1)