1 // Tests that the combination of -fcoro-aligned-allocation and -fsized-deallocation works well.
2 // Test the compiler will chose sized deallocation correctly.
3 // RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -std=c++20 \
4 // RUN: -fcoro-aligned-allocation -emit-llvm %s -o - -disable-llvm-passes \
5 // RUN: | FileCheck %s
6
7 #include "Inputs/coroutine.h"
8
9 namespace std {
10 typedef __SIZE_TYPE__ size_t;
11 enum class align_val_t : size_t {};
12 }
13
14 struct task {
15 struct promise_type {
initial_suspendtask::promise_type16 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask::promise_type17 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask::promise_type18 auto get_return_object() { return task{}; }
unhandled_exceptiontask::promise_type19 void unhandled_exception() {}
return_valuetask::promise_type20 void return_value(int) {}
21 };
22 };
23
24 // CHECK: define{{.*}}@_Z1fv
25 // CHECK: coro.free:
26 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
27 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
28 // CHECK: call{{.*}}void @_ZdlPvmSt11align_val_t(ptr{{.*}}, i64{{.*}}%[[coro_size]], i64{{.*}}%[[coro_align]])
29
f()30 task f() {
31 co_return 43;
32 }
33
34 struct task2 {
35 struct promise_type {
initial_suspendtask2::promise_type36 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask2::promise_type37 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask2::promise_type38 auto get_return_object() { return task2{}; }
unhandled_exceptiontask2::promise_type39 void unhandled_exception() {}
return_valuetask2::promise_type40 void return_value(int) {}
41 void operator delete(void *ptr);
42 };
43 };
44
45 // CHECK: define{{.*}}@_Z2f2v
46 // CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
47 // CHECK: coro.free:
48 // CHECK: call{{.*}}void @_ZN5task212promise_typedlEPv(ptr{{.*}} %[[FREE_HANDLE]])
49
f2()50 task2 f2() {
51 co_return 43;
52 }
53
54 struct task3 {
55 struct promise_type {
initial_suspendtask3::promise_type56 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask3::promise_type57 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask3::promise_type58 auto get_return_object() { return task3{}; }
unhandled_exceptiontask3::promise_type59 void unhandled_exception() {}
return_valuetask3::promise_type60 void return_value(int) {}
61 void operator delete(void *ptr, std::size_t);
62 void operator delete(void *ptr);
63 };
64 };
65
66 // CHECK: define{{.*}}@_Z2f3v
67 // CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
68 // CHECK: coro.free:
69 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
70 // CHECK: call{{.*}}void @_ZN5task312promise_typedlEPvm(ptr{{.*}} %[[FREE_HANDLE]], i64{{.*}}%[[coro_size]]
71
f3()72 task3 f3() {
73 co_return 43;
74 }
75
76 struct task4 {
77 struct promise_type {
initial_suspendtask4::promise_type78 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask4::promise_type79 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask4::promise_type80 auto get_return_object() { return task4{}; }
unhandled_exceptiontask4::promise_type81 void unhandled_exception() {}
return_valuetask4::promise_type82 void return_value(int) {}
83 void operator delete(void *ptr, std::size_t);
84 void operator delete(void *ptr, std::align_val_t);
85 void operator delete(void *ptr);
86 };
87 };
88
89 // CHECK: define{{.*}}@_Z2f4v
90 // CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
91 // CHECK: coro.free:
92 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
93 // CHECK: call{{.*}}void @_ZN5task412promise_typedlEPvSt11align_val_t(ptr{{.*}} %[[FREE_HANDLE]], i64{{.*}}%[[coro_align]])
94
f4()95 task4 f4() {
96 co_return 43;
97 }
98
99 struct task5 {
100 struct promise_type {
initial_suspendtask5::promise_type101 auto initial_suspend() { return std::suspend_always{}; }
final_suspendtask5::promise_type102 auto final_suspend() noexcept { return std::suspend_always{}; }
get_return_objecttask5::promise_type103 auto get_return_object() { return task5{}; }
unhandled_exceptiontask5::promise_type104 void unhandled_exception() {}
return_valuetask5::promise_type105 void return_value(int) {}
106 void operator delete(void *ptr, std::size_t);
107 void operator delete(void *ptr, std::size_t, std::align_val_t);
108 void operator delete(void *ptr);
109 };
110 };
111
112 // CHECK: define{{.*}}@_Z2f5v
113 // CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
114 // CHECK: coro.free:
115 // CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
116 // CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
117 // CHECK: call{{.*}}void @_ZN5task512promise_typedlEPvmSt11align_val_t(ptr{{.*}} %[[FREE_HANDLE]], i64{{.*}}%[[coro_size]], i64{{.*}}%[[coro_align]])
118
f5()119 task5 f5() {
120 co_return 43;
121 }
122