98 lines · plain
1; Tests that coro-split will convert coro.await.suspend.handle to a musttail call.2; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck --check-prefixes=CHECK %s3; RUN: opt < %s -passes='pgo-instr-gen,cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck --check-prefixes=CHECK %s4 5define void @f() #0 {6entry:7 %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)8 %alloc = call ptr @malloc(i64 16) #39 %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc)10 11 %save = call token @llvm.coro.save(ptr null)12 %addr1 = call ptr @llvm.coro.subfn.addr(ptr null, i8 0)13 call fastcc void %addr1(ptr null)14 15 %suspend = call i8 @llvm.coro.suspend(token %save, i1 false)16 switch i8 %suspend, label %exit [17 i8 0, label %await.suspend18 i8 1, label %exit19 ]20await.suspend:21 %save2 = call token @llvm.coro.save(ptr null)22 %br0 = call i8 @switch_result()23 switch i8 %br0, label %unreach [24 i8 0, label %await.resume325 i8 1, label %await.resume126 i8 2, label %await.resume227 ]28await.resume1:29 %hdl = call ptr @g()30 call void @llvm.coro.await.suspend.handle(ptr null, ptr %hdl, ptr @await_suspend_function)31 br label %final.suspend32await.resume2:33 %hdl2 = call ptr @h()34 call void @llvm.coro.await.suspend.handle(ptr null, ptr %hdl2, ptr @await_suspend_function)35 br label %final.suspend36await.resume3:37 call void @llvm.coro.await.suspend.handle(ptr null, ptr null, ptr @await_suspend_function)38 br label %final.suspend39final.suspend:40 %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false)41 switch i8 %suspend2, label %exit [42 i8 0, label %pre.exit43 i8 1, label %exit44 ]45pre.exit:46 br label %exit47exit:48 call void @llvm.coro.end(ptr null, i1 false, token none)49 ret void50unreach:51 unreachable52}53 54; Verify that in the initial function resume is not marked with musttail.55; CHECK-LABEL: @f(56; CHECK: %[[addr1:.+]] = call ptr @llvm.coro.subfn.addr(ptr null, i8 0)57; CHECK-NOT: musttail call fastcc void %[[addr1]](ptr null)58 59; Verify that in the resume part resume call is marked with musttail.60; CHECK-LABEL: @f.resume(61; CHECK: %[[hdl:.+]] = call ptr @g()62; CHECK-NEXT: call ptr @await_suspend_function63; CHECK-NEXT: %[[addr2:.+]] = call ptr @llvm.coro.subfn.addr64; CHECK-NEXT: musttail call fastcc void %[[addr2]]65; CHECK-NEXT: ret void66; CHECK: %[[hdl2:.+]] = call ptr @h()67; CHECK-NEXT: call ptr @await_suspend_function68; CHECK-NEXT: %[[addr3:.+]] = call ptr @llvm.coro.subfn.addr69; CHECK-NEXT: musttail call fastcc void %[[addr3]]70; CHECK-NEXT: ret void71; CHECK: call ptr @await_suspend_function72; CHECK: %[[addr4:.+]] = call ptr @llvm.coro.subfn.addr73; CHECK-NEXT: musttail call fastcc void %[[addr4]]74; CHECK-NEXT: ret void75 76 77 78declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) #179declare i1 @llvm.coro.alloc(token) #280declare i64 @llvm.coro.size.i64() #381declare ptr @llvm.coro.begin(token, ptr writeonly) #282declare token @llvm.coro.save(ptr) #283declare ptr @llvm.coro.frame() #384declare i8 @llvm.coro.suspend(token, i1) #285declare ptr @llvm.coro.free(token, ptr nocapture readonly) #186declare void @llvm.coro.end(ptr, i1, token) #287declare ptr @llvm.coro.subfn.addr(ptr nocapture readonly, i8) #188declare ptr @malloc(i64)89declare i8 @switch_result()90declare ptr @g()91declare ptr @h()92declare ptr @await_suspend_function(ptr %awaiter, ptr %hdl)93 94attributes #0 = { presplitcoroutine }95attributes #1 = { argmemonly nounwind readonly }96attributes #2 = { nounwind }97attributes #3 = { nounwind readnone }98