1; Tests that coro-split removes cleanup code after coro.end in resume functions 2; and retains it in the start function. 3; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s 4 5define ptr @f(i1 %val) presplitcoroutine personality i32 3 { 6entry: 7 %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) 8 %hdl = call ptr @llvm.coro.begin(token %id, ptr null) 9 call void @print(i32 0) 10 br i1 %val, label %resume, label %susp 11 12susp: 13 %0 = call i8 @llvm.coro.suspend(token none, i1 false) 14 switch i8 %0, label %suspend [i8 0, label %resume 15 i8 1, label %suspend] 16resume: 17 invoke void @print(i32 1) to label %suspend unwind label %lpad 18 19suspend: 20 call i1 @llvm.coro.end(ptr %hdl, i1 0, token none) 21 call void @print(i32 0) ; should not be present in f.resume 22 ret ptr %hdl 23 24lpad: 25 %lpval = landingpad { ptr, i32 } 26 cleanup 27 28 call void @print(i32 2) 29 %need.resume = call i1 @llvm.coro.end(ptr null, i1 true, token none) 30 br i1 %need.resume, label %eh.resume, label %cleanup.cont 31 32cleanup.cont: 33 call void @print(i32 3) ; should not be present in f.resume 34 br label %eh.resume 35 36eh.resume: 37 resume { ptr, i32 } %lpval 38} 39 40; Verify that start function contains both print calls the one before and after coro.end 41; CHECK-LABEL: define ptr @f( 42; CHECK: invoke void @print(i32 1) 43; CHECK: to label %AfterCoroEnd unwind label %lpad 44 45; CHECK: AfterCoroEnd: 46; CHECK: call void @print(i32 0) 47; CHECK: ret ptr %hdl 48 49; CHECK: lpad: 50; CHECK-NEXT: %lpval = landingpad { ptr, i32 } 51; CHECK-NEXT: cleanup 52; CHECK-NEXT: call void @print(i32 2) 53; CHECK-NEXT: call void @print(i32 3) 54; CHECK-NEXT: resume { ptr, i32 } %lpval 55 56; VERIFY Resume Parts 57 58; Verify that resume function does not contains both print calls appearing after coro.end 59; CHECK-LABEL: define internal fastcc void @f.resume 60; CHECK: invoke void @print(i32 1) 61; CHECK: to label %CoroEnd unwind label %lpad 62 63; CHECK: CoroEnd: 64; CHECK-NEXT: ret void 65 66; CHECK: lpad: 67; CHECK-NEXT: %lpval = landingpad { ptr, i32 } 68; CHECK-NEXT: cleanup 69; CHECK-NEXT: call void @print(i32 2) 70; Checks that the coroutine would be marked as done if it exits in unwinding path. 71; CHECK-NEXT: store ptr null, ptr %hdl, align 8 72; CHECK-NEXT: resume { ptr, i32 } %lpval 73 74declare ptr @llvm.coro.free(token, ptr) 75declare i32 @llvm.coro.size.i32() 76declare i8 @llvm.coro.suspend(token, i1) 77declare void @llvm.coro.resume(ptr) 78declare void @llvm.coro.destroy(ptr) 79 80declare token @llvm.coro.id(i32, ptr, ptr, ptr) 81declare ptr @llvm.coro.alloc(token) 82declare ptr @llvm.coro.begin(token, ptr) 83declare i1 @llvm.coro.end(ptr, i1, token) 84 85declare noalias ptr @malloc(i32) 86declare void @print(i32) 87declare void @free(ptr) 88