xref: /llvm-project/llvm/test/Transforms/RewriteStatepointsForGC/base-inference.ll (revision f01a3a893c147c1594b9a3fbd817456b209dabbf)
182400ae0SPhilip Reames; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2dcd5f6f2SRoman Lebedev; RUN: opt < %s -passes=rewrite-statepoints-for-gc -S 2>&1 | FileCheck %s
382400ae0SPhilip Reames
482400ae0SPhilip Reames; A collection of tests for exercising the base inference logic in the
582400ae0SPhilip Reames; findBasePointers.  That is, the logic which proves a potentially derived
682400ae0SPhilip Reames; pointer is actually a base pointer itself.
782400ae0SPhilip Reames
8*f01a3a89SNikita Popovdefine ptr addrspace(1) @test(ptr addrspace(1) %a) gc "statepoint-example" {
982400ae0SPhilip Reames; CHECK-LABEL: @test(
10*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[A:%.*]]) ]
11*f01a3a89SNikita Popov; CHECK-NEXT:    [[A_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
12*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[A_RELOCATED]]
1382400ae0SPhilip Reames;
1482400ae0SPhilip Reames  call void @foo()
15*f01a3a89SNikita Popov  ret ptr addrspace(1) %a
1682400ae0SPhilip Reames}
1782400ae0SPhilip Reames
18*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_select(i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2) gc "statepoint-example" {
1982400ae0SPhilip Reames; CHECK-LABEL: @test_select(
20*f01a3a89SNikita Popov; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[C:%.*]], ptr addrspace(1) [[A1:%.*]], ptr addrspace(1) [[A2:%.*]]
21*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[SEL]]) ]
22*f01a3a89SNikita Popov; CHECK-NEXT:    [[SEL_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
23*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[SEL_RELOCATED]]
2482400ae0SPhilip Reames;
25*f01a3a89SNikita Popov  %sel = select i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2
2682400ae0SPhilip Reames  call void @foo()
27*f01a3a89SNikita Popov  ret ptr addrspace(1) %sel
2882400ae0SPhilip Reames}
2982400ae0SPhilip Reames
30*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_phi1(i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2) gc "statepoint-example" {
3182400ae0SPhilip Reames; CHECK-LABEL: @test_phi1(
3282400ae0SPhilip Reames; CHECK-NEXT:  entry:
3382400ae0SPhilip Reames; CHECK-NEXT:    br i1 [[C:%.*]], label [[TAKEN:%.*]], label [[UNTAKEN:%.*]]
3482400ae0SPhilip Reames; CHECK:       taken:
3582400ae0SPhilip Reames; CHECK-NEXT:    br label [[MERGE:%.*]]
3682400ae0SPhilip Reames; CHECK:       untaken:
3782400ae0SPhilip Reames; CHECK-NEXT:    br label [[MERGE]]
3882400ae0SPhilip Reames; CHECK:       merge:
39*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI:%.*]] = phi ptr addrspace(1) [ [[A1:%.*]], [[TAKEN]] ], [ [[A2:%.*]], [[UNTAKEN]] ]
40*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[PHI]]) ]
41*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
42*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[PHI_RELOCATED]]
4382400ae0SPhilip Reames;
4482400ae0SPhilip Reamesentry:
4582400ae0SPhilip Reames  br i1 %c, label %taken, label %untaken
4682400ae0SPhilip Reamestaken:
4782400ae0SPhilip Reames  br label %merge
4882400ae0SPhilip Reamesuntaken:
4982400ae0SPhilip Reames  br label %merge
5082400ae0SPhilip Reamesmerge:
51*f01a3a89SNikita Popov  %phi = phi ptr addrspace(1) [%a1, %taken], [%a2, %untaken]
5282400ae0SPhilip Reames  call void @foo()
53*f01a3a89SNikita Popov  ret ptr addrspace(1) %phi
5482400ae0SPhilip Reames}
5582400ae0SPhilip Reames
56*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_phi_lcssa(i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2) gc "statepoint-example" {
5782400ae0SPhilip Reames; CHECK-LABEL: @test_phi_lcssa(
5882400ae0SPhilip Reames; CHECK-NEXT:  entry:
5982400ae0SPhilip Reames; CHECK-NEXT:    br label [[MERGE:%.*]]
6082400ae0SPhilip Reames; CHECK:       merge:
61*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[A1:%.*]]) ]
62*f01a3a89SNikita Popov; CHECK-NEXT:    [[A1_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
63*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[A1_RELOCATED]]
6482400ae0SPhilip Reames;
6582400ae0SPhilip Reamesentry:
6682400ae0SPhilip Reames  br label %merge
6782400ae0SPhilip Reamesmerge:
68*f01a3a89SNikita Popov  %phi = phi ptr addrspace(1) [%a1, %entry]
6982400ae0SPhilip Reames  call void @foo()
70*f01a3a89SNikita Popov  ret ptr addrspace(1) %phi
7182400ae0SPhilip Reames}
7282400ae0SPhilip Reames
7382400ae0SPhilip Reames
74*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_loop1(i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2) gc "statepoint-example" {
7582400ae0SPhilip Reames; CHECK-LABEL: @test_loop1(
7682400ae0SPhilip Reames; CHECK-NEXT:  entry:
7782400ae0SPhilip Reames; CHECK-NEXT:    br label [[LOOP:%.*]]
7882400ae0SPhilip Reames; CHECK:       loop:
79*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI:%.*]] = phi ptr addrspace(1) [ [[A1:%.*]], [[ENTRY:%.*]] ], [ [[A2:%.*]], [[LOOP]] ]
8082400ae0SPhilip Reames; CHECK-NEXT:    br i1 [[C:%.*]], label [[EXIT:%.*]], label [[LOOP]]
8182400ae0SPhilip Reames; CHECK:       exit:
82*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[PHI]]) ]
83*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
84*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[PHI_RELOCATED]]
8582400ae0SPhilip Reames;
8682400ae0SPhilip Reamesentry:
8782400ae0SPhilip Reames  br label %loop
8882400ae0SPhilip Reamesloop:
89*f01a3a89SNikita Popov  %phi = phi ptr addrspace(1) [%a1, %entry], [%a2, %loop]
9082400ae0SPhilip Reames  br i1 %c, label %exit, label %loop
9182400ae0SPhilip Reamesexit:
92*f01a3a89SNikita Popov  %phi2 = phi ptr addrspace(1) [%phi, %loop]
9382400ae0SPhilip Reames  call void @foo()
94*f01a3a89SNikita Popov  ret ptr addrspace(1) %phi2
9582400ae0SPhilip Reames}
9682400ae0SPhilip Reames
97*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_loop2(i1 %c, ptr addrspace(1) %a1) gc "statepoint-example" {
9882400ae0SPhilip Reames; CHECK-LABEL: @test_loop2(
9982400ae0SPhilip Reames; CHECK-NEXT:  entry:
10082400ae0SPhilip Reames; CHECK-NEXT:    br label [[LOOP:%.*]]
10182400ae0SPhilip Reames; CHECK:       loop:
102*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI:%.*]] = phi ptr addrspace(1) [ [[A1:%.*]], [[ENTRY:%.*]] ], [ [[O2:%.*]], [[LOOP]] ]
103*f01a3a89SNikita Popov; CHECK-NEXT:    [[O2]] = load ptr addrspace(1), ptr addrspace(1) [[PHI]], align 8
10482400ae0SPhilip Reames; CHECK-NEXT:    br i1 [[C:%.*]], label [[EXIT:%.*]], label [[LOOP]]
10582400ae0SPhilip Reames; CHECK:       exit:
106*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[PHI]]) ]
107*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
108*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[PHI_RELOCATED]]
10982400ae0SPhilip Reames;
11082400ae0SPhilip Reamesentry:
11182400ae0SPhilip Reames  br label %loop
11282400ae0SPhilip Reamesloop:
113*f01a3a89SNikita Popov  %phi = phi ptr addrspace(1) [%a1, %entry], [%o2, %loop]
114*f01a3a89SNikita Popov  %o2 = load ptr addrspace(1), ptr addrspace(1) %phi
11582400ae0SPhilip Reames  br i1 %c, label %exit, label %loop
11682400ae0SPhilip Reamesexit:
117*f01a3a89SNikita Popov  %phi2 = phi ptr addrspace(1) [%phi, %loop]
11882400ae0SPhilip Reames  call void @foo()
119*f01a3a89SNikita Popov  ret ptr addrspace(1) %phi2
12082400ae0SPhilip Reames}
12182400ae0SPhilip Reames
12282400ae0SPhilip Reames; %phi1 and phi2 are not base pointers, but they do have a single
12382400ae0SPhilip Reames; base pointer which is %a1
124*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_loop3(i1 %c, ptr addrspace(1) %a1) gc "statepoint-example" {
12582400ae0SPhilip Reames; CHECK-LABEL: @test_loop3(
12682400ae0SPhilip Reames; CHECK-NEXT:  entry:
12782400ae0SPhilip Reames; CHECK-NEXT:    br label [[LOOP:%.*]]
12882400ae0SPhilip Reames; CHECK:       loop:
129*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI:%.*]] = phi ptr addrspace(1) [ [[A1:%.*]], [[ENTRY:%.*]] ], [ [[GEP:%.*]], [[LOOP]] ]
130*f01a3a89SNikita Popov; CHECK-NEXT:    [[GEP]] = getelementptr i8, ptr addrspace(1) [[PHI]], i64 16
13182400ae0SPhilip Reames; CHECK-NEXT:    br i1 [[C:%.*]], label [[EXIT:%.*]], label [[LOOP]]
13282400ae0SPhilip Reames; CHECK:       exit:
133*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[PHI]], ptr addrspace(1) [[A1]]) ]
134*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 1, i32 0)
135*f01a3a89SNikita Popov; CHECK-NEXT:    [[A1_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 1, i32 1)
136*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[PHI_RELOCATED]]
13782400ae0SPhilip Reames;
13882400ae0SPhilip Reamesentry:
13982400ae0SPhilip Reames  br label %loop
14082400ae0SPhilip Reamesloop:
141*f01a3a89SNikita Popov  %phi = phi ptr addrspace(1) [%a1, %entry], [%gep, %loop]
142*f01a3a89SNikita Popov  %gep = getelementptr i8, ptr addrspace(1) %phi, i64 16
14382400ae0SPhilip Reames  br i1 %c, label %exit, label %loop
14482400ae0SPhilip Reamesexit:
145*f01a3a89SNikita Popov  %phi2 = phi ptr addrspace(1) [%phi, %loop]
14682400ae0SPhilip Reames  call void @foo()
147*f01a3a89SNikita Popov  ret ptr addrspace(1) %phi2
14882400ae0SPhilip Reames}
14982400ae0SPhilip Reames
150*f01a3a89SNikita Popovdefine <2 x ptr addrspace(1)> @test_vec_passthrough(<2 x ptr addrspace(1)> %a) gc "statepoint-example" {
15182400ae0SPhilip Reames; CHECK-LABEL: @test_vec_passthrough(
152*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(<2 x ptr addrspace(1)> [[A:%.*]]) ]
153*f01a3a89SNikita Popov; CHECK-NEXT:    [[A_RELOCATED:%.*]] = call coldcc <2 x ptr addrspace(1)> @llvm.experimental.gc.relocate.v2p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
154*f01a3a89SNikita Popov; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[A_RELOCATED]]
15582400ae0SPhilip Reames;
15682400ae0SPhilip Reames  call void @foo()
157*f01a3a89SNikita Popov  ret <2 x ptr addrspace(1)> %a
15882400ae0SPhilip Reames}
15982400ae0SPhilip Reames
16082400ae0SPhilip Reames
161*f01a3a89SNikita Popovdefine <2 x ptr addrspace(1)> @test_insert(ptr addrspace(1) %a) gc "statepoint-example" {
16282400ae0SPhilip Reames; CHECK-LABEL: @test_insert(
163*f01a3a89SNikita Popov; CHECK-NEXT:    [[VEC:%.*]] = insertelement <2 x ptr addrspace(1)> zeroinitializer, ptr addrspace(1) [[A:%.*]], i64 0
164*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(<2 x ptr addrspace(1)> [[VEC]]) ]
165*f01a3a89SNikita Popov; CHECK-NEXT:    [[VEC_RELOCATED:%.*]] = call coldcc <2 x ptr addrspace(1)> @llvm.experimental.gc.relocate.v2p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
166*f01a3a89SNikita Popov; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[VEC_RELOCATED]]
16782400ae0SPhilip Reames;
168*f01a3a89SNikita Popov  %vec = insertelement <2 x ptr addrspace(1)> zeroinitializer, ptr addrspace(1) %a, i64 0
16982400ae0SPhilip Reames  call void @foo()
170*f01a3a89SNikita Popov  ret <2 x ptr addrspace(1)> %vec
17182400ae0SPhilip Reames}
17282400ae0SPhilip Reames
173*f01a3a89SNikita Popovdefine ptr addrspace(1) @test_extract(<2 x ptr addrspace(1)> %a) gc "statepoint-example" {
17482400ae0SPhilip Reames; CHECK-LABEL: @test_extract(
175*f01a3a89SNikita Popov; CHECK-NEXT:    [[EE:%.*]] = extractelement <2 x ptr addrspace(1)> [[A:%.*]], i64 0
176*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[EE]]) ]
177*f01a3a89SNikita Popov; CHECK-NEXT:    [[EE_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
178*f01a3a89SNikita Popov; CHECK-NEXT:    ret ptr addrspace(1) [[EE_RELOCATED]]
17982400ae0SPhilip Reames;
180*f01a3a89SNikita Popov  %ee = extractelement <2 x ptr addrspace(1)> %a, i64 0
18182400ae0SPhilip Reames  call void @foo()
182*f01a3a89SNikita Popov  ret ptr addrspace(1) %ee
18382400ae0SPhilip Reames}
18482400ae0SPhilip Reames
185*f01a3a89SNikita Popovdefine <2 x ptr addrspace(1)> @test_shuffle(<2 x ptr addrspace(1)> %a1) gc "statepoint-example" {
18682400ae0SPhilip Reames; CHECK-LABEL: @test_shuffle(
187*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES:%.*]] = shufflevector <2 x ptr addrspace(1)> [[A1:%.*]], <2 x ptr addrspace(1)> [[A1]], <2 x i32> zeroinitializer
188*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(<2 x ptr addrspace(1)> [[RES]]) ]
189*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES_RELOCATED:%.*]] = call coldcc <2 x ptr addrspace(1)> @llvm.experimental.gc.relocate.v2p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
190*f01a3a89SNikita Popov; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[RES_RELOCATED]]
19182400ae0SPhilip Reames;
192*f01a3a89SNikita Popov  %res = shufflevector <2 x ptr addrspace(1)> %a1, <2 x ptr addrspace(1)> %a1, <2 x i32> zeroinitializer
19382400ae0SPhilip Reames  call void @foo()
194*f01a3a89SNikita Popov  ret <2 x ptr addrspace(1)> %res
19582400ae0SPhilip Reames}
19682400ae0SPhilip Reames
197*f01a3a89SNikita Popovdefine <2 x ptr addrspace(1)> @test_shuffle2(<2 x ptr addrspace(1)> %a1, <2 x ptr addrspace(1)> %a2) gc "statepoint-example" {
19882400ae0SPhilip Reames; CHECK-LABEL: @test_shuffle2(
199*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES:%.*]] = shufflevector <2 x ptr addrspace(1)> [[A1:%.*]], <2 x ptr addrspace(1)> [[A2:%.*]], <2 x i32> zeroinitializer
200*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(<2 x ptr addrspace(1)> [[RES]]) ]
201*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES_RELOCATED:%.*]] = call coldcc <2 x ptr addrspace(1)> @llvm.experimental.gc.relocate.v2p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
202*f01a3a89SNikita Popov; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[RES_RELOCATED]]
20382400ae0SPhilip Reames;
204*f01a3a89SNikita Popov  %res = shufflevector <2 x ptr addrspace(1)> %a1, <2 x ptr addrspace(1)> %a2, <2 x i32> zeroinitializer
20582400ae0SPhilip Reames  call void @foo()
206*f01a3a89SNikita Popov  ret <2 x ptr addrspace(1)> %res
20782400ae0SPhilip Reames}
20882400ae0SPhilip Reames
209*f01a3a89SNikita Popovdefine <4 x ptr addrspace(1)> @test_shuffle_concat(<2 x ptr addrspace(1)> %a1, <2 x ptr addrspace(1)> %a2) gc "statepoint-example" {
21082400ae0SPhilip Reames; CHECK-LABEL: @test_shuffle_concat(
211*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES:%.*]] = shufflevector <2 x ptr addrspace(1)> [[A1:%.*]], <2 x ptr addrspace(1)> [[A2:%.*]], <4 x i32> <i32 0, i32 1, i32 2, i32 3>
212*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(<4 x ptr addrspace(1)> [[RES]]) ]
213*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES_RELOCATED:%.*]] = call coldcc <4 x ptr addrspace(1)> @llvm.experimental.gc.relocate.v4p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
214*f01a3a89SNikita Popov; CHECK-NEXT:    ret <4 x ptr addrspace(1)> [[RES_RELOCATED]]
21582400ae0SPhilip Reames;
216*f01a3a89SNikita Popov  %res = shufflevector <2 x ptr addrspace(1)> %a1, <2 x ptr addrspace(1)> %a2, <4 x i32> <i32 0, i32 1, i32 2, i32 3>
21782400ae0SPhilip Reames  call void @foo()
218*f01a3a89SNikita Popov  ret <4 x ptr addrspace(1)> %res
21982400ae0SPhilip Reames}
22082400ae0SPhilip Reames
2215cabf472SPhilip Reames; TODO: Special case worth handling - we interpret the shuffle as if we need
2225cabf472SPhilip Reames; to select the base pointers from either input when the mask is known.
223*f01a3a89SNikita Popovdefine <2 x ptr addrspace(1)> @test_shuffle_broadcast(ptr addrspace(1) %a) gc "statepoint-example" {
2245cabf472SPhilip Reames; CHECK-LABEL: @test_shuffle_broadcast(
2255cabf472SPhilip Reames; CHECK-NEXT:  entry:
226*f01a3a89SNikita Popov; CHECK-NEXT:    [[IE:%.*]] = insertelement <2 x ptr addrspace(1)> zeroinitializer, ptr addrspace(1) [[A:%.*]], i64 0
227*f01a3a89SNikita Popov; CHECK-NEXT:    [[BROADCAST:%.*]] = shufflevector <2 x ptr addrspace(1)> [[IE]], <2 x ptr addrspace(1)> undef, <2 x i32> zeroinitializer
228*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(<2 x ptr addrspace(1)> [[BROADCAST]]) ]
229*f01a3a89SNikita Popov; CHECK-NEXT:    [[BROADCAST_RELOCATED:%.*]] = call coldcc <2 x ptr addrspace(1)> @llvm.experimental.gc.relocate.v2p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
230*f01a3a89SNikita Popov; CHECK-NEXT:    ret <2 x ptr addrspace(1)> [[BROADCAST_RELOCATED]]
2315cabf472SPhilip Reames;
2325cabf472SPhilip Reamesentry:
233*f01a3a89SNikita Popov  %ie = insertelement <2 x ptr addrspace(1)> zeroinitializer, ptr addrspace(1) %a, i64 0
234*f01a3a89SNikita Popov  %broadcast = shufflevector <2 x ptr addrspace(1)> %ie, <2 x ptr addrspace(1)> undef, <2 x i32> zeroinitializer
2355cabf472SPhilip Reames  call void @foo()
236*f01a3a89SNikita Popov  ret <2 x ptr addrspace(1)> %broadcast
2375cabf472SPhilip Reames}
23882400ae0SPhilip Reames
239b7fc3729SPhilip Reames; Show a case where only a portion of the sub-graph propagates base pointers.
240*f01a3a89SNikita Popovdefine i8 @test_subgraph(i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2) gc "statepoint-example" {
241b7fc3729SPhilip Reames; CHECK-LABEL: @test_subgraph(
242b7fc3729SPhilip Reames; CHECK-NEXT:  entry:
243*f01a3a89SNikita Popov; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[C:%.*]], ptr addrspace(1) [[A1:%.*]], ptr addrspace(1) [[A2:%.*]]
244b7fc3729SPhilip Reames; CHECK-NEXT:    br i1 [[C]], label [[TAKEN:%.*]], label [[MERGE:%.*]]
245b7fc3729SPhilip Reames; CHECK:       taken:
246*f01a3a89SNikita Popov; CHECK-NEXT:    [[GEP:%.*]] = getelementptr i8, ptr addrspace(1) [[SEL]], i64 8
247b7fc3729SPhilip Reames; CHECK-NEXT:    br label [[MERGE]]
248b7fc3729SPhilip Reames; CHECK:       merge:
249*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI:%.*]] = phi ptr addrspace(1) [ [[GEP]], [[TAKEN]] ], [ [[SEL]], [[ENTRY:%.*]] ]
250*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[PHI]], ptr addrspace(1) [[SEL]]) ]
251*f01a3a89SNikita Popov; CHECK-NEXT:    [[PHI_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 1, i32 0)
252*f01a3a89SNikita Popov; CHECK-NEXT:    [[SEL_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 1, i32 1)
253*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES:%.*]] = load i8, ptr addrspace(1) [[PHI_RELOCATED]], align 1
254b7fc3729SPhilip Reames; CHECK-NEXT:    ret i8 [[RES]]
255b7fc3729SPhilip Reames;
256b7fc3729SPhilip Reamesentry:
257*f01a3a89SNikita Popov  %sel = select i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2
258b7fc3729SPhilip Reames  br i1 %c, label %taken, label %merge
259b7fc3729SPhilip Reamestaken:
260*f01a3a89SNikita Popov  %gep = getelementptr i8, ptr addrspace(1) %sel, i64 8
261b7fc3729SPhilip Reames  br label %merge
262b7fc3729SPhilip Reamesmerge:
263*f01a3a89SNikita Popov  %phi = phi ptr addrspace(1) [%gep, %taken], [%sel, %entry]
264b7fc3729SPhilip Reames  call void @foo()
265*f01a3a89SNikita Popov  %res = load i8, ptr addrspace(1) %phi
266b7fc3729SPhilip Reames  ret i8 %res
267b7fc3729SPhilip Reames}
268b7fc3729SPhilip Reames
269b7fc3729SPhilip Reames; An example of a non-trivial subgraph computing base pointers.
270*f01a3a89SNikita Popovdefine i8 @test_subgraph2(i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2) gc "statepoint-example" {
271b7fc3729SPhilip Reames; CHECK-LABEL: @test_subgraph2(
272b7fc3729SPhilip Reames; CHECK-NEXT:  entry:
273*f01a3a89SNikita Popov; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[C:%.*]], ptr addrspace(1) [[A1:%.*]], ptr addrspace(1) [[A2:%.*]]
274*f01a3a89SNikita Popov; CHECK-NEXT:    [[IE:%.*]] = insertelement <2 x ptr addrspace(1)> zeroinitializer, ptr addrspace(1) [[SEL]], i64 0
275*f01a3a89SNikita Popov; CHECK-NEXT:    [[BROADCAST:%.*]] = shufflevector <2 x ptr addrspace(1)> [[IE]], <2 x ptr addrspace(1)> [[IE]], <2 x i32> zeroinitializer
276*f01a3a89SNikita Popov; CHECK-NEXT:    [[EE:%.*]] = extractelement <2 x ptr addrspace(1)> [[BROADCAST]], i32 1
277*f01a3a89SNikita Popov; CHECK-NEXT:    [[STATEPOINT_TOKEN:%.*]] = call token (i64, i32, ptr, i32, i32, ...) @llvm.experimental.gc.statepoint.p0(i64 2882400000, i32 0, ptr elementtype(void ()) @foo, i32 0, i32 0, i32 0, i32 0) [ "gc-live"(ptr addrspace(1) [[EE]]) ]
278*f01a3a89SNikita Popov; CHECK-NEXT:    [[EE_RELOCATED:%.*]] = call coldcc ptr addrspace(1) @llvm.experimental.gc.relocate.p1(token [[STATEPOINT_TOKEN]], i32 0, i32 0)
279*f01a3a89SNikita Popov; CHECK-NEXT:    [[RES:%.*]] = load i8, ptr addrspace(1) [[EE_RELOCATED]], align 1
280b7fc3729SPhilip Reames; CHECK-NEXT:    ret i8 [[RES]]
281b7fc3729SPhilip Reames;
282b7fc3729SPhilip Reamesentry:
283*f01a3a89SNikita Popov  %sel = select i1 %c, ptr addrspace(1) %a1, ptr addrspace(1) %a2
284*f01a3a89SNikita Popov  %ie = insertelement <2 x ptr addrspace(1)> zeroinitializer, ptr addrspace(1) %sel, i64 0
285*f01a3a89SNikita Popov  %broadcast = shufflevector <2 x ptr addrspace(1)> %ie, <2 x ptr addrspace(1)> %ie, <2 x i32> zeroinitializer
286*f01a3a89SNikita Popov  %ee = extractelement <2 x ptr addrspace(1)> %broadcast, i32 1
287b7fc3729SPhilip Reames  call void @foo()
288*f01a3a89SNikita Popov  %res = load i8, ptr addrspace(1) %ee
289b7fc3729SPhilip Reames  ret i8 %res
290b7fc3729SPhilip Reames}
29182400ae0SPhilip Reames
29282400ae0SPhilip Reames
29382400ae0SPhilip Reamesdeclare void @foo()
294