93 lines · plain
1; Tests that CoroSplit can succesfully determine allocas should live on the frame2; if their aliases are used across suspension points through PHINode.3; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s4 5define ptr @f(i1 %n) presplitcoroutine {6entry:7 %x = alloca i648 %y = alloca i649 %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)10 %size = call i32 @llvm.coro.size.i32()11 %alloc = call ptr @malloc(i32 %size)12 %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc)13 br i1 %n, label %flag_true, label %flag_false14 15flag_true:16 call void @llvm.lifetime.start.p0(ptr %x)17 br label %merge18 19flag_false:20 call void @llvm.lifetime.start.p0(ptr %y)21 br label %merge22 23merge:24 %alias_phi = phi ptr [ %x, %flag_true ], [ %y, %flag_false ]25 store i8 1, ptr %alias_phi26 %sp1 = call i8 @llvm.coro.suspend(token none, i1 false)27 switch i8 %sp1, label %suspend [i8 0, label %resume28 i8 1, label %cleanup]29resume:30 call void @print(ptr %alias_phi)31 br label %cleanup32 33cleanup:34 %mem = call ptr @llvm.coro.free(token %id, ptr %hdl)35 call void @free(ptr %mem)36 br label %suspend37 38suspend:39 call void @llvm.coro.end(ptr %hdl, i1 0, token none)40 ret ptr %hdl41}42 43declare ptr @llvm.coro.free(token, ptr)44declare i32 @llvm.coro.size.i32()45declare i8 @llvm.coro.suspend(token, i1)46declare void @llvm.coro.resume(ptr)47declare void @llvm.coro.destroy(ptr)48 49declare token @llvm.coro.id(i32, ptr, ptr, ptr)50declare i1 @llvm.coro.alloc(token)51declare ptr @llvm.coro.begin(token, ptr)52declare void @llvm.coro.end(ptr, i1, token)53 54declare void @llvm.lifetime.start.p0(ptr nocapture)55 56declare void @print(ptr)57declare noalias ptr @malloc(i32)58declare void @free(ptr)59 60; Verify that both x and y are put in the frame.61; CHECK: %f.Frame = type { ptr, ptr, i64, i64, ptr, i1 }62 63; CHECK-LABEL: @f(64; CHECK-NEXT: entry:65; CHECK-NEXT: [[ID:%.*]] = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr @f.resumers)66; CHECK-NEXT: [[ALLOC:%.*]] = call ptr @malloc(i32 48)67; CHECK-NEXT: [[HDL:%.*]] = call noalias nonnull ptr @llvm.coro.begin(token [[ID]], ptr [[ALLOC]])68; CHECK-NEXT: store ptr @f.resume, ptr [[HDL]], align 869; CHECK-NEXT: [[DESTROY_ADDR:%.*]] = getelementptr inbounds nuw [[F_FRAME:%f.Frame]], ptr [[HDL]], i32 0, i32 170; CHECK-NEXT: store ptr @f.destroy, ptr [[DESTROY_ADDR]], align 871; CHECK-NEXT: [[X_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[HDL]], i32 0, i32 272; CHECK-NEXT: [[Y_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[HDL]], i32 0, i32 373; CHECK-NEXT: br i1 [[N:%.*]], label [[MERGE:%.*]], label [[MERGE_FROM_FLAG_FALSE:%.*]]74; CHECK: merge.from.flag_false:75; CHECK-NEXT: br label [[MERGE:%.*]]76; CHECK: merge:77; CHECK-NEXT: [[ALIAS_PHI:%.*]] = phi ptr [ [[Y_RELOAD_ADDR]], [[MERGE_FROM_FLAG_FALSE]] ], [ [[X_RELOAD_ADDR]], [[ENTRY:%.*]] ]78; CHECK-NEXT: [[ALIAS_PHI_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[HDL]], i32 0, i32 479; CHECK-NEXT: store ptr [[ALIAS_PHI]], ptr [[ALIAS_PHI_SPILL_ADDR]], align 880; CHECK-NEXT: store i8 1, ptr [[ALIAS_PHI]], align 181; CHECK-NEXT: [[INDEX_ADDR1:%.*]] = getelementptr inbounds nuw [[F_FRAME]], ptr [[HDL]], i32 0, i32 582; CHECK-NEXT: store i1 false, ptr [[INDEX_ADDR1]], align 183; CHECK-NEXT: ret ptr [[HDL]]84;85;86; CHECK-LABEL: @f.resume(87; CHECK-NEXT: entry.resume:88; CHECK-NEXT: [[ALIAS_PHI_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], ptr [[HDL]], i32 0, i32 489; CHECK-NEXT: [[ALIAS_PHI_RELOAD:%.*]] = load ptr, ptr [[ALIAS_PHI_RELOAD_ADDR]], align 890; CHECK-NEXT: call void @print(ptr [[ALIAS_PHI_RELOAD]])91; CHECK-NEXT: call void @free(ptr [[FRAMEPTR:%.*]])92; CHECK-NEXT: ret void93