tvm_tirx::ir::module

Struct IRModule

Source
#[repr(C)]
pub struct IRModule { /* private fields */ }

Implementations§

Source§

impl IRModule

Source

pub fn new( functions: impl Into<Map<GlobalVar, BaseFunc>>, attrs: impl Into<DictAttrs>, global_infos: impl Into<Map<String, Array<GlobalInfo>>>, global_var_map: impl Into<Map<String, GlobalVar>>, ) -> Self

Lossless complete-field allocation.

Source§

impl IRModule

Source

pub fn from_functions( functions: &Map<GlobalVar, BaseFunc>, attrs: Option<&DictAttrs>, global_infos: &Map<String, Array<GlobalInfo>>, ) -> Result<Self>

The module of functions, with the attributes attrs (empty if None) and the module-level information global_infos; it indexes the functions by name, which must be distinct (C++ IRModule(functions, attrs, global_infos)).

Source

pub fn from_expr( expr: impl IntoExpr, functions: &Map<GlobalVar, BaseFunc>, ) -> Result<Self>

The module of functions and expr, a function bound to its global symbol or to main (ir.IRModuleFromExpr).

Source

pub fn shallow_copy(&self) -> Result<Self>

A copy of the module that shares its functions (ir.IRModuleClone).

Source

pub fn add( &self, var: impl Into<GlobalVar>, func: impl Into<BaseFunc>, update: bool, ) -> Result<()>

Add func as var, replacing an existing definition; the native side currently ignores update (ir.IRModuleAdd).

Source

pub fn remove(&self, name: &str) -> Result<()>

Remove the function named name (ir.IRModuleRemove).

Source

pub fn remove_global_var(&self, var: impl Into<GlobalVar>) -> Result<()>

Remove the function of var (ir.IRModuleRemove).

Source

pub fn contains(&self, name: &str) -> Result<bool>

Whether the module has a global variable named name (ir.IRModuleContains).

Source

pub fn contains_global_var(&self, var: impl Into<GlobalVar>) -> Result<bool>

Whether the module defines a function for var (ir.IRModuleContains).

Source

pub fn has_global_var(&self, name: &str) -> Result<bool>

Whether the module has a global variable named name (ir.IRModuleContainGlobalVar).

Source

pub fn get_global_var(&self, name: &str) -> Result<GlobalVar>

The global variable named name; fails if there is none (ir.IRModuleGetGlobalVar).

Source

pub fn get_global_vars(&self) -> Result<Array<GlobalVar>>

The global variables of the module, sorted by name (ir.IRModuleGetGlobalVars).

Source

pub fn lookup(&self, var: impl Into<GlobalVar>) -> Result<BaseFunc>

The function of var; fails if there is none (ir.IRModuleLookup).

Source

pub fn lookup_name(&self, name: &str) -> Result<BaseFunc>

The function named name; fails if there is none (ir.IRModuleLookupStr).

Source

pub fn update(&self, other: &IRModule) -> Result<()>

Add or replace the functions of other in the module (ir.IRModuleUpdate).

Source

pub fn update_func( &self, var: impl Into<GlobalVar>, func: impl Into<BaseFunc>, ) -> Result<()>

Set the function of var to func (ir.IRModuleUpdateFunction).

Source

pub fn update_global_info( &self, name: &str, global_info: impl IntoIterator<Item = impl Into<GlobalInfo>>, ) -> Result<()>

Set the module-level information name to global_info (ir.IRModuleUpdateGlobalInfo).

Source

pub fn get_attr(&self, key: &str) -> Result<Option<Any>>

The attribute key of the module, if set (ir.IRModuleGetAttr).

Source

pub fn with_attr(self, key: &str, value: impl Into<Any>) -> Result<IRModule>

The module with the attribute key set to value (ir.IRModuleWithAttr).

Consumes self: the module is updated in place when this was its only handle, and copied otherwise.

Source

pub fn with_attrs(self, attrs: &Map<String, Any>) -> Result<IRModule>

The module with the attributes of attrs set (ir.IRModuleWithAttrs).

Consumes self, as IRModule::with_attr does.

Source

pub fn without_attr(self, key: &str) -> Result<IRModule>

The module without the attribute key (ir.IRModuleWithoutAttr).

Consumes self, as IRModule::with_attr does.

Trait Implementations§

Source§

impl AnyCompatible for IRModule

Source§

unsafe fn is_undefined(field: *const Self) -> bool

