102 lines · plain
1; Coroutine calls marked with `coro_elide_safe` should be elided.2; Inside `caller`, we expect the `callee` coroutine to be elided.3; Inside `caller_conditional`, `callee` is only called on an unlikely4; path, hence we expect the `callee` coroutine NOT to be elided.5;6; RUN: opt < %s -S -passes='cgscc(coro-annotation-elide)' | FileCheck %s7 8%struct.Task = type { ptr }9 10declare void @print(i32) nounwind11 12; resume part of the coroutine13define fastcc void @callee.resume(ptr dereferenceable(1)) {14 tail call void @print(i32 0)15 ret void16}17 18; destroy part of the coroutine19define fastcc void @callee.destroy(ptr) {20 tail call void @print(i32 1)21 ret void22}23 24; cleanup part of the coroutine25define fastcc void @callee.cleanup(ptr) {26 tail call void @print(i32 2)27 ret void28}29 30@callee.resumers = internal constant [3 x ptr] [31 ptr @callee.resume, ptr @callee.destroy, ptr @callee.cleanup]32 33declare void @alloc(i1) nounwind34 35; CHECK-LABEL: define ptr @callee36define ptr @callee(i8 %arg) {37entry:38 %task = alloca %struct.Task, align 839 %id = call token @llvm.coro.id(i32 0, ptr null,40 ptr @callee,41 ptr @callee.resumers)42 %alloc = call i1 @llvm.coro.alloc(token %id)43 %hdl = call ptr @llvm.coro.begin(token %id, ptr null)44 store ptr %hdl, ptr %task45 ret ptr %task46}47 48; CHECK-LABEL: define ptr @callee.noalloc49define ptr @callee.noalloc(i8 %arg, ptr dereferenceable(32) align(8) %frame) {50 entry:51 %task = alloca %struct.Task, align 852 %id = call token @llvm.coro.id(i32 0, ptr null,53 ptr @callee,54 ptr @callee.resumers)55 %hdl = call ptr @llvm.coro.begin(token %id, ptr null)56 store ptr %hdl, ptr %task57 ret ptr %task58}59 60; CHECK-LABEL: define ptr @caller()61; Function Attrs: presplitcoroutine62define ptr @caller() #0 {63entry:64 %task = call ptr @callee(i8 0) coro_elide_safe65 ret ptr %task66 ; CHECK: %[[TASK:.+]] = alloca %struct.Task, align 867 ; CHECK-NEXT: %[[FRAME:.+]] = alloca [32 x i8], align 868 ; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr %[[TASK]])69 ; CHECK-NEXT: %[[ID:.+]] = call token @llvm.coro.id(i32 0, ptr null, ptr @callee, ptr @callee.resumers)70 ; CHECK-NEXT: %[[HDL:.+]] = call ptr @llvm.coro.begin(token %[[ID]], ptr null)71 ; CHECK-NEXT: store ptr %[[HDL]], ptr %[[TASK]], align 872 ; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr %[[TASK]])73 ; CHECK-NEXT: ret ptr %[[TASK]]74}75 76; CHECK-LABEL: define ptr @caller_conditional(i1 %cond)77; Function Attrs: presplitcoroutine78define ptr @caller_conditional(i1 %cond) #0 {79entry:80 br i1 %cond, label %call, label %ret81 82call:83 ; CHECK-NOT: alloca84 ; CHECK-NOT: @llvm.coro.id({{.*}}, ptr @callee, {{.*}})85 ; CHECK: %task = call ptr @callee(i8 0)86 ; CHECK-NEXT: br label %ret87 %task = call ptr @callee(i8 0) coro_elide_safe88 br label %ret89 90ret:91 %retval = phi ptr [ %task, %call ], [ null, %entry ]92 ret ptr %retval93}94 95declare token @llvm.coro.id(i32, ptr, ptr, ptr)96declare ptr @llvm.coro.begin(token, ptr)97declare ptr @llvm.coro.frame()98declare ptr @llvm.coro.subfn.addr(ptr, i8)99declare i1 @llvm.coro.alloc(token)100 101attributes #0 = { presplitcoroutine }102