fb289ecde0
Summary: Certain implicitly generated coroutine statements, such as the calls to 'return_value()' or `return_void()` or `get_return_object_on_allocation_failure()`, cannot be built until the promise type is no longer dependent. This means they are not built until after the coroutine body statement has been transformed. This patch fixes an issue where these statements would never be built for coroutine templates. It also fixes a small issue where diagnostics about `get_return_object_on_allocation_failure()` were incorrectly suppressed. Reviewers: rsmith, majnemer, GorNishanov, aaron.ballman Reviewed By: GorNishanov Subscribers: cfe-commits Differential Revision: https://reviews.llvm.org/D31487 llvm-svn: 300504
168 lines
5.5 KiB
C++
168 lines
5.5 KiB
C++
// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -fcoroutines-ts -std=c++14 -emit-llvm %s -o - -disable-llvm-passes | FileCheck %s
|
|
|
|
namespace std {
|
|
namespace experimental {
|
|
template <typename... T>
|
|
struct coroutine_traits; // expected-note {{declared here}}
|
|
|
|
template <class Promise = void>
|
|
struct coroutine_handle {
|
|
coroutine_handle() = default;
|
|
static coroutine_handle from_address(void *) { return {}; }
|
|
};
|
|
|
|
template <>
|
|
struct coroutine_handle<void> {
|
|
static coroutine_handle from_address(void *) { return {}; }
|
|
coroutine_handle() = default;
|
|
template <class PromiseType>
|
|
coroutine_handle(coroutine_handle<PromiseType>) {}
|
|
};
|
|
|
|
} // end namespace experimental
|
|
|
|
struct nothrow_t {};
|
|
constexpr nothrow_t nothrow = {};
|
|
|
|
} // end namespace std
|
|
|
|
// Required when get_return_object_on_allocation_failure() is defined by
|
|
// the promise.
|
|
using SizeT = decltype(sizeof(int));
|
|
void* operator new(SizeT __sz, const std::nothrow_t&) noexcept;
|
|
void operator delete(void* __p, const std::nothrow_t&) noexcept;
|
|
|
|
|
|
struct suspend_always {
|
|
bool await_ready() { return false; }
|
|
void await_suspend(std::experimental::coroutine_handle<>) {}
|
|
void await_resume() {}
|
|
};
|
|
|
|
struct global_new_delete_tag {};
|
|
|
|
template<>
|
|
struct std::experimental::coroutine_traits<void, global_new_delete_tag> {
|
|
struct promise_type {
|
|
void get_return_object() {}
|
|
suspend_always initial_suspend() { return {}; }
|
|
suspend_always final_suspend() { return {}; }
|
|
void return_void() {}
|
|
};
|
|
};
|
|
|
|
// CHECK-LABEL: f0(
|
|
extern "C" void f0(global_new_delete_tag) {
|
|
// CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16
|
|
// CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64()
|
|
// CHECK: call i8* @_Znwm(i64 %[[SIZE]])
|
|
|
|
// CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame()
|
|
// CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]])
|
|
// CHECK: call void @_ZdlPv(i8* %[[MEM]])
|
|
co_return;
|
|
}
|
|
|
|
struct promise_new_tag {};
|
|
|
|
template<>
|
|
struct std::experimental::coroutine_traits<void, promise_new_tag> {
|
|
struct promise_type {
|
|
void *operator new(unsigned long);
|
|
void get_return_object() {}
|
|
suspend_always initial_suspend() { return {}; }
|
|
suspend_always final_suspend() { return {}; }
|
|
void return_void() {}
|
|
};
|
|
};
|
|
|
|
// CHECK-LABEL: f1(
|
|
extern "C" void f1(promise_new_tag ) {
|
|
// CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16
|
|
// CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64()
|
|
// CHECK: call i8* @_ZNSt12experimental16coroutine_traitsIJv15promise_new_tagEE12promise_typenwEm(i64 %[[SIZE]])
|
|
|
|
// CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame()
|
|
// CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]])
|
|
// CHECK: call void @_ZdlPv(i8* %[[MEM]])
|
|
co_return;
|
|
}
|
|
|
|
struct promise_delete_tag {};
|
|
|
|
template<>
|
|
struct std::experimental::coroutine_traits<void, promise_delete_tag> {
|
|
struct promise_type {
|
|
void operator delete(void*);
|
|
void get_return_object() {}
|
|
suspend_always initial_suspend() { return {}; }
|
|
suspend_always final_suspend() { return {}; }
|
|
void return_void() {}
|
|
};
|
|
};
|
|
|
|
// CHECK-LABEL: f2(
|
|
extern "C" void f2(promise_delete_tag) {
|
|
// CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16
|
|
// CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64()
|
|
// CHECK: call i8* @_Znwm(i64 %[[SIZE]])
|
|
|
|
// CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame()
|
|
// CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]])
|
|
// CHECK: call void @_ZNSt12experimental16coroutine_traitsIJv18promise_delete_tagEE12promise_typedlEPv(i8* %[[MEM]])
|
|
co_return;
|
|
}
|
|
|
|
struct promise_sized_delete_tag {};
|
|
|
|
template<>
|
|
struct std::experimental::coroutine_traits<void, promise_sized_delete_tag> {
|
|
struct promise_type {
|
|
void operator delete(void*, unsigned long);
|
|
void get_return_object() {}
|
|
suspend_always initial_suspend() { return {}; }
|
|
suspend_always final_suspend() { return {}; }
|
|
void return_void() {}
|
|
};
|
|
};
|
|
|
|
// CHECK-LABEL: f3(
|
|
extern "C" void f3(promise_sized_delete_tag) {
|
|
// CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16
|
|
// CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64()
|
|
// CHECK: call i8* @_Znwm(i64 %[[SIZE]])
|
|
|
|
// CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame()
|
|
// CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]])
|
|
// CHECK: %[[SIZE2:.+]] = call i64 @llvm.coro.size.i64()
|
|
// CHECK: call void @_ZNSt12experimental16coroutine_traitsIJv24promise_sized_delete_tagEE12promise_typedlEPvm(i8* %[[MEM]], i64 %[[SIZE2]])
|
|
co_return;
|
|
}
|
|
|
|
struct promise_on_alloc_failure_tag {};
|
|
|
|
template<>
|
|
struct std::experimental::coroutine_traits<int, promise_on_alloc_failure_tag> {
|
|
struct promise_type {
|
|
int get_return_object() {}
|
|
suspend_always initial_suspend() { return {}; }
|
|
suspend_always final_suspend() { return {}; }
|
|
void return_void() {}
|
|
static int get_return_object_on_allocation_failure() { return -1; }
|
|
};
|
|
};
|
|
|
|
// CHECK-LABEL: f4(
|
|
extern "C" int f4(promise_on_alloc_failure_tag) {
|
|
// CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16
|
|
// CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64()
|
|
// CHECK: %[[MEM:.+]] = call i8* @_ZnwmRKSt9nothrow_t(i64 %[[SIZE]], %"struct.std::nothrow_t"* dereferenceable(1) @_ZStL7nothrow)
|
|
// CHECK: %[[OK:.+]] = icmp ne i8* %[[MEM]], null
|
|
// CHECK: br i1 %[[OK]], label %[[OKBB:.+]], label %[[ERRBB:.+]]
|
|
|
|
// CHECK: [[ERRBB]]:
|
|
// CHECK: %[[RETVAL:.+]] = call i32 @_ZNSt12experimental16coroutine_traitsIJi28promise_on_alloc_failure_tagEE12promise_type39get_return_object_on_allocation_failureEv(
|
|
// CHECK: ret i32 %[[RETVAL]]
|
|
co_return;
|
|
}
|