1 // RUN: %clang_cc1 -no-opaque-pointers -triple x86_64-unknown-linux-gnu -emit-llvm -std=c++20 \
2 // RUN:   -O0 %s -o - | FileCheck %s
3 // RUN: %clang_cc1 -no-opaque-pointers -triple x86_64-unknown-linux-gnu -emit-llvm -std=c++20 \
4 // RUN:   -fno-inline -O0 %s -o - | FileCheck %s
5 
6 namespace std {
7 
8 struct handle {};
9 
10 struct awaitable {
11   bool await_ready() noexcept { return true; }
12   // CHECK-NOT: await_suspend
13   inline void __attribute__((__always_inline__)) await_suspend(handle) noexcept {}
14   bool await_resume() noexcept { return true; }
15 };
16 
17 template <typename T>
18 struct coroutine_handle {
19   static handle from_address(void *address) noexcept { return {}; }
20 };
21 
22 template <typename T = void>
23 struct coroutine_traits {
24   struct promise_type {
25     awaitable initial_suspend() { return {}; }
26     awaitable final_suspend() noexcept { return {}; }
27     void return_void() {}
28     T get_return_object() { return T(); }
29     void unhandled_exception() {}
30   };
31 };
32 } // namespace std
33 
34 // CHECK-LABEL: @_Z3foov
35 // CHECK-LABEL: entry:
36 // CHECK: [[CAST0:%[0-9]+]] = bitcast %"struct.std::awaitable"* %ref.tmp{{.*}} to i8*
37 // CHECK-NEXT: call void @llvm.lifetime.start.p0i8(i64 1, i8* [[CAST0]])
38 // CHECK: [[CAST1:%[0-9]+]] = bitcast %"struct.std::awaitable"* %ref.tmp{{.*}} to i8*
39 // CHECK-NEXT: call void @llvm.lifetime.end.p0i8(i64 1, i8* [[CAST1]])
40 
41 // CHECK: [[CAST2:%[0-9]+]] = bitcast %"struct.std::awaitable"* %ref.tmp{{.*}} to i8*
42 // CHECK-NEXT: call void @llvm.lifetime.start.p0i8(i64 1, i8* [[CAST2]])
43 // CHECK: [[CAST3:%[0-9]+]] = bitcast %"struct.std::awaitable"* %ref.tmp{{.*}} to i8*
44 // CHECK-NEXT: call void @llvm.lifetime.end.p0i8(i64 1, i8* [[CAST3]])
45 void foo() { co_return; }
46 
47 // Check that bar is not inlined even it's marked as always_inline.
48 
49 // CHECK-LABEL:   define {{.*}} void @_Z3bazv()
50 // CHECK:         call void @_Z3barv(
51 __attribute__((__always_inline__)) void bar() {
52   co_return;
53 }
54 void baz() {
55   bar();
56   co_return;
57 }
58