Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 20 additions & 2 deletions compiler/noirc_frontend/src/hir_def/types.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1547,12 +1547,15 @@ impl Type {
) -> Result<(), UnificationError> {
use Type::*;

let lhs = match self {
let lhs = self.follow_bindings_shallow();
let rhs = other.follow_bindings_shallow();

let lhs = match lhs.as_ref() {
Type::InfixExpr(..) => Cow::Owned(self.canonicalize()),
other => Cow::Borrowed(other),
};

let rhs = match other {
let rhs = match rhs.as_ref() {
Type::InfixExpr(..) => Cow::Owned(other.canonicalize()),
other => Cow::Borrowed(other),
};
Expand Down Expand Up @@ -2386,6 +2389,21 @@ impl Type {
}
}

/// Follow bindings if this is a type variable or generic to the first non-typevariable
/// type. Unlike `follow_bindings`, this won't recursively follow any bindings on any
/// fields or arguments of this type.
pub fn follow_bindings_shallow(&self) -> Cow<Type> {
match self {
Type::TypeVariable(var) | Type::NamedGeneric(var, _) => {
if let TypeBinding::Bound(typ) = &*var.borrow() {
return Cow::Owned(typ.follow_bindings_shallow().into_owned());
}
Cow::Borrowed(self)
}
other => Cow::Borrowed(other),
}
}

pub fn from_generics(generics: &GenericTypeVars) -> Vec<Type> {
vecmap(generics, |var| Type::TypeVariable(var.clone()))
}
Expand Down
6 changes: 4 additions & 2 deletions compiler/noirc_frontend/src/monomorphization/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -959,7 +959,8 @@ impl<'interner> Monomorphizer<'interner> {

/// Convert a non-tuple/struct type to a monomorphized type
fn convert_type(typ: &HirType, location: Location) -> Result<ast::Type, MonomorphizationError> {
Ok(match typ {
let typ = typ.follow_bindings_shallow();
Ok(match typ.as_ref() {
HirType::FieldElement => ast::Type::Field,
HirType::Integer(sign, bits) => ast::Type::Integer(*sign, *bits),
HirType::Bool => ast::Type::Bool,
Expand Down Expand Up @@ -1125,7 +1126,8 @@ impl<'interner> Monomorphizer<'interner> {

// Similar to `convert_type` but returns an error if any type variable can't be defaulted.
fn check_type(typ: &HirType, location: Location) -> Result<(), MonomorphizationError> {
match typ {
let typ = typ.follow_bindings_shallow();
match typ.as_ref() {
HirType::FieldElement
| HirType::Integer(..)
| HirType::Bool
Expand Down