llzk 0.5.0

Rust bindings to the LLZK C API.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

//! Utilities related to type unification.

use std::ptr::null;

use melior::{StringRef, ir::TypeLike};
use mlir_sys::MlirStringRef;

/// Return `true` iff the two types are equivalent or could be equivalent after full
/// instantiation of struct parameters.
///
/// `rhs_reverse_prefix` describes the symbol path prefix to prepend to reverse references found on
/// the right-hand side type before attempting unification.
pub fn types_unify_with_prefix<'c>(
    lhs: impl TypeLike<'c>,
    rhs: impl TypeLike<'c>,
    rhs_reverse_prefix: &[StringRef<'_>],
) -> bool {
    let rhs_reverse_prefix: Vec<MlirStringRef> = rhs_reverse_prefix
        .iter()
        .map(|name| name.to_raw())
        .collect();
    let rhs_reverse_prefix_ptr = if rhs_reverse_prefix.is_empty() {
        null()
    } else {
        rhs_reverse_prefix.as_ptr()
    };

    unsafe {
        llzk_sys::llzkTypesUnify(
            lhs.to_raw(),
            rhs.to_raw(),
            isize::try_from(rhs_reverse_prefix.len()).expect("reverse prefix count too large"),
            rhs_reverse_prefix_ptr,
        )
    }
}

/// Return `true` iff the two types are equivalent without any symbol prefix adjustment.
#[inline]
pub fn types_unify<'c>(lhs: impl TypeLike<'c>, rhs: impl TypeLike<'c>) -> bool {
    types_unify_with_prefix(lhs, rhs, &[])
}