6dcb0eb301
Summary: This patch adds passing a coroutine_handle object to await_suspend calls. It builds the coroutine_handle using coroutine_handle<PromiseType>::from_address(__builtin_coro_frame()). (a revision of https://reviews.llvm.org/D26316 that for some reason refuses to apply via arc patch) Reviewers: GorNishanov Subscribers: mehdi_amini, cfe-commits, EricWF Differential Revision: https://reviews.llvm.org/D30769 llvm-svn: 297356
130 lines
4.0 KiB
C++
130 lines
4.0 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>) {}
|
|
};
|
|
|
|
}
|
|
}
|
|
|
|
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;
|
|
}
|