Whether field, which holds a value of this type or all-zero bytes, holds all-zero bytes that are no value of this type, as a null object reference, a string of type index None or a zero device type: a field that a creator left undefined for reflection to set, as C++ UnsafeInit leaves a reference undefined (null). Unlike a C++ undefined reference, it is no value; a nullable reference is Option<T>, whose None is a defined value. Read more
Source§

fn type_str() -> String

the type string of the type
Source§

unsafe fn copy_to_any_view(src: &Self, data: &mut TVMFFIAny)

the value to copy to TVMFFIAny
Source§

unsafe fn check_any_strict(data: &TVMFFIAny) -> bool

Source§

unsafe fn copy_from_any_view_after_check(data: &TVMFFIAny) -> Self

Copy value from TVMFFIAny after checking caller must ensure that the value is compatible with the type
Source§

unsafe fn move_to_any(src: Self, data: &mut TVMFFIAny)

consume the value to move to Any
Source§

unsafe fn move_from_any_after_check(data: &mut TVMFFIAny) -> Self

the value to move from TVMFFIAny NOTE: pay very careful attention to avoid memory leak! Read more
Source§

unsafe fn try_cast_from_any_view(data: &TVMFFIAny) -> Result<Self, ()>

try to cast the value from AnyView
Source§

fn get_mismatch_type_info(data: &TVMFFIAny) -> String

Get the type key of a type when TryCastFromAnyView fails.
Source§

fn type_schema() -> String

The type schema of the type: the JSON object C++ TypeTraits<T>::TypeSchema writes, which the type schemas of functions and fields embed. Read more
Source§

fn to_any(&self) -> Any

Borrow self into an owned Any, increfing object-backed values. Read more
Source§

impl<'a> ArgIntoRef for &'a IRModule

Source§

type Target = IRModule

Source§

fn to_ref(&self) -> &Self::Target

Source§

impl ArgIntoRef for IRModule

Source§

type Target = IRModule

Source§

fn to_ref(&self) -> &Self::Target

Source§

impl Clone for IRModule

Source§

fn clone(&self) -> IRModule

Returns a copy of the value. Read more
1.0.0 · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Deref for IRModule

Source§

type Target = IRModuleObj

The resulting type after dereferencing.
Source§

fn deref(&self) -> &IRModuleObj

Dereferences the value.
Source§

impl From<&IRModule> for IRModule

Source§

fn from(value: &IRModule) -> Self

Converts to this type from the input type.
Source§

impl<'a> IntoArgHolder for &'a IRModule

Source§

impl IntoArgHolder for IRModule

Source§

impl ObjectRefCore for IRModule

Source§

type ContainerType = <ObjectArc<IRModuleObj> as Deref>::Target

Source§

fn data(this: &Self) -> &ObjectArc<Self::ContainerType>

Source§

fn into_data(this: Self) -> ObjectArc<Self::ContainerType>

Source§

unsafe fn from_data(data: ObjectArc<Self::ContainerType>) -> Self

Construct a reference view from an owning container handle. Read more
Source§

fn same_as<Other>(&self, other: &Other) -> bool
where Other: ObjectRefCore,

Return whether two object references point to the same allocation.
Source§

fn downcast_ref<T>(&self) -> Option<&T>
where T: ObjectCore,

Borrow the underlying object as T when its runtime type is T or a subtype. Read more
Source§

impl TryFrom<Any> for IRModule

Source§

type Error = Error

The type returned in the event of a conversion error.
Source§

fn try_from(value: Any) -> Result<Self, Self::Error>

Performs the conversion.
Source§

impl<'a> TryFrom<AnyView<'a>> for IRModule

Source§

type Error = Error

The type returned in the event of a conversion error.
Source§

fn try_from(value: AnyView<'a>) -> Result<Self, Self::Error>

Performs the conversion.

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> ArgType for T
where T: ObjectRefCore + AnyCompatible + 'static,

Source§

fn check(value: &TVMFFIAny) -> bool

Whether value, which is not None, is accepted (C++ CheckAnyStrict).
Source§

fn type_str() -> String

The expected type, for an error (C++ TypeStr).
Source§

fn mismatch(value: &TVMFFIAny) -> String

The rejected value, for an error (C++ GetMismatchTypeInfo): its type key.
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dst: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dst. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> ObjectRefCast for T

Source§

fn try_cast<B>(self) -> Result<B, Error>

Consume self and rewrap the underlying object as B without copying.
Source§

impl<P, T> Receiver for P
where P: Deref<Target = T> + ?Sized, T: ?Sized,

Source§

type Target = T

🔬This is a nightly-only experimental API. (arbitrary_self_types)
The target type on which the method may be called.
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.