1 // RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -std=c++20 \
2 // RUN: -fcoro-aligned-allocation -emit-llvm %s -o - -disable-llvm-passes \
3 // RUN: | FileCheck %s
4
5 #include "Inputs/coroutine.h"
6
7 namespace std {
8 typedef __SIZE_TYPE__ size_t;
9 enum class align_val_t : size_t {};
10 }
11
12 struct task {
13 struct promise_type {
initial_suspendtask::promise_type14 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask::promise_type15 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask::promise_type16 auto get_return_object() { return task{}; }
unhandled_exceptiontask::promise_type17 void unhandled_exception() {}
return_valuetask::promise_type18 void return_value(int) {}
19 };
20 };
21
22 // CHECK: define{{.*}}@_Z1fv(
23 // CHECK: coro.alloc:
24 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
25 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
26 // CHECK: %[[aligned_new:.+]] = call{{.*}}@_ZnwmSt11align_val_t({{.*}}%[[coro_size]],{{.*}}%[[coro_align]])
27
28 // CHECK: coro.free:
29 // CHECK: %[[coro_size_for_free:.+]] = call{{.*}}@llvm.coro.size
30 // CHECK: %[[coro_align_for_free:.+]] = call{{.*}}@llvm.coro.align
31 // CHECK: call void @_ZdlPvmSt11align_val_t({{.*}}%[[coro_size_for_free]],{{.*}}%[[coro_align_for_free]])
32
f()33 task f() {
34 co_return 43;
35 }
36
37 struct task2 {
38 struct promise_type {
initial_suspendtask2::promise_type39 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask2::promise_type40 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask2::promise_type41 auto get_return_object() { return task2{}; }
unhandled_exceptiontask2::promise_type42 void unhandled_exception() {}
return_valuetask2::promise_type43 void return_value(int) {}
get_return_object_on_allocation_failuretask2::promise_type44 static task2 get_return_object_on_allocation_failure() { return task2{}; }
45 };
46 };
47
48 namespace std {
49 struct nothrow_t {};
50 constexpr nothrow_t nothrow = {};
51 }
52
53 void *operator new(std::size_t, std::align_val_t, std::nothrow_t) noexcept;
54
55 // CHECK: define{{.*}}@_Z2f2v(
56 // CHECK: coro.alloc:
57 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
58 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
59 // CHECK: %[[aligned_new:.+]] = call{{.*}}@_ZnwmSt11align_val_tSt9nothrow_t({{.*}}%[[coro_size]],{{.*}}%[[coro_align]])
60
61 // CHECK: coro.free:
62 // CHECK: %[[coro_size_for_free:.+]] = call{{.*}}@llvm.coro.size
63 // CHECK: %[[coro_align_for_free:.+]] = call{{.*}}@llvm.coro.align
64 // CHECK: call void @_ZdlPvmSt11align_val_t({{.*}}%[[coro_size_for_free]],{{.*}}%[[coro_align_for_free]])
65
f2()66 task2 f2() {
67 co_return 43;
68 }
69
70 struct task3 {
71 struct promise_type {
initial_suspendtask3::promise_type72 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask3::promise_type73 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask3::promise_type74 auto get_return_object() { return task3{}; }
unhandled_exceptiontask3::promise_type75 void unhandled_exception() {}
return_valuetask3::promise_type76 void return_value(int) {}
77 void operator delete(void *ptr);
78 };
79 };
80
81 // CHECK: define{{.*}}@_Z2f3v
82 // CHECK: coro.free:
83 // CHECK: call{{.*}}void @_ZN5task312promise_typedlEPv(
84
f3()85 task3 f3() {
86 co_return 43;
87 }
88
89 struct task4 {
90 struct promise_type {
initial_suspendtask4::promise_type91 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask4::promise_type92 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask4::promise_type93 auto get_return_object() { return task4{}; }
unhandled_exceptiontask4::promise_type94 void unhandled_exception() {}
return_valuetask4::promise_type95 void return_value(int) {}
96 void operator delete(void *ptr, std::align_val_t);
97 void operator delete(void *ptr);
98 };
99 };
100
101 // CHECK: define{{.*}}@_Z2f4v
102 // CHECK: coro.free:
103 // CHECK: %[[coro_align_for_free:.+]] = call{{.*}}@llvm.coro.align
104 // CHECK: call{{.*}}void @_ZN5task412promise_typedlEPvSt11align_val_t({{.*}}, i64{{.*}}[[coro_align_for_free]]
105
f4()106 task4 f4() {
107 co_return 43;
108 }
109
110 struct task5 {
111 struct promise_type {
initial_suspendtask5::promise_type112 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask5::promise_type113 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask5::promise_type114 auto get_return_object() { return task5{}; }
unhandled_exceptiontask5::promise_type115 void unhandled_exception() {}
return_valuetask5::promise_type116 void return_value(int) {}
117 void *operator new(std::size_t);
118 };
119 };
120
121 // CHECK: define{{.*}}@_Z2f5v
122 // CHECK: coro.alloc:
123 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
124 // CHECK: call{{.*}}ptr @_ZN5task512promise_typenwEm(i64{{.*}}%[[coro_size]])
f5()125 task5 f5() {
126 co_return 43;
127 }
128
129 struct task6 {
130 struct promise_type {
initial_suspendtask6::promise_type131 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask6::promise_type132 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask6::promise_type133 auto get_return_object() { return task6{}; }
unhandled_exceptiontask6::promise_type134 void unhandled_exception() {}
return_valuetask6::promise_type135 void return_value(int) {}
136 void *operator new(std::size_t);
137 void *operator new(std::size_t, int i);
138 };
139 };
140
141 // CHECK: define{{.*}}@_Z2f6i
142 // CHECK: coro.alloc:
143 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
144 // CHECK: call{{.*}}ptr @_ZN5task612promise_typenwEmi(i64{{.*}}%[[coro_size]],
f6(int i)145 task6 f6(int i) {
146 co_return i;
147 }
148
149 struct task7 {
150 struct promise_type {
initial_suspendtask7::promise_type151 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask7::promise_type152 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask7::promise_type153 auto get_return_object() { return task7{}; }
unhandled_exceptiontask7::promise_type154 void unhandled_exception() {}
return_valuetask7::promise_type155 void return_value(int) {}
156 void *operator new(std::size_t);
157 void *operator new(std::size_t, int i);
158 void *operator new(std::size_t, std::align_val_t);
159 };
160 };
161
162 // CHECK: define{{.*}}@_Z2f7i
163 // CHECK: coro.alloc:
164 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
165 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
166 // CHECK: call{{.*}}ptr @_ZN5task712promise_typenwEmSt11align_val_t(i64{{.*}}%[[coro_size]], i64{{.*}}[[coro_align]])
f7(int i)167 task7 f7(int i) {
168 co_return i;
169 }
170
171 struct task8 {
172 struct promise_type {
initial_suspendtask8::promise_type173 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask8::promise_type174 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask8::promise_type175 auto get_return_object() { return task8{}; }
unhandled_exceptiontask8::promise_type176 void unhandled_exception() {}
return_valuetask8::promise_type177 void return_value(int) {}
178 void *operator new(std::size_t);
179 void *operator new(std::size_t, int i);
180 void *operator new(std::size_t, std::align_val_t);
181 void *operator new(std::size_t, std::align_val_t, int i);
182 };
183 };
184
185 // CHECK: define{{.*}}@_Z2f8i
186 // CHECK: coro.alloc:
187 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
188 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
189 // CHECK: call{{.*}}ptr @_ZN5task812promise_typenwEmSt11align_val_ti(i64{{.*}}%[[coro_size]], i64{{.*}}[[coro_align]],
f8(int i)190 task8 f8(int i) {
191 co_return i;
192 }
193