test/Transforms/RewriteStatepointsForGC/live-vector.ll

   1 ; Test that we can correctly handle vectors of pointers in statepoint
   2 ; rewriting.  Currently, we scalarize, but that's an implementation detail.
   3 ; RUN: opt %s -rewrite-statepoints-for-gc -S | FileCheck  %s
   4
   5 ; A non-vector relocation for comparison
   6 define i64 addrspace(1)* @test(i64 addrspace(1)* %obj) gc "statepoint-example" {
   7 ; CHECK-LABEL: test
   8 ; CHECK: gc.statepoint
   9 ; CHECK-NEXT: gc.relocate
  10 ; CHECK-NEXT: bitcast
  11 ; CHECK-NEXT: ret i64 addrspace(1)* %obj.relocated.casted
  12 entry:
  13   %safepoint_token = call token (i64, i32, void ()*, i32, i32, ...) @llvm.experimental.gc.statepoint.p0f_isVoidf(i64 0, i32 0, void ()* @do_safepoint, i32 0, i32 0, i32 0, i32 0)
  14   ret i64 addrspace(1)* %obj
  15 }
  16
  17 ; A base vector from a argument
  18 define <2 x i64 addrspace(1)*> @test2(<2 x i64 addrspace(1)*> %obj) gc "statepoint-example" {
  19 ; CHECK-LABEL: test2
  20 ; CHECK: extractelement
  21 ; CHECK-NEXT: extractelement
  22 ; CHECK-NEXT: gc.statepoint
  23 ; CHECK-NEXT: gc.relocate
  24 ; CHECK-NEXT: bitcast
  25 ; CHECK-NEXT: gc.relocate
  26 ; CHECK-NEXT: bitcast
  27 ; CHECK-NEXT: insertelement
  28 ; CHECK-NEXT: insertelement
  29 ; CHECK-NEXT: ret <2 x i64 addrspace(1)*> %7
  30 entry:
  31   %safepoint_token = call token (i64, i32, void ()*, i32, i32, ...) @llvm.experimental.gc.statepoint.p0f_isVoidf(i64 0, i32 0, void ()* @do_safepoint, i32 0, i32 0, i32 0, i32 0)
  32   ret <2 x i64 addrspace(1)*> %obj
  33 }
  34
  35 ; A base vector from a load
  36 define <2 x i64 addrspace(1)*> @test3(<2 x i64 addrspace(1)*>* %ptr) gc "statepoint-example" {
  37 ; CHECK-LABEL: test3
  38 ; CHECK: load
  39 ; CHECK-NEXT: extractelement
  40 ; CHECK-NEXT: extractelement
  41 ; CHECK-NEXT: gc.statepoint
  42 ; CHECK-NEXT: gc.relocate
  43 ; CHECK-NEXT: bitcast
  44 ; CHECK-NEXT: gc.relocate
  45 ; CHECK-NEXT: bitcast
  46 ; CHECK-NEXT: insertelement
  47 ; CHECK-NEXT: insertelement
  48 ; CHECK-NEXT: ret <2 x i64 addrspace(1)*> %7
  49 entry:
  50   %obj = load <2 x i64 addrspace(1)*>, <2 x i64 addrspace(1)*>* %ptr
  51   %safepoint_token = call token (i64, i32, void ()*, i32, i32, ...) @llvm.experimental.gc.statepoint.p0f_isVoidf(i64 0, i32 0, void ()* @do_safepoint, i32 0, i32 0, i32 0, i32 0)
  52   ret <2 x i64 addrspace(1)*> %obj
  53 }
  54
  55 declare i32 @fake_personality_function()
  56
  57 ; When a statepoint is an invoke rather than a call
  58 define <2 x i64 addrspace(1)*> @test4(<2 x i64 addrspace(1)*>* %ptr) gc "statepoint-example" personality i32 ()* @fake_personality_function {
  59 ; CHECK-LABEL: test4
  60 ; CHECK: load
  61 ; CHECK-NEXT: extractelement
  62 ; CHECK-NEXT: extractelement
  63 ; CHECK-NEXT: gc.statepoint
  64 entry:
  65   %obj = load <2 x i64 addrspace(1)*>, <2 x i64 addrspace(1)*>* %ptr
  66   invoke token (i64, i32, void ()*, i32, i32, ...) @llvm.experimental.gc.statepoint.p0f_isVoidf(i64 0, i32 0, void ()* @do_safepoint, i32 0, i32 0, i32 0, i32 0)
  67           to label %normal_return unwind label %exceptional_return
  68
  69 ; CHECK-LABEL: normal_return:
  70 ; CHECK: gc.relocate
  71 ; CHECK-NEXT: bitcast
  72 ; CHECK-NEXT: gc.relocate
  73 ; CHECK-NEXT: bitcast
  74 ; CHECK-NEXT: insertelement
  75 ; CHECK-NEXT: insertelement
  76 ; CHECK-NEXT: ret <2 x i64 addrspace(1)*> %8
  77 normal_return:                                    ; preds = %entry
  78   ret <2 x i64 addrspace(1)*> %obj
  79
  80 ; CHECK-LABEL: exceptional_return:
  81 ; CHECK: gc.relocate
  82 ; CHECK-NEXT: bitcast
  83 ; CHECK-NEXT: gc.relocate
  84 ; CHECK-NEXT: bitcast
  85 ; CHECK-NEXT: insertelement
  86 ; CHECK-NEXT: insertelement
  87 ; CHECK-NEXT: ret <2 x i64 addrspace(1)*> %14
  88 exceptional_return:                               ; preds = %entry
  89   %landing_pad4 = landingpad token
  90           cleanup
  91   ret <2 x i64 addrspace(1)*> %obj
  92 }
  93
  94 ; Can we handle an insert element with a constant offset?  This effectively
  95 ; tests both the equal and inequal case since we have to relocate both indices
  96 ; in the vector.
  97 define <2 x i64 addrspace(1)*> @test5(i64 addrspace(1)* %p)
  98      gc "statepoint-example" {
  99 ; CHECK-LABEL: test5
 100 ; CHECK: insertelement
 101 ; CHECK-NEXT: extractelement
 102 ; CHECK-NEXT: extractelement
 103 ; CHECK-NEXT: gc.statepoint
 104 ; CHECK-NEXT: gc.relocate
 105 ; CHECK-NEXT: bitcast
 106 ; CHECK-NEXT: gc.relocate
 107 ; CHECK-NEXT: bitcast
 108 ; CHECK-NEXT: insertelement
 109 ; CHECK-NEXT: insertelement
 110 ; CHECK-NEXT: ret <2 x i64 addrspace(1)*> %7
 111 entry:
 112   %vec = insertelement <2 x i64 addrspace(1)*> undef, i64 addrspace(1)* %p, i32 0
 113   %safepoint_token = call token (i64, i32, void ()*, i32, i32, ...) @llvm.experimental.gc.statepoint.p0f_isVoidf(i64 0, i32 0, void ()* @do_safepoint, i32 0, i32 0, i32 0, i32 0)
 114   ret <2 x i64 addrspace(1)*> %vec
 115 }
 116
 117
 118 ; A base vector from a load
 119 define <2 x i64 addrspace(1)*> @test6(i1 %cnd, <2 x i64 addrspace(1)*>* %ptr)
 120     gc "statepoint-example" {
 121 ; CHECK-LABEL: test6
 122 ; CHECK-LABEL: merge:
 123 ; CHECK-NEXT: = phi
 124 ; CHECK-NEXT: extractelement
 125 ; CHECK-NEXT: extractelement
 126 ; CHECK-NEXT: gc.statepoint
 127 ; CHECK-NEXT: gc.relocate
 128 ; CHECK-NEXT: bitcast
 129 ; CHECK-NEXT: gc.relocate
 130 ; CHECK-NEXT: bitcast
 131 ; CHECK-NEXT: insertelement
 132 ; CHECK-NEXT: insertelement
 133 ; CHECK-NEXT: ret <2 x i64 addrspace(1)*>
 134 entry:
 135   br i1 %cnd, label %taken, label %untaken
 136 taken:
 137   %obja = load <2 x i64 addrspace(1)*>, <2 x i64 addrspace(1)*>* %ptr
 138   br label %merge
 139 untaken:
 140   %objb = load <2 x i64 addrspace(1)*>, <2 x i64 addrspace(1)*>* %ptr
 141   br label %merge
 142
 143 merge:
 144   %obj = phi <2 x i64 addrspace(1)*> [%obja, %taken], [%objb, %untaken]
 145   %safepoint_token = call token (i64, i32, void ()*, i32, i32, ...) @llvm.experimental.gc.statepoint.p0f_isVoidf(i64 0, i32 0, void ()* @do_safepoint, i32 0, i32 0, i32 0, i32 0)
 146   ret <2 x i64 addrspace(1)*> %obj
 147 }
 148
 149
 150 declare void @do_safepoint()
 151
 152 declare token @llvm.experimental.gc.statepoint.p0f_isVoidf(i64, i32, void ()*, i32, i32, ...)