Merge pull request #20561 from ChayimFriedman2/no-table-for-you

minor: Don't require a full `InferenceTable` for `CastTy`
This commit is contained in:
Chayim Refael Friedman 2025-08-28 16:38:39 +00:00 committed by GitHub
commit 6e59defe11
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 51 additions and 58 deletions

View File

@ -6,7 +6,9 @@ use stdx::never;
use crate::{ use crate::{
Adjustment, Binders, DynTy, InferenceDiagnostic, Interner, PlaceholderIndex, Adjustment, Binders, DynTy, InferenceDiagnostic, Interner, PlaceholderIndex,
QuantifiedWhereClauses, Ty, TyExt, TyKind, TypeFlags, WhereClause, from_chalk_trait_id, QuantifiedWhereClauses, Ty, TyExt, TyKind, TypeFlags, WhereClause,
db::HirDatabase,
from_chalk_trait_id,
infer::{coerce::CoerceNever, unify::InferenceTable}, infer::{coerce::CoerceNever, unify::InferenceTable},
}; };
@ -30,7 +32,7 @@ pub(crate) enum CastTy {
} }
impl CastTy { impl CastTy {
pub(crate) fn from_ty(table: &mut InferenceTable<'_>, t: &Ty) -> Option<Self> { pub(crate) fn from_ty(db: &dyn HirDatabase, t: &Ty) -> Option<Self> {
match t.kind(Interner) { match t.kind(Interner) {
TyKind::Scalar(Scalar::Bool) => Some(Self::Int(Int::Bool)), TyKind::Scalar(Scalar::Bool) => Some(Self::Int(Int::Bool)),
TyKind::Scalar(Scalar::Char) => Some(Self::Int(Int::Char)), TyKind::Scalar(Scalar::Char) => Some(Self::Int(Int::Char)),
@ -43,8 +45,8 @@ impl CastTy {
let (AdtId::EnumId(id), _) = t.as_adt()? else { let (AdtId::EnumId(id), _) = t.as_adt()? else {
return None; return None;
}; };
let enum_data = id.enum_variants(table.db); let enum_data = id.enum_variants(db);
if enum_data.is_payload_free(table.db) { Some(Self::Int(Int::CEnum)) } else { None } if enum_data.is_payload_free(db) { Some(Self::Int(Int::CEnum)) } else { None }
} }
TyKind::Raw(m, ty) => Some(Self::Ptr(ty.clone(), *m)), TyKind::Raw(m, ty) => Some(Self::Ptr(ty.clone(), *m)),
TyKind::Function(_) => Some(Self::FnPtr), TyKind::Function(_) => Some(Self::FnPtr),
@ -142,58 +144,50 @@ impl CastCheck {
where where
F: FnMut(ExprId, Vec<Adjustment>), F: FnMut(ExprId, Vec<Adjustment>),
{ {
let (t_from, t_cast) = let (t_from, t_cast) = match (
match (CastTy::from_ty(table, &self.expr_ty), CastTy::from_ty(table, &self.cast_ty)) { CastTy::from_ty(table.db, &self.expr_ty),
(Some(t_from), Some(t_cast)) => (t_from, t_cast), CastTy::from_ty(table.db, &self.cast_ty),
(None, Some(t_cast)) => match self.expr_ty.kind(Interner) { ) {
TyKind::FnDef(..) => { (Some(t_from), Some(t_cast)) => (t_from, t_cast),
let sig = self.expr_ty.callable_sig(table.db).expect("FnDef had no sig"); (None, Some(t_cast)) => match self.expr_ty.kind(Interner) {
let sig = table.eagerly_normalize_and_resolve_shallow_in(sig); TyKind::FnDef(..) => {
let fn_ptr = TyKind::Function(sig.to_fn_ptr()).intern(Interner); let sig = self.expr_ty.callable_sig(table.db).expect("FnDef had no sig");
if let Ok((adj, _)) = table.coerce(&self.expr_ty, &fn_ptr, CoerceNever::Yes) let sig = table.eagerly_normalize_and_resolve_shallow_in(sig);
{ let fn_ptr = TyKind::Function(sig.to_fn_ptr()).intern(Interner);
apply_adjustments(self.source_expr, adj); if let Ok((adj, _)) = table.coerce(&self.expr_ty, &fn_ptr, CoerceNever::Yes) {
} else { apply_adjustments(self.source_expr, adj);
return Err(CastError::IllegalCast); } else {
} return Err(CastError::IllegalCast);
(CastTy::FnPtr, t_cast)
} }
TyKind::Ref(mutbl, _, inner_ty) => {
return match t_cast {
CastTy::Int(_) | CastTy::Float => match inner_ty.kind(Interner) {
TyKind::Scalar(
Scalar::Int(_) | Scalar::Uint(_) | Scalar::Float(_),
)
| TyKind::InferenceVar(
_,
TyVariableKind::Integer | TyVariableKind::Float,
) => Err(CastError::NeedDeref),
_ => Err(CastError::NeedViaPtr), (CastTy::FnPtr, t_cast)
}, }
// array-ptr-cast TyKind::Ref(mutbl, _, inner_ty) => {
CastTy::Ptr(t, m) => { return match t_cast {
let t = table.eagerly_normalize_and_resolve_shallow_in(t); CastTy::Int(_) | CastTy::Float => match inner_ty.kind(Interner) {
if !table.is_sized(&t) { TyKind::Scalar(Scalar::Int(_) | Scalar::Uint(_) | Scalar::Float(_))
return Err(CastError::IllegalCast); | TyKind::InferenceVar(
} _,
self.check_ref_cast( TyVariableKind::Integer | TyVariableKind::Float,
table, ) => Err(CastError::NeedDeref),
inner_ty,
*mutbl, _ => Err(CastError::NeedViaPtr),
&t, },
m, // array-ptr-cast
apply_adjustments, CastTy::Ptr(t, m) => {
) let t = table.eagerly_normalize_and_resolve_shallow_in(t);
if !table.is_sized(&t) {
return Err(CastError::IllegalCast);
} }
_ => Err(CastError::NonScalar), self.check_ref_cast(table, inner_ty, *mutbl, &t, m, apply_adjustments)
}; }
} _ => Err(CastError::NonScalar),
_ => return Err(CastError::NonScalar), };
}, }
_ => return Err(CastError::NonScalar), _ => return Err(CastError::NonScalar),
}; },
_ => return Err(CastError::NonScalar),
};
// rustc checks whether the `expr_ty` is foreign adt with `non_exhaustive` sym // rustc checks whether the `expr_ty` is foreign adt with `non_exhaustive` sym

View File

@ -31,7 +31,7 @@ use crate::{
display::{DisplayTarget, HirDisplay, hir_display_with_store}, display::{DisplayTarget, HirDisplay, hir_display_with_store},
error_lifetime, error_lifetime,
generics::generics, generics::generics,
infer::{CaptureKind, CapturedItem, TypeMismatch, cast::CastTy, unify::InferenceTable}, infer::{CaptureKind, CapturedItem, TypeMismatch, cast::CastTy},
inhabitedness::is_ty_uninhabited_from, inhabitedness::is_ty_uninhabited_from,
layout::LayoutError, layout::LayoutError,
mapping::ToChalk, mapping::ToChalk,
@ -948,8 +948,7 @@ impl<'ctx> MirLowerCtx<'ctx> {
let cast_kind = if source_ty.as_reference().is_some() { let cast_kind = if source_ty.as_reference().is_some() {
CastKind::PointerCoercion(PointerCast::ArrayToPointer) CastKind::PointerCoercion(PointerCast::ArrayToPointer)
} else { } else {
let mut table = InferenceTable::new(self.db, self.env.clone()); cast_kind(self.db, &source_ty, &target_ty)?
cast_kind(&mut table, &source_ty, &target_ty)?
}; };
Rvalue::Cast(cast_kind, it, target_ty) Rvalue::Cast(cast_kind, it, target_ty)
@ -2017,9 +2016,9 @@ impl<'ctx> MirLowerCtx<'ctx> {
} }
} }
fn cast_kind(table: &mut InferenceTable<'_>, source_ty: &Ty, target_ty: &Ty) -> Result<CastKind> { fn cast_kind(db: &dyn HirDatabase, source_ty: &Ty, target_ty: &Ty) -> Result<CastKind> {
let from = CastTy::from_ty(table, source_ty); let from = CastTy::from_ty(db, source_ty);
let cast = CastTy::from_ty(table, target_ty); let cast = CastTy::from_ty(db, target_ty);
Ok(match (from, cast) { Ok(match (from, cast) {
(Some(CastTy::Ptr(..) | CastTy::FnPtr), Some(CastTy::Int(_))) => { (Some(CastTy::Ptr(..) | CastTy::FnPtr), Some(CastTy::Int(_))) => {
CastKind::PointerExposeAddress CastKind::PointerExposeAddress