xref: /llvm-project/llvm/test/Transforms/IROutliner/alloca-addrspace.ll (revision e8444078ed9ad040ad4ce9b47bfd3745db0953a7)
1192c0e5aSMatt Arsenault; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --include-generated-funcs
2192c0e5aSMatt Arsenault; RUN: opt -S -passes=iroutliner -ir-outlining-no-cost < %s | FileCheck %s
3192c0e5aSMatt Arsenault
4192c0e5aSMatt Arsenault; Check alloca with non-0 address spaces works correctly.
5192c0e5aSMatt Arsenault
6192c0e5aSMatt Arsenaulttarget datalayout = "A5"
7192c0e5aSMatt Arsenault
8192c0e5aSMatt Arsenaultdefine i32 @outlineable() {
9192c0e5aSMatt Arsenaultbb:
10192c0e5aSMatt Arsenault  %i = tail call i32 @func(i32 0, i32 1)
11192c0e5aSMatt Arsenault  %i1 = or i32 0, %i
12192c0e5aSMatt Arsenault  %i2 = tail call i32 @func(i32 %i1, i32 0)
13192c0e5aSMatt Arsenault  %i3 = or i32 %i1, %i2
14192c0e5aSMatt Arsenault  ret i32 0
15192c0e5aSMatt Arsenault}
16192c0e5aSMatt Arsenault
17192c0e5aSMatt Arsenaultdeclare i32 @func(i32, i32)
18192c0e5aSMatt Arsenault; CHECK-LABEL: define {{[^@]+}}@outlineable() {
19192c0e5aSMatt Arsenault; CHECK-NEXT:  bb:
20192c0e5aSMatt Arsenault; CHECK-NEXT:    [[I1_LOC:%.*]] = alloca i32, align 4, addrspace(5)
21*e8444078SMatt Arsenault; CHECK-NEXT:    call void @llvm.lifetime.start.p5(i64 -1, ptr addrspace(5) [[I1_LOC]])
22192c0e5aSMatt Arsenault; CHECK-NEXT:    call void @outlined_ir_func_0(i32 0, i32 1, ptr addrspace(5) [[I1_LOC]], i32 0)
23192c0e5aSMatt Arsenault; CHECK-NEXT:    [[I1_RELOAD:%.*]] = load i32, ptr addrspace(5) [[I1_LOC]], align 4
24*e8444078SMatt Arsenault; CHECK-NEXT:    call void @llvm.lifetime.end.p5(i64 -1, ptr addrspace(5) [[I1_LOC]])
25192c0e5aSMatt Arsenault; CHECK-NEXT:    call void @outlined_ir_func_0(i32 [[I1_RELOAD]], i32 0, ptr addrspace(5) null, i32 -1)
26192c0e5aSMatt Arsenault; CHECK-NEXT:    ret i32 0
27192c0e5aSMatt Arsenault;
28192c0e5aSMatt Arsenault;
29192c0e5aSMatt Arsenault; CHECK-LABEL: define {{[^@]+}}@outlined_ir_func_0
30192c0e5aSMatt Arsenault; CHECK-SAME: (i32 [[TMP0:%.*]], i32 [[TMP1:%.*]], ptr addrspace(5) [[TMP2:%.*]], i32 [[TMP3:%.*]]) #[[ATTR1:[0-9]+]] {
31192c0e5aSMatt Arsenault; CHECK-NEXT:  newFuncRoot:
32192c0e5aSMatt Arsenault; CHECK-NEXT:    br label [[BB_TO_OUTLINE:%.*]]
33192c0e5aSMatt Arsenault; CHECK:       bb_to_outline:
34192c0e5aSMatt Arsenault; CHECK-NEXT:    [[I:%.*]] = tail call i32 @func(i32 [[TMP0]], i32 [[TMP1]])
35192c0e5aSMatt Arsenault; CHECK-NEXT:    [[I1:%.*]] = or i32 [[TMP0]], [[I]]
36192c0e5aSMatt Arsenault; CHECK-NEXT:    br label [[BB_AFTER_OUTLINE_EXITSTUB:%.*]]
37192c0e5aSMatt Arsenault; CHECK:       bb_after_outline.exitStub:
38192c0e5aSMatt Arsenault; CHECK-NEXT:    switch i32 [[TMP3]], label [[FINAL_BLOCK_0:%.*]] [
39192c0e5aSMatt Arsenault; CHECK-NEXT:    i32 0, label [[OUTPUT_BLOCK_0_0:%.*]]
40192c0e5aSMatt Arsenault; CHECK-NEXT:    ]
41192c0e5aSMatt Arsenault; CHECK:       output_block_0_0:
42192c0e5aSMatt Arsenault; CHECK-NEXT:    store i32 [[I1]], ptr addrspace(5) [[TMP2]], align 4
43192c0e5aSMatt Arsenault; CHECK-NEXT:    br label [[FINAL_BLOCK_0]]
44192c0e5aSMatt Arsenault; CHECK:       final_block_0:
45192c0e5aSMatt Arsenault; CHECK-NEXT:    ret void
46192c0e5aSMatt Arsenault;
47