107 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py2; RUN: opt -S -passes=iroutliner -ir-outlining-no-cost < %s | FileCheck %s3 4; These functions are constructed slightly differently so that they require5; different output blocks for the values used outside of the region. We are6; checking that two output blocks are created with different values.7 8define void @outline_outputs1() #0 {9; CHECK-LABEL: @outline_outputs1(10; CHECK-NEXT: entry:11; CHECK-NEXT: [[DOTLOC:%.*]] = alloca i32, align 412; CHECK-NEXT: [[ADD_LOC:%.*]] = alloca i32, align 413; CHECK-NEXT: [[A:%.*]] = alloca i32, align 414; CHECK-NEXT: [[B:%.*]] = alloca i32, align 415; CHECK-NEXT: [[OUTPUT:%.*]] = alloca i32, align 416; CHECK-NEXT: [[RESULT:%.*]] = alloca i32, align 417; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr [[ADD_LOC]])18; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr [[DOTLOC]])19; CHECK-NEXT: call void @outlined_ir_func_0(ptr [[A]], ptr [[B]], ptr [[OUTPUT]], ptr [[ADD_LOC]], ptr [[DOTLOC]], i32 0)20; CHECK-NEXT: [[ADD_RELOAD:%.*]] = load i32, ptr [[ADD_LOC]], align 421; CHECK-NEXT: [[DOTRELOAD:%.*]] = load i32, ptr [[DOTLOC]], align 422; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr [[ADD_LOC]])23; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr [[DOTLOC]])24; CHECK-NEXT: [[TMP0:%.*]] = load i32, ptr [[OUTPUT]], align 425; CHECK-NEXT: call void @outlined_ir_func_1(i32 [[DOTRELOAD]], i32 [[ADD_RELOAD]], ptr [[RESULT]])26; CHECK-NEXT: ret void27;28entry:29 %a = alloca i32, align 430 %b = alloca i32, align 431 %output = alloca i32, align 432 %result = alloca i32, align 433 store i32 2, ptr %a, align 434 store i32 3, ptr %b, align 435 %0 = load i32, ptr %a, align 436 %1 = load i32, ptr %b, align 437 %add = add i32 %0, %138 %sub = sub i32 %0, %139 store i32 %add, ptr %output, align 440 %2 = load i32, ptr %output, align 441 %3 = load i32, ptr %output, align 442 %mul = mul i32 %2, %add43 store i32 %mul, ptr %result, align 444 ret void45}46 47define void @outline_outputs2() #0 {48; CHECK-LABEL: @outline_outputs2(49; CHECK-NEXT: entry:50; CHECK-NEXT: [[DOTLOC:%.*]] = alloca i32, align 451; CHECK-NEXT: [[SUB_LOC:%.*]] = alloca i32, align 452; CHECK-NEXT: [[A:%.*]] = alloca i32, align 453; CHECK-NEXT: [[B:%.*]] = alloca i32, align 454; CHECK-NEXT: [[OUTPUT:%.*]] = alloca i32, align 455; CHECK-NEXT: [[RESULT:%.*]] = alloca i32, align 456; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr [[SUB_LOC]])57; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr [[DOTLOC]])58; CHECK-NEXT: call void @outlined_ir_func_0(ptr [[A]], ptr [[B]], ptr [[OUTPUT]], ptr [[SUB_LOC]], ptr [[DOTLOC]], i32 1)59; CHECK-NEXT: [[SUB_RELOAD:%.*]] = load i32, ptr [[SUB_LOC]], align 460; CHECK-NEXT: [[DOTRELOAD:%.*]] = load i32, ptr [[DOTLOC]], align 461; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr [[SUB_LOC]])62; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr [[DOTLOC]])63; CHECK-NEXT: call void @outlined_ir_func_1(i32 [[DOTRELOAD]], i32 [[SUB_RELOAD]], ptr [[RESULT]])64; CHECK-NEXT: ret void65;66entry:67 %a = alloca i32, align 468 %b = alloca i32, align 469 %output = alloca i32, align 470 %result = alloca i32, align 471 store i32 2, ptr %a, align 472 store i32 3, ptr %b, align 473 %0 = load i32, ptr %a, align 474 %1 = load i32, ptr %b, align 475 %add = add i32 %0, %176 %sub = sub i32 %0, %177 store i32 %add, ptr %output, align 478 %2 = load i32, ptr %output, align 479 %mul = mul i32 %2, %sub80 store i32 %mul, ptr %result, align 481 ret void82}83 84; CHECK: define internal void @outlined_ir_func_0(ptr [[ARG0:%.*]], ptr [[ARG1:%.*]], ptr [[ARG2:%.*]], ptr [[ARG3:%.*]], ptr [[ARG4:%.*]], i32 [[ARG5:%.*]]) #1 {85; CHECK: entry_to_outline:86; CHECK-NEXT: store i32 2, ptr [[ARG0]], align 487; CHECK-NEXT: store i32 3, ptr [[ARG1]], align 488; CHECK-NEXT: [[TMP0:%.*]] = load i32, ptr [[ARG0]], align 489; CHECK-NEXT: [[TMP1:%.*]] = load i32, ptr [[ARG1]], align 490; CHECK-NEXT: [[ADD:%.*]] = add i32 [[TMP0]], [[TMP1]]91; CHECK-NEXT: [[SUB:%.*]] = sub i32 [[TMP0]], [[TMP1]]92; CHECK-NEXT: store i32 [[ADD]], ptr [[ARG2]], align 493; CHECK-NEXT: [[TMP2:%.*]] = load i32, ptr [[ARG2]], align 494 95; CHECK: _after_outline.exitStub:96; CHECK-NEXT: switch i32 [[ARG5]], label [[BLOCK:%.*]] [97; CHECK-NEXT: i32 0, label %[[BLOCK_0:.*]]98; CHECK-NEXT: i32 1, label %[[BLOCK_1:.*]]99 100; CHECK: [[BLOCK_0]]:101; CHECK-NEXT: store i32 [[ADD]], ptr [[ARG3]], align 4102; CHECK-NEXT: store i32 [[TMP2]], ptr [[ARG4]], align 4103 104; CHECK: [[BLOCK_1]]:105; CHECK-NEXT: store i32 [[SUB]], ptr [[ARG3]], align 4106; CHECK-NEXT: store i32 [[TMP2]], ptr [[ARG4]], align 4107