xref: /llvm-project/llvm/test/Transforms/IROutliner/outlining-gep.ll (revision f4b925ee7078f058602fd323e25f45f1ae91ca34)
1b6b1006fSAndrew Litteken; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
29dd9575cSRoman Lebedev; RUN: opt -S -passes=verify,iroutliner -ir-outlining-no-cost < %s | FileCheck %s
3b6b1006fSAndrew Litteken
4b6b1006fSAndrew Litteken; This test checks to make sure that we outline getelementptr instructions only
5b6b1006fSAndrew Litteken; when all the operands after the first are the exact same. In this case, we
6b6b1006fSAndrew Litteken; outline from the first two functions, but not the third.
7b6b1006fSAndrew Litteken
8b6b1006fSAndrew Litteken%struct.RT = type { i8, [10 x [20 x i32]], i8 }
9b6b1006fSAndrew Litteken%struct.ST = type { i32, double, %struct.RT }
10b6b1006fSAndrew Litteken
11*f4b925eeSMatt Arsenaultdefine void @function1(ptr %s, i64 %t) {
12b6b1006fSAndrew Litteken; CHECK-LABEL: @function1(
13b6b1006fSAndrew Litteken; CHECK-NEXT:  entry:
14b6b1006fSAndrew Litteken; CHECK-NEXT:    [[A:%.*]] = alloca i32, align 4
15b6b1006fSAndrew Litteken; CHECK-NEXT:    [[B:%.*]] = alloca i32, align 4
16*f4b925eeSMatt Arsenault; CHECK-NEXT:    call void @outlined_ir_func_0(ptr [[A]], ptr [[B]], ptr [[S:%.*]], i64 [[T:%.*]])
17b6b1006fSAndrew Litteken; CHECK-NEXT:    ret void
18b6b1006fSAndrew Litteken;
19b6b1006fSAndrew Littekenentry:
20b6b1006fSAndrew Litteken  %a = alloca i32, align 4
21b6b1006fSAndrew Litteken  %b = alloca i32, align 4
22*f4b925eeSMatt Arsenault  store i32 2, ptr %a, align 4
23*f4b925eeSMatt Arsenault  store i32 3, ptr %b, align 4
24*f4b925eeSMatt Arsenault  %0 = getelementptr inbounds %struct.ST, ptr %s, i64 %t, i32 1
25b6b1006fSAndrew Litteken  ret void
26b6b1006fSAndrew Litteken}
27b6b1006fSAndrew Litteken
28*f4b925eeSMatt Arsenaultdefine void @function2(ptr %s, i64 %t) {
29b6b1006fSAndrew Litteken; CHECK-LABEL: @function2(
30b6b1006fSAndrew Litteken; CHECK-NEXT:  entry:
31b6b1006fSAndrew Litteken; CHECK-NEXT:    [[A:%.*]] = alloca i32, align 4
32b6b1006fSAndrew Litteken; CHECK-NEXT:    [[B:%.*]] = alloca i32, align 4
33*f4b925eeSMatt Arsenault; CHECK-NEXT:    call void @outlined_ir_func_0(ptr [[A]], ptr [[B]], ptr [[S:%.*]], i64 [[T:%.*]])
34b6b1006fSAndrew Litteken; CHECK-NEXT:    ret void
35b6b1006fSAndrew Litteken;
36b6b1006fSAndrew Littekenentry:
37b6b1006fSAndrew Litteken  %a = alloca i32, align 4
38b6b1006fSAndrew Litteken  %b = alloca i32, align 4
39*f4b925eeSMatt Arsenault  store i32 2, ptr %a, align 4
40*f4b925eeSMatt Arsenault  store i32 3, ptr %b, align 4
41*f4b925eeSMatt Arsenault  %0 = getelementptr inbounds %struct.ST, ptr %s, i64 %t, i32 1
42b6b1006fSAndrew Litteken  ret void
43b6b1006fSAndrew Litteken}
44b6b1006fSAndrew Litteken
45*f4b925eeSMatt Arsenaultdefine void @function3(ptr %s, i64 %t) {
46b6b1006fSAndrew Litteken; CHECK-LABEL: @function3(
47b6b1006fSAndrew Litteken; CHECK-NEXT:  entry:
48b6b1006fSAndrew Litteken; CHECK-NEXT:    [[A:%.*]] = alloca i32, align 4
49b6b1006fSAndrew Litteken; CHECK-NEXT:    [[B:%.*]] = alloca i32, align 4
50*f4b925eeSMatt Arsenault; CHECK-NEXT:    store i32 2, ptr [[A]], align 4
51*f4b925eeSMatt Arsenault; CHECK-NEXT:    store i32 3, ptr [[B]], align 4
52*f4b925eeSMatt Arsenault; CHECK-NEXT:    [[TMP0:%.*]] = getelementptr inbounds [[STRUCT_ST:%.*]], ptr [[S:%.*]], i64 [[T:%.*]], i32 0
53b6b1006fSAndrew Litteken; CHECK-NEXT:    ret void
54b6b1006fSAndrew Litteken;
55b6b1006fSAndrew Littekenentry:
56b6b1006fSAndrew Litteken  %a = alloca i32, align 4
57b6b1006fSAndrew Litteken  %b = alloca i32, align 4
58*f4b925eeSMatt Arsenault  store i32 2, ptr %a, align 4
59*f4b925eeSMatt Arsenault  store i32 3, ptr %b, align 4
60*f4b925eeSMatt Arsenault  %0 = getelementptr inbounds %struct.ST, ptr %s, i64 %t, i32 0
61b6b1006fSAndrew Litteken  ret void
62b6b1006fSAndrew Litteken}
63b6b1006fSAndrew Litteken
64*f4b925eeSMatt Arsenault; CHECK: define internal void @outlined_ir_func_0(ptr [[ARG0:%.*]], ptr [[ARG1:%.*]], ptr [[ARG2:%.*]], i64 [[ARG3:%.*]])
65b6b1006fSAndrew Litteken; CHECK: entry_to_outline:
66*f4b925eeSMatt Arsenault; CHECK-NEXT:    store i32 2, ptr [[ARG0]], align 4
67*f4b925eeSMatt Arsenault; CHECK-NEXT:    store i32 3, ptr [[ARG1]], align 4
68*f4b925eeSMatt Arsenault; CHECK-NEXT:    [[TMP0:%.*]] = getelementptr inbounds %struct.ST, ptr [[ARG2]], i64 [[ARG3]], i32 1
69