327141fb1d
implement the option2 of P2014R0 This implements the option2 of https://www.open-std.org/jtc1/sc22/wg21/docs/papers/2020/p2014r0.pdf. This also fixes https://github.com/llvm/llvm-project/issues/56671. Although wg21 didn't get consensus for the direction of the problem, we're happy to have some implementation and user experience first. And from issue56671, the option2 should be the pursued one. Reviewed By: ychen Differential Revision: https://reviews.llvm.org/D133341
124 lines
3.9 KiB
C++
124 lines
3.9 KiB
C++
// Tests that the combination of -fcoro-aligned-allocation and -fsized-deallocation works well.
|
|
// Test the compiler will chose sized deallocation correctly.
|
|
// This is only enabled with `-fsized-deallocation` which is off by default.
|
|
// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -std=c++20 \
|
|
// RUN: -fcoro-aligned-allocation -S -emit-llvm %s -o - -disable-llvm-passes \
|
|
// RUN: -fsized-deallocation \
|
|
// RUN: | FileCheck %s
|
|
|
|
#include "Inputs/coroutine.h"
|
|
|
|
namespace std {
|
|
typedef __SIZE_TYPE__ size_t;
|
|
enum class align_val_t : size_t {};
|
|
}
|
|
|
|
struct task {
|
|
struct promise_type {
|
|
auto initial_suspend() { return std::suspend_always{}; }
|
|
auto final_suspend() noexcept { return std::suspend_always{}; }
|
|
auto get_return_object() { return task{}; }
|
|
void unhandled_exception() {}
|
|
void return_value(int) {}
|
|
};
|
|
};
|
|
|
|
// CHECK: define{{.*}}@_Z1fv
|
|
// CHECK: coro.free:
|
|
// CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
|
|
// CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
|
|
// CHECK: call{{.*}}void @_ZdlPvmSt11align_val_t(ptr{{.*}}, i64{{.*}}%[[coro_size]], i64{{.*}}%[[coro_align]])
|
|
|
|
task f() {
|
|
co_return 43;
|
|
}
|
|
|
|
struct task2 {
|
|
struct promise_type {
|
|
auto initial_suspend() { return std::suspend_always{}; }
|
|
auto final_suspend() noexcept { return std::suspend_always{}; }
|
|
auto get_return_object() { return task2{}; }
|
|
void unhandled_exception() {}
|
|
void return_value(int) {}
|
|
void operator delete(void *ptr);
|
|
};
|
|
};
|
|
|
|
// CHECK: define{{.*}}@_Z2f2v
|
|
// CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
|
|
// CHECK: coro.free:
|
|
// CHECK: call{{.*}}void @_ZN5task212promise_typedlEPv(ptr{{.*}} %[[FREE_HANDLE]])
|
|
|
|
task2 f2() {
|
|
co_return 43;
|
|
}
|
|
|
|
struct task3 {
|
|
struct promise_type {
|
|
auto initial_suspend() { return std::suspend_always{}; }
|
|
auto final_suspend() noexcept { return std::suspend_always{}; }
|
|
auto get_return_object() { return task3{}; }
|
|
void unhandled_exception() {}
|
|
void return_value(int) {}
|
|
void operator delete(void *ptr, std::size_t);
|
|
void operator delete(void *ptr);
|
|
};
|
|
};
|
|
|
|
// CHECK: define{{.*}}@_Z2f3v
|
|
// CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
|
|
// CHECK: coro.free:
|
|
// CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
|
|
// CHECK: call{{.*}}void @_ZN5task312promise_typedlEPvm(ptr{{.*}} %[[FREE_HANDLE]], i64{{.*}}%[[coro_size]]
|
|
|
|
task3 f3() {
|
|
co_return 43;
|
|
}
|
|
|
|
struct task4 {
|
|
struct promise_type {
|
|
auto initial_suspend() { return std::suspend_always{}; }
|
|
auto final_suspend() noexcept { return std::suspend_always{}; }
|
|
auto get_return_object() { return task4{}; }
|
|
void unhandled_exception() {}
|
|
void return_value(int) {}
|
|
void operator delete(void *ptr, std::size_t);
|
|
void operator delete(void *ptr, std::align_val_t);
|
|
void operator delete(void *ptr);
|
|
};
|
|
};
|
|
|
|
// CHECK: define{{.*}}@_Z2f4v
|
|
// CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
|
|
// CHECK: coro.free:
|
|
// CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
|
|
// CHECK: call{{.*}}void @_ZN5task412promise_typedlEPvSt11align_val_t(ptr{{.*}} %[[FREE_HANDLE]], i64{{.*}}%[[coro_align]])
|
|
|
|
task4 f4() {
|
|
co_return 43;
|
|
}
|
|
|
|
struct task5 {
|
|
struct promise_type {
|
|
auto initial_suspend() { return std::suspend_always{}; }
|
|
auto final_suspend() noexcept { return std::suspend_always{}; }
|
|
auto get_return_object() { return task5{}; }
|
|
void unhandled_exception() {}
|
|
void return_value(int) {}
|
|
void operator delete(void *ptr, std::size_t);
|
|
void operator delete(void *ptr, std::size_t, std::align_val_t);
|
|
void operator delete(void *ptr);
|
|
};
|
|
};
|
|
|
|
// CHECK: define{{.*}}@_Z2f5v
|
|
// CHECK: %[[FREE_HANDLE:.+]] = call{{.*}}ptr @llvm.coro.free(
|
|
// CHECK: coro.free:
|
|
// CHECK: %[[coro_size:.+]] = call{{.*}}@llvm.coro.size
|
|
// CHECK: %[[coro_align:.+]] = call{{.*}}@llvm.coro.align
|
|
// CHECK: call{{.*}}void @_ZN5task512promise_typedlEPvmSt11align_val_t(ptr{{.*}} %[[FREE_HANDLE]], i64{{.*}}%[[coro_size]], i64{{.*}}%[[coro_align]])
|
|
|
|
task5 f5() {
|
|
co_return 43;
|
|
}
|