Record coroutine kind in generics
This commit is contained in:
parent
d6c46a23ce
commit
dcca9a12cd
@ -344,11 +344,18 @@ enum Defaults {
|
||||
kind: hir::ExprKind::Closure(hir::Closure { kind, .. }), ..
|
||||
}) = node
|
||||
{
|
||||
// See `ClosureArgsParts`, `CoroutineArgsParts`, and `CoroutineClosureArgsParts`
|
||||
// for info on the usage of each of these fields.
|
||||
let dummy_args = match kind {
|
||||
ClosureKind::Closure => &["<closure_kind>", "<closure_signature>", "<upvars>"][..],
|
||||
ClosureKind::Coroutine(_) => {
|
||||
&["<resume_ty>", "<yield_ty>", "<return_ty>", "<witness>", "<upvars>"][..]
|
||||
}
|
||||
ClosureKind::Coroutine(_) => &[
|
||||
"<coroutine_kind>",
|
||||
"<resume_ty>",
|
||||
"<yield_ty>",
|
||||
"<return_ty>",
|
||||
"<witness>",
|
||||
"<upvars>",
|
||||
][..],
|
||||
ClosureKind::CoroutineClosure(_) => &[
|
||||
"<closure_kind>",
|
||||
"<closure_signature_parts>",
|
||||
|
@ -765,7 +765,14 @@ fn polymorphize<'tcx>(
|
||||
let def_id = instance.def_id();
|
||||
let upvars_ty = match tcx.type_of(def_id).skip_binder().kind() {
|
||||
ty::Closure(..) => Some(args.as_closure().tupled_upvars_ty()),
|
||||
ty::Coroutine(..) => Some(args.as_coroutine().tupled_upvars_ty()),
|
||||
ty::Coroutine(..) => {
|
||||
assert_eq!(
|
||||
args.as_coroutine().kind_ty(),
|
||||
tcx.types.unit,
|
||||
"polymorphization does not support coroutines from async closures"
|
||||
);
|
||||
Some(args.as_coroutine().tupled_upvars_ty())
|
||||
}
|
||||
_ => None,
|
||||
};
|
||||
let has_upvars = upvars_ty.is_some_and(|ty| !ty.tuple_fields().is_empty());
|
||||
|
17
tests/ui/coroutine/polymorphize-args.rs
Normal file
17
tests/ui/coroutine/polymorphize-args.rs
Normal file
@ -0,0 +1,17 @@
|
||||
// compile-flags: -Zpolymorphize=on
|
||||
// build-pass
|
||||
|
||||
#![feature(coroutines, coroutine_trait)]
|
||||
|
||||
use std::ops::Coroutine;
|
||||
use std::pin::Pin;
|
||||
use std::thread;
|
||||
|
||||
fn main() {
|
||||
let mut foo = || yield;
|
||||
thread::spawn(move || match Pin::new(&mut foo).resume(()) {
|
||||
s => panic!("bad state: {:?}", s),
|
||||
})
|
||||
.join()
|
||||
.unwrap();
|
||||
}
|
Loading…
Reference in New Issue
Block a user