91 lines · cpp
1// RUN: %clang_cc1 --std=c++20 -triple x86_64-linux-gnu -emit-llvm %s -o - | FileCheck %s2 3#include "Inputs/coroutine.h"4 5struct Printy {6 Printy(const char *name) : name(name) {}7 ~Printy() {}8 const char *name;9};10 11struct coroutine {12 struct promise_type;13 std::coroutine_handle<promise_type> handle;14 ~coroutine() {15 if (handle) handle.destroy();16 }17};18 19struct coroutine::promise_type {20 coroutine get_return_object() {21 return {std::coroutine_handle<promise_type>::from_promise(*this)};22 }23 std::suspend_never initial_suspend() noexcept { return {}; }24 std::suspend_always final_suspend() noexcept { return {}; }25 void return_void() {}26 void unhandled_exception() {}27};28 29struct Awaiter : std::suspend_always {30 Printy await_resume() { return {"awaited"}; }31};32 33int foo() { return 2; }34 35coroutine ArrayInitCoro() {36 // Verify that:37 // - We do the necessary stores for array cleanups.38 // - Array cleanups are called by await.cleanup.39 // - We activate the cleanup after the first element and deactivate it in await.ready (see cleanup.isactive).40 41 // CHECK-LABEL: define dso_local void @_Z13ArrayInitCorov42 // CHECK: %arrayinit.endOfInit = alloca ptr, align 843 // CHECK: %cleanup.isactive = alloca i1, align 144 Printy arr[2] = {45 Printy("a"),46 // CHECK: store i1 true, ptr %cleanup.isactive.reload.addr, align 147 // CHECK-NEXT: store ptr %arr.reload.addr, ptr %arrayinit.endOfInit.reload.addr, align 848 // CHECK-NEXT: call void @_ZN6PrintyC1EPKc(ptr noundef nonnull align 8 dereferenceable(8) %arr.reload.addr, ptr noundef @.str)49 // CHECK-NEXT: %arrayinit.element = getelementptr inbounds %struct.Printy, ptr %arr.reload.addr, i64 150 // CHECK-NEXT: %arrayinit.element.spill.addr = getelementptr inbounds %_Z13ArrayInitCorov.Frame, ptr %0, i32 0, i32 1051 // CHECK-NEXT: store ptr %arrayinit.element, ptr %arrayinit.element.spill.addr, align 852 // CHECK-NEXT: store ptr %arrayinit.element, ptr %arrayinit.endOfInit.reload.addr, align 853 co_await Awaiter{}54 // CHECK-NEXT: @_ZNSt14suspend_always11await_readyEv55 // CHECK-NEXT: br i1 %{{.+}}, label %await.ready, label %CoroSave3056 };57 // CHECK: await.cleanup: ; preds = %AfterCoroSuspend{{.*}}58 // CHECK-NEXT: br label %cleanup{{.*}}.from.await.cleanup59 60 // CHECK: cleanup{{.*}}.from.await.cleanup: ; preds = %await.cleanup61 // CHECK: br label %cleanup{{.*}}62 63 // CHECK: await.ready:64 // CHECK-NEXT: %arrayinit.element.reload.addr = getelementptr inbounds %_Z13ArrayInitCorov.Frame, ptr %0, i32 0, i32 1065 // CHECK-NEXT: %arrayinit.element.reload = load ptr, ptr %arrayinit.element.reload.addr, align 866 // CHECK-NEXT: call void @_ZN7Awaiter12await_resumeEv67 // CHECK-NEXT: store i1 false, ptr %cleanup.isactive.reload.addr, align 168 // CHECK-NEXT: br label %cleanup{{.*}}.from.await.ready69 70 // CHECK: cleanup{{.*}}: ; preds = %cleanup{{.*}}.from.await.ready, %cleanup{{.*}}.from.await.cleanup71 // CHECK: %cleanup.is_active = load i1, ptr %cleanup.isactive.reload.addr, align 172 // CHECK-NEXT: br i1 %cleanup.is_active, label %cleanup.action, label %cleanup.done73 74 // CHECK: cleanup.action:75 // CHECK: %arraydestroy.isempty = icmp eq ptr %arr.reload.addr, %{{.*}}76 // CHECK-NEXT: br i1 %arraydestroy.isempty, label %arraydestroy.done{{.*}}, label %arraydestroy.body.from.cleanup.action77 // Ignore rest of the array cleanup.78}79 80coroutine ArrayInitWithCoReturn() {81 // CHECK-LABEL: define dso_local void @_Z21ArrayInitWithCoReturnv82 // Verify that we start to emit the array destructor.83 // CHECK: %arrayinit.endOfInit = alloca ptr, align 884 Printy arr[2] = {"a", ({85 if (foo()) {86 co_return;87 }88 "b";89 })};90}91