Base class for all Defs. More...

#include <mim/def.h>

Inheritance diagram for mim::Def:

Public Member Functions
Getters
World &	world () const noexcept
Driver &	driver () const noexcept
constexpr flags_t	flags () const noexcept
constexpr u32	gid () const noexcept
	Global id - unique number for this Def.
constexpr u32	mark () const noexcept
	Used internally by free_vars().
constexpr size_t	hash () const noexcept
constexpr Node	node () const noexcept
std::string_view	node_name () const
Diagnostics
fe::Error &	error () const noexcept
Loc	err_loc () const
	Returns a blame Loc from World::get_loc, this Def, or its nearest located dependency, in that order.
template<class... Args>
fe::Error &	blame (fe::cite_string< Args... > s, Args &&... args) const
	Reports an error that blames this; chain Error::n for Notes and Error::bail to throw.
Judgement
What kind of Judgement represents this Def?
Judge	judge () const noexcept
	Def must never become polymorphic: a vptr costs 8 bytes on every node in the World, and Def::ops_ptr hands out the operands at this + 1, so the vptr would also shift them.
bool	is_form () const noexcept
bool	is_intro () const noexcept
bool	is_elim () const noexcept
bool	is_meta () const noexcept
type
const Def *	type () const noexcept
	Yields the "raw" type of this Def (maybe nullptr).
const Def *	unfold_type () const
	Yields the type of this Def and builds a new Type (UInc n) if necessary.
template<class T>
const T *	isa_type () const
	Is Def::unfold_type a `T`? Yields nullptr for Univ, which has no type at all.
bool	is_term () const
	Is this Def a term, i.e. is its type() a Type?
const Def *	arity () const
	Number of elements available to Extract / Insert (may be dynamic).
ops
template<size_t N = std::dynamic_extent>
constexpr auto	ops () const noexcept
const Def *	op (size_t i) const noexcept
constexpr size_t	num_ops () const noexcept
Setting Ops (Mutables Only)
You can set and change the Def::ops of a mutable after construction. However, you have to obey the following rules: If Def::is_set() is ... false, set the operands from left to right. true, Def::unset() the operands first and then start over: mut->unset()->set({a, b, c}); MimIR assumes that a mutable is final, when its last operand is set. Then, Def::check() will be invoked.
bool	is_set () const
Def *	set (size_t i, const Def *)
	Successively set from left to right.
Def *	set (Defs ops)
	Set `ops` all at once (no Def::unset necessary beforehand).
Def *	unset ()
	Unsets all Def::ops; works even, if not set at all or only partially set.
Def *	set_type (const Def *)
	Update type.
deps
All dependencies of a Def and includes: Def::type() (if not nullptr) and the other Def::ops() (only included, if Def::is_set()) in this order.
Defs	deps () const noexcept
const Def *	dep (size_t i) const noexcept
size_t	num_deps () const noexcept
has_dep
Checks whether one Def::deps() contains specific elements defined in Dep. This works up to the next mutable. For example, consider the Tuple tup: (?, lam (x: Nat) = y): bool has_hole = tup->has_dep(Dep::Hole); // true bool has_mut = tup->has_dep(Dep::Mut); // true bool has_var = tup->has_dep(Dep::Var); // false - y is contained in another mutable mim::Def::has_dep bool has_dep() const noexcept Definition def.h:408 mim::Def::has_var const Var * has_var() Only returns not nullptr, if Var of this mutable has ever been created. Definition def.h:483 mim::Dep::Var @ Var Depends on a Var. Definition def.h:136 mim::Dep::Hole @ Hole Depends on a Hole. Definition def.h:137 mim::Dep::Mut @ Mut Depends on a mutable. Definition def.h:135
Dep	dep () const noexcept
bool	has_dep () const noexcept
bool	has_dep (Dep d) const noexcept
proj
Splits this Def via Extracts or directly accessing the Def::ops in the case of Sigmas or Arrays. std::array<const Def, 2> ab = def->projs<2>(); std::array<u64, 2> xy = def->projs<2>([](auto def) { return Lit::as(def); }); auto [a, b] = def->projs<2>(); auto [x, y] = def->projs<2>([](auto def) { return Lit::as(def); }); fe::Vector<const Def> projs1 = def->projs(); // "projs1" has def->num_projs() many elements fe::Vector<const Def> projs2 = def->projs(n);// "projs2" has n elements - asserts if incorrect // same as above but applies Lit::as<nat_t>(def) to each element fe::Vector<const Lit> lits1 = def->projs( [](auto def) { return Lit::as(def); }); fe::Vector<const Lit> lits2 = def->projs(n, [](auto def) { return Lit::as(def); }); mim::Lit::as static T as(const Def def) Definition def.h:943
nat_t	num_projs () const
	Yields Def::arity(), if it is a Lit, or 1 otherwise.
nat_t	num_tprojs () const
	As above but yields 1, if Flags::scalarize_threshold is exceeded.
const Def *	proj (nat_t a, nat_t i) const
	Similar to World::extract while assuming an arity of `a`, but also works on Sigmas and Arrays.
const Def *	proj (nat_t i) const
	As above but takes Def::num_projs as arity.
const Def *	tproj (nat_t i) const
	As above but takes Def::num_tprojs.
template<nat_t A = std::dynamic_extent, class F>
auto	projs (F f) const
	Splits this Def via Def::projections into an Array (if A == std::dynamic_extent) or std::array (otherwise).
template<class F>
auto	tprojs (F f) const
template<class F>
auto	projs (nat_t a, F f) const
template<nat_t A = std::dynamic_extent>
auto	projs () const
auto	tprojs () const
auto	projs (nat_t a) const
var
Retrieve Var for mutables. See also proj
nat_t	num_vars () noexcept
nat_t	num_tvars () noexcept
const Def *	var (nat_t a, nat_t i) noexcept
const Def *	var (nat_t i) noexcept
const Def *	tvar (nat_t i) noexcept
template<nat_t A = std::dynamic_extent, class F>
auto	vars (F f) noexcept
template<class F>
auto	tvars (F f) noexcept
template<nat_t A = std::dynamic_extent>
auto	vars () noexcept
auto	tvars () noexcept
template<class F>
auto	vars (nat_t a, F f) noexcept
auto	vars (nat_t a) noexcept
const Def *	var ()
	Not necessarily a Var: E.g., if the return type is [], this will yield ().
const Def *	var_type ()
	If this is a binder, compute the type of its Variable.
const Var *	has_var ()
	Only returns not nullptr, if Var of this mutable has ever been created.
const Var *	has_var () const
	As above if this is a mutable.
template<class D = Def>
std::pair< D , const Var >	isa_binder () const
	Is this a mutable that introduces a Var?
Free Vars and Muts
MimIR splits the free-variable analysis into a local and a global layer: local_muts() / local_vars() only look at the immutable fan-out and are cheap, cached, and hash-consed. free_vars() close over the mutable boundary as well and are the actual set of free Vars. They are computed on demand via a fixed-point iteration and cached in mutables. Mutating a mutable transitively invalidates these caches by following users().
Muts	local_muts () const
	Mutables reachable by following immutable deps(); mut->local_muts() is by definition the set { mut }.
Vars	local_vars () const
	Vars reachable by following immutable deps().
Vars	free_vars () const
	Global set of free Vars: extends local_vars() by transitively following mutables as well.
Vars	free_vars ()
	As above but drives (and caches) the fixed-point iteration for mutables.
Muts	users ()
	Set of mutables where this mutable is locally referenced.
bool	is_open () const
	Same as has_free_vars().
bool	is_closed () const
	Same as !has_free_vars().
bool	is_ground () const
	Immutable that contains neither mutables nor Vars.
Def *	outermost_binder () const
	Transitively walks up free_vars() till the outermoust binder has been found.
bool	nests (Def *mut)
	Does `this` nest `mut`?
bool	nests (const Def *def)
	Does `this` nest `def`?
free_vars predicates
free_vars() of an immutable is not cached: it merges free_vars() of every local_muts() entry on every call, and each merge allocates, hashes, and probes the pool. Since free_vars() is a union, any predicate over it distributes over that union - so these answer the question without ever materializing the merged set. Prefer them over free_vars().contains(...) / .empty() / has_intersection(...).
bool	has_free_var (const Var *) const
	Same as free_vars().contains(var).
bool	has_free_vars () const
	Same as !free_vars().empty().
bool	has_free_vars_in (Vars) const
	Same as vars.has_intersection(free_vars()).
external
bool	is_external () const noexcept
void	externalize ()
void	internalize ()
void	transfer_external (Def *to)
bool	is_annex () const noexcept
dirty
Scratch bit for Phases to mark muts that need re-examination. See also Phase::taint
bool	is_dirty () const noexcept
void	dirty (bool dirty=true) noexcept
Casts
See also Downcasts for Built-ins
bool	is_mutable () const noexcept
template<class T = Def>
const T *	isa_imm () const
template<class T = Def>
const T *	as_imm () const
template<class T = Def, bool invert = false>
T *	isa_mut () const
	If this is mutable, it will cast constness away and perform a dynamic_cast to `T`.
template<class T = Def, bool invert = false>
T *	as_mut () const
	Asserts that this is a mutable, casts constness away and performs a static_cast to `T`.
template<class T = Def, class... Args>
T *	expect_mut (std::format_string< Args... > fmt, Args &&... args) const
	Like Def::as_mut but - instead of merely asserting in Debug builds - throws via fe::throwf when the cast fails; the mutable counterpart of fe::RuntimeCast::expect (which Def inherits for the general case).
Dbg Getters
Dbg	dbg () const
	Looks up Def::dbg_ in Driver::dbg.
DbgKey	dbg_key () const
	Cheap handle for other->set(this->dbg_key()).
Loc	loc () const
Sym	sym () const
std::string	unique_name () const
	name + "_" + Def::gid
Dbg Setters
Every subclass S of Def has the same setters that return S/const S via the mixin Setters.
template<bool Ow = false>
const Def *	set (Loc l) const
template<bool Ow = false>
Def *	set (Loc l)
template<bool Ow = false>
const Def *	set (Sym s) const
template<bool Ow = false>
Def *	set (Sym s)
template<bool Ow = false>
const Def *	set (std::string s) const
template<bool Ow = false>
Def *	set (std::string s)
template<bool Ow = false>
const Def *	set (Loc l, Sym s) const
template<bool Ow = false>
Def *	set (Loc l, Sym s)
template<bool Ow = false>
const Def *	set (Loc l, std::string s) const
template<bool Ow = false>
Def *	set (Loc l, std::string s)
template<bool Ow = false>
const Def *	set (Dbg d) const
template<bool Ow = false>
Def *	set (Dbg d)
template<bool Ow = false>
const Def *	set (DbgKey key) const
	Adopts the Dbg behind `key` - just copies the interned index, so nothing is re-interned.
template<bool Ow = false>
Def *	set (DbgKey key)
debug_prefix/suffix
Prepends/Appends a prefix/suffix to Def::name - but only in Debug build.
const Def *	debug_prefix (std::string) const
const Def *	debug_suffix (std::string) const
Rebuild
const Def *	immutabilize ()
bool	is_immutabilizable ()
template<size_t N = std::dynamic_extent>
constexpr auto	reduce (const Def *arg) const
size_t	reduction_offset () const noexcept
	First Def::op that needs to be dealt with during reduction; e.g.
Type Checking
const Def *	check (size_t i, const Def *def)
	Checks whether the ith operand can be set to def.
const Def *	check ()
	After all Def::ops have been Def::set, this method will be invoked to check the type of this mutable.
bool	needs_zonk () const
	Yields true, if Def::local_muts() contain a Hole that is set.
const Def *	zonk () const
	If Holes have been filled, reconstruct the program without them.
const Def *	zonk_mut () const
	If mutable, zonk()s all ops and tries to immutabilize it; otherwise just zonk.
dump
Note While this output uses Mim syntax, it does usually not produce programs that can be read back. It uses an unscheduled visiting algorithm, and is only meant for debugging purposes.
void	dump () const
void	dump (int max) const
void	write (int max) const
void	write (int max, const char *file) const
std::ostream &	stream (std::ostream &, int max) const
dot
Streams dot to `os`, configured via `cfg` (see DotConfig).
void	dot (std::ostream &os, DotConfig cfg={}) const
void	dot (const char *file=nullptr, DotConfig cfg={}) const
	Same as above but write to `file` or std::cout if `file` is nullptr.
void	dot (const std::string &file, DotConfig cfg={}) const

Static Public Member Functions
static DefVec	zonk (Defs defs)
	zonks all `defs` and returns a new DefVec.

Protected Member Functions
C'tors and D'tors
	Def (World , Node, const Def type, Defs ops, flags_t flags)
	Constructor for an immutable Def.
	Def (Node, const Def *type, Defs ops, flags_t flags)
	As above but World retrieved from `type`.
	Def (Node, const Def *type, size_t num_ops, flags_t flags)
	Constructor for a mutable Def.
	Def (Node, Def *binder)
	Constructor for a Var; stores its `binder`.
Wrappers for World::sym
These are here to have Def::setters inline without including mim/world.h.
Sym	sym (const char *) const
Sym	sym (std::string_view) const
Sym	sym (std::string) const
void	set_dbg (Dbg) const
	Interns `dbg` via Driver::dbg and stores the key in Def::dbg_.
void	set_dbg_ (Dbg, bool ow) const
	Backs Def::set(Dbg).
void	set_dbg_key_ (DbgKey, bool ow) const
	Backs Def::set(DbgKey).

Protected Attributes
union mim::Def:: { ... }
flags_t	flags_
u8	curry_ = 0
u8	trip_ = 0

Friends
struct	DefKey
class	World
void	swap (World &, World &) noexcept
std::ostream &	operator<< (std::ostream &os, const Def *def)
	This will stream `def` as an operand.

Syntactic Comparison
Establishes an arbitrary but deterministic total order on Defs that is stable across runs.
enum class	Cmp { L , G , E , U }
static Cmp	cmp (const Def a, const Def b)
static bool	less (const Def a, const Def b)
static bool	greater (const Def a, const Def b)

Detailed Description

Base class for all Defs.

These are the most important subclasses:

Type Formation	Term Introduction	Term Elimination
Pi	Lam	App
Sigma / Arr	Tuple / Pack	Extract
	Insert	Insert
Uniq
Join	Inj	Match
Meet	Merge	Split
Reform	Rule
Nat	Lit
Idx	Lit

In addition there is:

Var: A variable. Currently the following Defs may be binders:
- Pi, Lam, Sigma, Arr, Pack
Axm: To introduce new entities.
Proxy: Used for intermediate values during optimizations.
Hole: A metavariable filled in by the type inference (always mutable as holes are filled in later).
Type, Univ, UMax, UInc: To keep track of type levels.

The data layout (see World::alloc and Def::deps) looks like this:

Def| type | op(0) ... op(num_ops-1) |
          |-----------ops-----------|
                 deps
   |--------------------------------| if type() != nullptr &&  is_set()
          |-------------------------| if type() == nullptr &&  is_set()
   |------|                           if type() != nullptr && !is_set()
   ||                                 if type() == nullptr && !is_set()

Attention: This means that any subclass of Def must not introduce additional members.

See also: Immutables vs. Mutables

Definition at line 273 of file def.h.

Member Enumeration Documentation

◆ Cmp

enum class mim::Def::Cmp

strong

Enumerator
L	Less.
G	Greater.
E	Equal.
U	Unknown.

Definition at line 712 of file def.h.

Constructor & Destructor Documentation

◆ Def() [1/4]

mim::Def::Def	(	World *	world,
		Node	node,
		const Def *	type,
		Defs	ops,
		flags_t	flags )

protected

Constructor for an immutable Def.

Definition at line 42 of file def.cpp.

References flags(), flags_, gid(), local_muts(), local_vars(), mim::World::muts(), mim::World::next_gid(), node(), mim::node2dep(), op(), ops(), mim::Proxy, type(), mim::Univ, vars(), mim::World::vars(), World, and world().

◆ Def() [2/4]

mim::Def::Def	(	Node	n,
		const Def *	type,
		Defs	ops,
		flags_t	flags )

protected

As above but World retrieved from type.

Definition at line 94 of file def.cpp.

References flags(), ops(), and type().

◆ Def() [3/4]

mim::Def::Def	(	Node	node,
		const Def *	type,
		size_t	num_ops,
		flags_t	flags )

protected

Constructor for a mutable Def.

Definition at line 97 of file def.cpp.

References flags(), flags_, gid(), mim::World::next_gid(), node(), mim::node2dep(), num_ops(), type(), and world().

◆ Def() [4/4]

mim::Def::Def	(	Node	node,
		Def *	binder )

protected

Constructor for a Var; stores its binder.

Definition at line 113 of file def.cpp.

References flags_, gid(), node(), mim::node2dep(), mim::Var, and world().

Member Function Documentation

◆ arity()

const Def * mim::Def::arity ( ) const

Number of elements available to Extract / Insert (may be dynamic).

Definition at line 592 of file def.cpp.

References arity(), mim::Arr, isa_mut(), mim::World::lit_nat(), mim::World::lit_nat_0(), mim::World::lit_nat_1(), node(), num_ops(), op(), mim::Pack, mim::Sigma, mim::World::sigma(), type(), and world().

Referenced by arity(), mim::plug::autodiff::phase::Eval::augment_pack(), mim::Dispatch::Dispatch(), mim::World::ext(), mim::World::extract(), mim::World::insert(), mim::isa_dim(), mim::plug::regex::normalize_conj(), mim::plug::regex::normalize_disj(), mim::plug::tensor::normalize_fastest_axis(), mim::plug::mem::normalize_lea(), num_projs(), num_tprojs(), mim::Rewriter::rewrite_imm_Seq(), mim::Rewriter::rewrite_mut_Seq(), mim::Select::Select(), mim::Arr::set(), mim::Seq::set(), mim::Arr::set_arity(), mim::tuple2str(), and var_type().

◆ as_imm()

template<class T = Def>

const T * mim::Def::as_imm ( ) const

inline

Definition at line 575 of file def.h.

References as_mut().

◆ as_mut()

template<class T = Def, bool invert = false>

T * mim::Def::as_mut ( ) const

inline

Asserts that this is a mutable, casts constness away and performs a static_cast to T.

Definition at line 589 of file def.h.

Referenced by as_imm(), mim::plug::clos::clos_type(), mim::plug::autodiff::phase::Eval::derive_(), mim::ast::TuplePtrn::emit_body(), mim::sexpr::Emitter::finalize(), mim::plug::tensor::is_copy_comb(), mim::plug::tensor::is_identity_post(), mim::RWPhase::rewrite_external(), mim::LamSpec::rewrite_imm_App(), mim::plug::affine::phase::LowerIndex::rewrite_imm_App(), mim::plug::mem::phase::SEO::rewrite_imm_App(), and mim::plug::gpu::phase::SplitOffKernels::rewrite_mut_Lam().

◆ blame()

template<class... Args>

fe::Error & mim::Def::blame	(	fe::cite_string< Args... >	s,
		Args &&...	args ) const

inline

Reports an error that blames this; chain Error::n for Notes and Error::bail to throw.

Definition at line 310 of file def.h.

References blame(), err_loc(), and error().

Referenced by mim::World::app(), blame(), check(), mim::World::extract(), mim::World::insert(), mim::World::match(), mim::plug::refly::normalize_equiv(), mim::plug::regex::normalize_not(), mim::World::seq(), mim::World::type(), and mim::World::uniq().

◆ check() [1/2]

const Def * mim::Def::check ( )

After all Def::ops have been Def::set, this method will be invoked to check the type of this mutable.

The method returns a possibly updated version of its type (e.g. where Holes have been resolved). If different from Def::type, it will update its Def::type to a Def::zonked version of that.

Definition at line 454 of file check.cpp.

References mim::Checker::alpha(), mim::Arr, mim::Checker::assignable(), mim::Pi::infer(), mim::Reform::infer(), mim::Sigma::infer(), node(), ops(), mim::Pi, mim::Reform, mim::Rule, mim::Sigma, type(), mim::type_of(), and world().

◆ check() [2/2]

const Def * mim::Def::check	(	size_t	i,
		const Def *	def )

Checks whether the ith operand can be set to def.

The method returns a possibly updated version of def (e.g. where Holes have been resolved). This is the actual def that will be set as the ith operand.

Definition at line 438 of file check.cpp.

References mim::Checker::assignable(), blame(), mim::type_of(), and world().

Referenced by set(), and set().

◆ cmp()

Def::Cmp mim::Def::cmp	(	const Def *	a,
		const Def *	b )

staticnodiscard

Definition at line 502 of file def.cpp.

References mim::Var::binder(), cmp(), E, flags(), free_vars(), G, is_mutable(), is_set(), L, node(), num_ops(), op(), type(), and U.

Referenced by cmp().

◆ dbg()

Dbg mim::Def::dbg ( ) const

Looks up Def::dbg_ in Driver::dbg.

Definition at line 430 of file def.cpp.

References mim::World::driver(), and world().

Referenced by debug_prefix(), debug_suffix(), loc(), set(), set(), set(), set(), set_dbg_(), set_dbg_key_(), and sym().

◆ dbg_key()

DbgKey mim::Def::dbg_key ( ) const

inline

Cheap handle for other->set(this->dbg_key()).

Definition at line 610 of file def.h.

Referenced by mim::plug::mem::phase::mk_phi(), mim::Rewriter::rewrite(), mim::VarRewriter::rewrite(), mim::plug::mem::phase::AddMem::rewrite_mut_Lam(), mim::Scalarize::rewrite_mut_Lam(), and mim::StaticArgOpt::rewrite_mut_Lam().

◆ debug_prefix()

const Def * mim::Def::debug_prefix ( std::string prefix ) const

Definition at line 494 of file def.cpp.

References dbg(), set(), set_dbg(), and sym().

◆ debug_suffix()

const Def * mim::Def::debug_suffix ( std::string suffix ) const

Definition at line 495 of file def.cpp.

References dbg(), set(), set_dbg(), and sym().

◆ dep() [1/2]

Dep mim::Def::dep ( ) const

inlinenoexcept

Definition at line 407 of file def.h.

◆ dep() [2/2]

const Def * mim::Def::dep ( size_t i ) const

inlinenoexcept

Definition at line 393 of file def.h.

References dep(), and deps().

Referenced by dep(), err_loc(), and has_dep().

◆ deps()

Defs mim::Def::deps ( ) const

noexcept

Definition at line 468 of file def.cpp.

References set().

Referenced by mim::collect(), dep(), mim::Scheduler::early(), err_loc(), mim::plug::ll_nvptx::HostEmitter::find_kernels(), is_immutabilizable(), mim::sexpr::Emitter::next_lams(), num_deps(), mim::post_order(), and zonk_mut().

◆ dirty()

void mim::Def::dirty ( bool dirty = true )

inlinenoexcept

Definition at line 565 of file def.h.

References dirty().

Referenced by dirty().

◆ dot() [1/3]

void mim::Def::dot	(	const char *	file = nullptr,
		DotConfig	cfg = {} ) const

Same as above but write to file or std::cout if file is nullptr.

Definition at line 205 of file dot.cpp.

References dot().

◆ dot() [2/3]

void mim::Def::dot	(	const std::string &	file,
		DotConfig	cfg = {} ) const

inline

Definition at line 729 of file def.h.

◆ dot() [3/3]

void mim::Def::dot	(	std::ostream &	os,
		DotConfig	cfg = {} ) const

Definition at line 203 of file dot.cpp.

References mim::DotConfig::max.

Referenced by dot().

◆ driver()

Driver & mim::Def::driver ( ) const

noexcept

Definition at line 399 of file def.cpp.

References mim::World::driver(), and world().

Referenced by error(), and set_dbg_key_().

◆ dump() [1/2]

void mim::Def::dump ( ) const

Definition at line 577 of file dump.cpp.

◆ dump() [2/2]

void mim::Def::dump ( int max ) const

Definition at line 578 of file dump.cpp.

References stream().

◆ err_loc()

Loc mim::Def::err_loc ( ) const

Returns a blame Loc from World::get_loc, this Def, or its nearest located dependency, in that order.

Definition at line 402 of file def.cpp.

References dep(), deps(), loc(), and world().

Referenced by blame().

◆ error()

fe::Error & mim::Def::error ( ) const

noexcept

Definition at line 400 of file def.cpp.

References driver().

Referenced by blame().

◆ expect_mut()

template<class T = Def, class... Args>

T * mim::Def::expect_mut	(	std::format_string< Args... >	fmt,
		Args &&...	args ) const

inline

Like Def::as_mut but - instead of merely asserting in Debug builds - throws via fe::throwf when the cast fails; the mutable counterpart of fe::RuntimeCast::expect (which Def inherits for the general case).

fmt / args describe what was expected; a plain string works, as does a format string plus arguments.

Definition at line 601 of file def.h.

References isa_mut().

◆ externalize()

void mim::Def::externalize ( )

Definition at line 607 of file def.cpp.

References mim::World::Externals::externalize(), mim::World::externals(), and world().

Referenced by mim::plug::gpu::phase::SplitOffKernels::rewrite_mut_Lam(), and transfer_external().

◆ flags()

flags_t mim::Def::flags ( ) const

inlineconstexprnoexcept

Definition at line 293 of file def.h.

References flags_.

Referenced by mim::Axm::base(), cmp(), Def(), Def(), Def(), mim::Axm::IsA< mem::M, mim::App >::id(), mim::Pi::is_implicit(), mim::Global::is_mutable(), mim::UInc::offset(), mim::Axm::plugin(), mim::Axm::sub(), and mim::Axm::tag().

◆ free_vars() [1/2]

Vars mim::Def::free_vars ( )

As above but drives (and caches) the fixed-point iteration for mutables.

Definition at line 285 of file def.cpp.

◆ free_vars() [2/2]

Vars mim::Def::free_vars ( ) const

Global set of free Vars: extends local_vars() by transitively following mutables as well.

Note: On a mutable this simply forwards to the caching non-const overload below.

Definition at line 249 of file def.cpp.

References isa_mut(), local_muts(), local_vars(), vars(), mim::World::vars(), and world().

Referenced by cmp(), mim::Prod::Def(), mim::Rule::is_in_rule(), nests(), and outermost_binder().

◆ gid()

u32 mim::Def::gid ( ) const

inlineconstexprnoexcept

Global id - unique number for this Def.

Definition at line 294 of file def.h.

Referenced by mim::plug::regex::compare_re(), Def(), Def(), Def(), mim::plug::refly::normalize_gid(), mim::UseHash::operator()(), and unique_name().

◆ greater()

bool mim::Def::greater	(	const Def *	a,
		const Def *	b )

staticnodiscard

Definition at line 543 of file def.cpp.

Referenced by mim::plug::core::normalize_bit2(), mim::plug::core::normalize_nat(), and mim::plug::core::normalize_ncmp().

◆ has_dep() [1/2]

bool mim::Def::has_dep ( ) const

inlinenoexcept

Definition at line 408 of file def.h.

Referenced by needs_zonk().

◆ has_dep() [2/2]

bool mim::Def::has_dep ( Dep d ) const

inlinenoexcept

Definition at line 409 of file def.h.

References dep().

◆ has_free_var()

bool mim::Def::has_free_var ( const Var * var ) const

Same as free_vars().contains(var).

Definition at line 273 of file def.cpp.

References mim::any_free_vars(), and var().

Referenced by mim::Lam::eta_reduce(), and is_immutabilizable().

◆ has_free_vars()

bool mim::Def::has_free_vars ( ) const

Same as !free_vars().empty().

Definition at line 277 of file def.cpp.

References mim::any_free_vars().

Referenced by is_closed(), and is_open().

◆ has_free_vars_in()

bool mim::Def::has_free_vars_in ( Vars vars ) const

Same as vars.has_intersection(free_vars()).

Definition at line 281 of file def.cpp.

References vars().

Referenced by mim::Nest::contains(), and mim::plug::cps::Conv::map().

◆ has_var() [1/2]

const Var * mim::Def::has_var ( )

inline

Only returns not nullptr, if Var of this mutable has ever been created.

Definition at line 483 of file def.h.

References has_var().

Referenced by mim::World::app(), mim::Lam::eta_reduce(), mim::World::extract(), has_var(), is_immutabilizable(), isa_binder(), nests(), nests(), mim::VarRewriter::rewrite_mut(), and mim::Rewriter::rewrite_mut_Seq().

◆ has_var() [2/2]

const Var * mim::Def::has_var ( ) const

inline

As above if this is a mutable.

Definition at line 485 of file def.h.

◆ hash()

size_t mim::Def::hash ( ) const

inlineconstexprnoexcept

Definition at line 296 of file def.h.

◆ immutabilize()

const Def * mim::Def::immutabilize ( )

Tries to make an immutable from a mutable. This usually works if the mutable isn't recursive and its var isn't used.

Definition at line 555 of file def.cpp.

References mim::Arr, is_immutabilizable(), mim::Lit::isa(), node(), ops(), mim::Pack, mim::Pi, mim::Rule, mim::Sigma, and world().

Referenced by mim::ast::TuplePtrn::emit_body(), mim::Zonker::rewire_mut(), and mim::Rewriter::rewrite_stub().

◆ internalize()

void mim::Def::internalize ( )

Definition at line 608 of file def.cpp.

References mim::World::externals(), mim::World::Externals::internalize(), and world().

Referenced by mim::optimize(), mim::InplaceRWPhase::rewrite_external(), and transfer_external().

◆ is_annex()

bool mim::Def::is_annex ( ) const

inlinenoexcept

Definition at line 557 of file def.h.

Referenced by mim::plug::cps::convertible(), and mim::isa_optimizable().

◆ is_closed()

bool mim::Def::is_closed ( ) const

Same as !has_free_vars().

Definition at line 353 of file def.cpp.

References has_free_vars(), and is_external().

Referenced by mim::Scheduler::early(), mim::World::Externals::externalize(), mim::World::for_each(), mim::Scheduler::late(), mim::plug::vec::normalize_diff(), and mim::plug::core::normalize_pe().

◆ is_dirty()

bool mim::Def::is_dirty ( ) const

inlinenoexcept

Definition at line 564 of file def.h.

◆ is_elim()

bool mim::Def::is_elim ( ) const

inlinenoexcept

Definition at line 322 of file def.h.

References mim::Elim, and judge().

◆ is_external()

bool mim::Def::is_external ( ) const

inlinenoexcept

Definition at line 553 of file def.h.

Referenced by mim::plug::cps::convertible(), mim::sexpr::Emitter::emit_head(), mim::sexpr::Emitter::emit_imported(), mim::World::Externals::externalize(), mim::World::Externals::internalize(), is_closed(), mim::isa_optimizable(), mim::plug::ll::Emitter::prepare(), mim::plug::clos::phase::ClosConv::rewrite_external(), mim::plug::clos::phase::LowerTypedClos::rewrite_external(), mim::RWPhase::rewrite_external(), and mim::plug::gpu::phase::RemoveDoubleSyncs::rewrite_imm_App().

◆ is_form()

bool mim::Def::is_form ( ) const

inlinenoexcept

Definition at line 320 of file def.h.

References mim::Form, is_form(), and judge().

Referenced by is_form().

◆ is_ground()

bool mim::Def::is_ground ( ) const

inline

Immutable that contains neither mutables nor Vars.

Definition at line 525 of file def.h.

References local_muts(), and local_vars().

Referenced by mim::EtaConv::rewrite().

◆ is_immutabilizable()

bool mim::Def::is_immutabilizable ( )

Definition at line 140 of file def.cpp.

References deps(), has_free_var(), has_var(), is_set(), local_muts(), and op().

Referenced by immutabilize(), mim::Rewriter::rewrite_mut_Seq(), and mim::Rewriter::rewrite_stub().

◆ is_intro()

bool mim::Def::is_intro ( ) const

inlinenoexcept

Definition at line 321 of file def.h.

References mim::Intro, and judge().

Referenced by mim::Rewriter::rewrite_imm_Seq(), and mim::Rewriter::rewrite_mut_Seq().

◆ is_meta()

bool mim::Def::is_meta ( ) const

inlinenoexcept

Definition at line 323 of file def.h.

References judge(), and mim::Meta.

◆ is_mutable()

bool mim::Def::is_mutable ( ) const

inlinenoexcept

Definition at line 571 of file def.h.

Referenced by cmp().

◆ is_open()

bool mim::Def::is_open ( ) const

inline

Same as has_free_vars().

Definition at line 521 of file def.h.

References has_free_vars().

Referenced by mim::plug::vec::normalize_zip().

◆ is_set()

bool mim::Def::is_set ( ) const

inline

Yields true if empty or the last op is set.

Definition at line 370 of file def.h.

References num_ops(), op(), and ops().

Referenced by mim::World::app(), cmp(), mim::plug::cps::convertible(), mim::World::for_each(), is_immutabilizable(), mim::isa_optimizable(), mim::Hole::isa_set(), mim::BetaRed::rewrite_imm_App(), mim::InplaceRWPhase::rewrite_mut(), mim::BranchNormalize::rewrite_mut_Lam(), mim::plug::cps::Conv::rewrite_mut_Lam(), mim::plug::mem::phase::AddMem::rewrite_mut_Lam(), mim::Rewriter::rewrite_mut_Seq(), mim::Rewriter::rewrite_stub(), mim::Scheduler::Scheduler(), mim::Hole::tuplefy(), and zonk_mut().

◆ is_term()

bool mim::Def::is_term ( ) const

Is this Def a term, i.e. is its type() a Type?

Definition at line 486 of file def.cpp.

References mim::Lit::isa(), and type().

◆ isa_binder()

template<class D = Def>

std::pair< D *, const Var * > mim::Def::isa_binder ( ) const

inline

Is this a mutable that introduces a Var?

Returns: {nullptr, nullptr} otherwise.

Definition at line 490 of file def.h.

References has_var(), isa_mut(), and var().

Referenced by mim::Hole::tuplefy().

◆ isa_imm()

template<class T = Def>

const T * mim::Def::isa_imm ( ) const

inline

Definition at line 574 of file def.h.

References isa_mut().

Referenced by mim::World::app(), mim::sexpr::Emitter::emit_bb(), mim::sexpr::Emitter::emit_decl(), mim::World::extract(), mim::isa_flex_rank(), and mim::plug::mem::phase::AddMem::rewrite().

◆ isa_mut()

template<class T = Def, bool invert = false>

T * mim::Def::isa_mut ( ) const

inline

If this is mutable, it will cast constness away and perform a dynamic_cast to T.

Definition at line 580 of file def.h.

Referenced by mim::plug::clos::phase::LowerTypedClosPrep::analyze(), mim::any_free_vars(), mim::World::app(), arity(), mim::plug::autodiff::phase::Eval::augment_(), mim::collect(), mim::sexpr::Emitter::emit_bb(), mim::sexpr::Emitter::emit_decl(), expect_mut(), mim::World::extract(), mim::Bound::find(), mim::Hole::find(), free_vars(), mim::plug::ll::Emitter::id(), mim::plug::mem::phase::is_dependent(), isa_binder(), mim::plug::clos::isa_clos_type(), isa_imm(), mim::plug::ll_nvptx::HostEmitter::isa_targetspecific_intrinsic(), mim::Hole::isa_unset(), mim::Scheduler::late(), local_muts(), nests(), outermost_binder(), proj(), mim::Analysis::rewrite(), mim::plug::clos::phase::LowerTypedClos::rewrite(), mim::plug::mem::phase::AddMem::rewrite(), mim::Rewriter::rewrite(), mim::VarRewriter::rewrite(), mim::Zonker::rewrite(), mim::BetaRed::rewrite_imm_App(), mim::LamSpec::rewrite_imm_App(), mim::plug::affine::phase::LowerFor::rewrite_imm_App(), mim::plug::mem::phase::SEO::rewrite_imm_App(), mim::plug::tensor::phase::LowerToMem::rewrite_imm_App(), mim::Scalarize::rewrite_imm_App(), mim::StaticArgOpt::rewrite_imm_App(), mim::plug::mem::phase::SEO::rewrite_imm_Var(), mim::Scheduler::Scheduler(), zonk(), and zonk_mut().

◆ isa_type()

template<class T>

const T * mim::Def::isa_type ( ) const

inline

Is Def::unfold_type a T? Yields nullptr for Univ, which has no type at all.

Definition at line 337 of file def.h.

References isa_type(), and unfold_type().

Referenced by mim::World::app(), isa_type(), mim::World::match(), and mim::World::type().

◆ judge()

Judge mim::Def::judge ( ) const

inlinenoexcept

Def must never become polymorphic: a vptr costs 8 bytes on every node in the World, and Def::ops_ptr hands out the operands at this + 1, so the vptr would also shift them.

Def carries its own Def::node() tag and dispatches on it instead - see the dispatch section in def.cpp.

Note: A subclass growing a virtual is caught by the sizeof(Def) == sizeof(T) assert in World::allocate. Table-driven and defined here instead of in def.cpp, so that Def::is_form & friends stay inlinable: libmim is a shared object, so an out-of-line judge() would be an opaque PLT call from every other TU.

Definition at line 811 of file def.h.

References CODE, MIM_NODE, and mim::Num_Nodes.

Referenced by is_elim(), is_form(), is_intro(), and is_meta().

◆ less()

bool mim::Def::less	(	const Def *	a,
		const Def *	b )

staticnodiscard

Definition at line 542 of file def.cpp.

◆ loc()

Loc mim::Def::loc ( ) const

inline

Definition at line 611 of file def.h.

References dbg().

Referenced by mim::World::app(), and err_loc().

◆ local_muts()

Muts mim::Def::local_muts ( ) const

inline

Mutables reachable by following immutable deps(); mut->local_muts() is by definition the set { mut }.

Definition at line 507 of file def.h.

References isa_mut().

Referenced by mim::any_free_vars(), Def(), free_vars(), is_ground(), is_immutabilizable(), needs_zonk(), and mim::VarRewriter::rewrite().

◆ local_vars()

Vars mim::Def::local_vars ( ) const

inline

Vars reachable by following immutable deps().

Note: var->local_vars() is by definition the set { var }.

Definition at line 514 of file def.h.

Referenced by mim::any_free_vars(), Def(), free_vars(), is_ground(), and mim::VarRewriter::rewrite().

◆ mark()

u32 mim::Def::mark ( ) const

inlineconstexprnoexcept

Used internally by free_vars().

Definition at line 295 of file def.h.

◆ needs_zonk()

bool mim::Def::needs_zonk ( ) const

Yields true, if Def::local_muts() contain a Hole that is set.

Rewriting (Def::zonking) will resolve the Hole to its operand.

Definition at line 12 of file check.cpp.

References has_dep(), mim::Hole, mim::Hole::isa_set(), and local_muts().

Referenced by mim::Zonker::rewrite(), and zonk().

◆ nests() [1/2]

bool mim::Def::nests ( const Def * def )

Does this nest def?

Also strict: a def that only uses this's own Var sits at this's level and is not nested.

Definition at line 384 of file def.cpp.

References free_vars(), has_var(), isa_mut(), and nests().

◆ nests() [2/2]

bool mim::Def::nests ( Def * mut )

Does this nest mut?

The relation is strict: f->nests(f) is false.

Definition at line 378 of file def.cpp.

References has_var(), and nests().

Referenced by nests(), and nests().

◆ node()

Node mim::Def::node ( ) const

inlineconstexprnoexcept

Definition at line 297 of file def.h.

Referenced by arity(), mim::Bound::Bound(), check(), cmp(), Def(), Def(), Def(), mim::Ext::Ext(), immutabilize(), mim::is_flex(), mim::Pi::isa_cn(), node_name(), reduction_offset(), mim::plug::clos::phase::LowerTypedClos::rewrite_imm(), mim::Rewriter::rewrite_imm(), mim::Rewriter::rewrite_mut(), mim::Seq::set(), and var_type().

◆ node_name()

std::string_view mim::Def::node_name ( ) const

Definition at line 459 of file def.cpp.

References CODE, MIM_NODE, and node().

Referenced by mim::plug::autodiff::phase::Eval::augment_().

◆ num_deps()

size_t mim::Def::num_deps ( ) const

inlinenoexcept

Definition at line 394 of file def.h.

References deps().

◆ num_ops()

size_t mim::Def::num_ops ( ) const

inlineconstexprnoexcept

Definition at line 352 of file def.h.

Referenced by arity(), cmp(), mim::plug::core::convert(), Def(), mim::World::extract(), mim::plug::mem::has_leading_mem(), is_set(), mim::World::match(), mim::plug::mem::normalize_load(), mim::plug::ll::Emitter::prepare(), mim::World::reduce(), mim::Zonker::rewire_mut(), mim::Rewriter::rewrite_stub(), mim::Scheduler::Scheduler(), set(), set(), and unset().

◆ num_projs()

nat_t mim::Def::num_projs ( ) const

inline

Yields Def::arity(), if it is a Lit, or 1 otherwise.

Definition at line 1132 of file def.h.

References arity(), and mim::Lit::isa().

Referenced by mim::plug::clos::clos_insert_env(), mim::plug::clos::clos_remove_env(), mim::plug::clos::clos_sub_env(), mim::plug::mem::mem_def(), proj(), projs(), and mim::plug::clos::phase::ClosConvPrep::rewrite_imm_App().

◆ num_tprojs()

nat_t mim::Def::num_tprojs ( ) const

As above but yields 1, if Flags::scalarize_threshold is exceeded.

Definition at line 618 of file def.cpp.

References arity(), mim::World::flags(), mim::Lit::isa(), mim::Flags::scalarize_threshold, and world().

Referenced by tproj(), and tprojs().

◆ num_tvars()

nat_t mim::Def::num_tvars ( )

inlinenoexcept

Definition at line 479 of file def.h.

References MIM_PROJ, var(), and var_type().

Referenced by mim::plug::ll::Emitter::emit_phi_args(), mim::plug::mem::phase::SEO::rewrite_imm_App(), and mim::Scalarize::rewrite_mut_Lam().

◆ num_vars()

nat_t mim::Def::num_vars ( )

inlinenoexcept

Definition at line 479 of file def.h.

Referenced by mim::Lam::ret_var(), mim::plug::affine::phase::LowerIndex::rewrite_imm_App(), and mim::plug::mem::phase::AddMem::rewrite_mut_Lam().

◆ op()

const Def * mim::Def::op ( size_t i ) const

inlinenoexcept

Definition at line 351 of file def.h.

References ops().

Referenced by mim::App::arg(), arity(), mim::Lam::body(), mim::App::callee(), mim::plug::clos::clos_type_to_pi(), cmp(), mim::Pi::codom(), Def(), mim::Pi::dom(), mim::Reform::dom(), mim::World::extract(), mim::Lam::filter(), mim::Hole::find(), mim::Bound::get(), mim::Rule::guard(), mim::Extract::index(), mim::Insert::index(), mim::Sigma::infer(), mim::Tuple::infer(), mim::Global::init(), is_immutabilizable(), is_set(), mim::plug::clos::isa_clos_type(), mim::Hole::isa_set(), mim::Type::level(), mim::Rule::lhs(), mim::World::match(), mim::plug::mem::normalize_load(), mim::plug::option::normalize_unwrap_unsafe(), mim::Hole::op(), mim::UInc::op(), mim::Uniq::op(), proj(), mim::Rewriter::rewrite_stub(), mim::Rule::rhs(), mim::Scheduler::Scheduler(), mim::Match::scrutinee(), set(), mim::Extract::tuple(), mim::Insert::tuple(), mim::Inj::value(), mim::Insert::value(), mim::Split::value(), and zonk_mut().

◆ ops()

template<size_t N = std::dynamic_extent>

auto mim::Def::ops ( ) const

inlineconstexprnoexcept

◆ outermost_binder()

Def * mim::Def::outermost_binder ( ) const

Transitively walks up free_vars() till the outermoust binder has been found.

Returns: nullptr, if is_closed() and not a mutable.

Definition at line 359 of file def.cpp.

References free_vars(), and isa_mut().

◆ proj() [1/2]

const Def * mim::Def::proj	(	nat_t	a,
		nat_t	i ) const

Similar to World::extract while assuming an arity of a, but also works on Sigmas and Arrays.

Definition at line 623 of file def.cpp.

References mim::World::extract(), isa_mut(), op(), type(), and world().

◆ proj() [2/2]

const Def * mim::Def::proj ( nat_t i ) const

inline

As above but takes Def::num_projs as arity.

Definition at line 434 of file def.h.

References num_projs(), and proj().

Referenced by proj().

◆ projs() [1/4]

template<nat_t A = std::dynamic_extent>

auto mim::Def::projs ( ) const

inline

Definition at line 463 of file def.h.

References projs().

◆ projs() [2/4]

template<nat_t A = std::dynamic_extent, class F>

auto mim::Def::projs ( F f ) const

inline

Splits this Def via Def::projections into an Array (if A == std::dynamic_extent) or std::array (otherwise).

Applies f to each element.

Definition at line 440 of file def.h.

References num_projs(), proj(), and projs().

Referenced by mim::plug::autodiff::phase::Eval::augment_tuple(), mim::sexpr::Emitter::emit_var(), mim::plug::tensor::is_pure_read(), mim::plug::mem::mem_def(), mim::plug::core::normalize_abs(), mim::plug::autodiff::normalize_add(), mim::plug::math::normalize_arith(), mim::plug::core::normalize_bit2(), mim::plug::tensor::normalize_broadcast(), mim::plug::buffer::normalize_Buf(), mim::plug::tuple::normalize_cat(), mim::plug::vec::normalize_cat(), mim::plug::refly::normalize_check(), mim::plug::math::normalize_cmp(), mim::plug::ord::normalize_contains(), mim::plug::tuple::normalize_contains(), mim::plug::refly::normalize_dbg(), mim::plug::vec::normalize_diff(), mim::plug::core::normalize_div(), mim::plug::refly::normalize_equiv(), mim::plug::core::normalize_extrema(), mim::plug::math::normalize_extrema(), mim::plug::tensor::normalize_fastest_axis(), mim::plug::vec::normalize_fold(), mim::plug::ord::normalize_get(), mim::plug::tensor::normalize_get(), mim::plug::core::normalize_icmp(), mim::plug::tensor::normalize_if_static(), mim::plug::ord::normalize_insert(), mim::plug::mem::normalize_lea(), mim::plug::mem::normalize_load(), mim::plug::btensor::normalize_map_reduce(), mim::plug::core::normalize_nat(), mim::plug::core::normalize_ncmp(), mim::plug::math::normalize_pow(), mim::plug::regex::normalize_range(), mim::plug::buffer::normalize_read(), mim::plug::tensor::normalize_set(), mim::plug::buffer::normalize_shape(), mim::plug::core::normalize_shr(), mim::plug::mem::normalize_store(), mim::plug::autodiff::normalize_sum(), mim::plug::core::normalize_wrap(), mim::plug::tuple::normalize_zip(), projs(), projs(), projs(), mim::plug::clos::phase::LowerTypedClos::rewrite(), mim::plug::affine::phase::LowerIndex::rewrite_imm_App(), mim::plug::buffer::LowerPtr::rewrite_imm_App(), and tprojs().

◆ projs() [3/4]

auto mim::Def::projs ( nat_t a ) const

inline

Definition at line 469 of file def.h.

References projs().

◆ projs() [4/4]

template<class F>

auto mim::Def::projs	(	nat_t	a,
		F	f ) const

inline

Definition at line 458 of file def.h.

References proj().

◆ reduce()

template<size_t N = std::dynamic_extent>

auto mim::Def::reduce ( const Def * arg ) const

inlineconstexpr

See also: World::reduce

Definition at line 660 of file def.h.

Referenced by mim::World::extract(), mim::Lam::reduce(), mim::Pi::reduce(), mim::Rule::reduce(), and mim::Seq::reduce().

◆ reduction_offset()

size_t mim::Def::reduction_offset ( ) const

noexcept

First Def::op that needs to be dealt with during reduction; e.g.

for a Pi we don't reduce the Pi::dom.

See also: World::reduce

Definition at line 578 of file def.cpp.

References mim::Arr, mim::Join, mim::Lam, mim::Meet, node(), mim::Pack, mim::Pi, mim::Rule, and mim::Sigma.

Referenced by mim::World::reduce().

◆ set() [1/16]

template<bool Ow = false>

Def * mim::Def::set ( Dbg d )

inline

Definition at line 631 of file def.h.

References set_dbg_().

◆ set() [2/16]

template<bool Ow = false>

const Def * mim::Def::set ( Dbg d ) const

inline

Definition at line 630 of file def.h.

References set_dbg_().

◆ set() [3/16]

template<bool Ow = false>

Def * mim::Def::set ( DbgKey key )

inline

Definition at line 635 of file def.h.

References set_dbg_key_().

◆ set() [4/16]

template<bool Ow = false>

const Def * mim::Def::set ( DbgKey key ) const

inline

Adopts the Dbg behind key - just copies the interned index, so nothing is re-interned.

Prefer a->set(b->dbg_key()) over a->set(b->dbg()).

Definition at line 634 of file def.h.

References set_dbg_key_().

◆ set() [5/16]

Def * mim::Def::set ( Defs ops )

Set ops all at once (no Def::unset necessary beforehand).

Definition at line 177 of file def.cpp.

References check(), num_ops(), and ops().

◆ set() [6/16]

template<bool Ow = false>

Def * mim::Def::set ( Loc l )

inline

Definition at line 621 of file def.h.

References dbg(), and set_dbg().

◆ set() [7/16]

template<bool Ow = false>

const Def * mim::Def::set ( Loc l ) const

inline

Definition at line 620 of file def.h.

References dbg(), and set_dbg().

◆ set() [8/16]

template<bool Ow = false>

Def * mim::Def::set	(	Loc	l,
		std::string	s )

inline

Definition at line 629 of file def.h.

References set(), and sym().

◆ set() [9/16]

template<bool Ow = false>

const Def * mim::Def::set	(	Loc	l,
		std::string	s ) const

inline

Definition at line 628 of file def.h.

References set(), and sym().

◆ set() [10/16]

template<bool Ow = false>

Def * mim::Def::set	(	Loc	l,
		Sym	s )

inline

Definition at line 627 of file def.h.

References set().

◆ set() [11/16]

template<bool Ow = false>

const Def * mim::Def::set	(	Loc	l,
		Sym	s ) const

inline

Definition at line 626 of file def.h.

References set().

◆ set() [12/16]

Def * mim::Def::set	(	size_t	i,
		const Def *	def )

Successively set from left to right.

Definition at line 196 of file def.cpp.

References check(), num_ops(), and op().

Referenced by mim::Lam::app(), debug_prefix(), debug_suffix(), deps(), mim::plug::autodiff::phase::Eval::derive_(), mim::ast::MatchExpr::Arm::emit(), mim::ast::AliasPtrn::emit_value(), mim::ast::GrpPtrn::emit_value(), mim::ast::IdPtrn::emit_value(), mim::Zonker::rewire_mut(), mim::VarRewriter::rewrite(), mim::LamSpec::rewrite_imm_App(), mim::InplaceRWPhase::rewrite_mut(), mim::plug::mem::phase::AddMem::rewrite_mut_Lam(), mim::Rewriter::rewrite_stub(), set(), set(), set(), set(), set(), set(), mim::Global::set(), mim::Hole::set(), mim::Lam::set(), mim::Lam::set(), mim::Pack::set(), mim::Pi::set(), mim::Reform::set(), mim::Reform::set(), mim::Rule::set(), mim::Seq::set(), mim::Sigma::set(), mim::Sigma::set(), mim::Arr::set_arity(), mim::Arr::set_body(), mim::Lam::set_body(), mim::Pi::set_codom(), mim::Pi::set_dom(), mim::Pi::set_dom(), mim::Lam::set_filter(), mim::Rule::set_guard(), mim::Rule::set_lhs(), and mim::Rule::set_rhs().

◆ set() [13/16]

template<bool Ow = false>

Def * mim::Def::set ( std::string s )

inline

Definition at line 625 of file def.h.

References set(), and sym().

◆ set() [14/16]

template<bool Ow = false>

const Def * mim::Def::set ( std::string s ) const

inline

Definition at line 624 of file def.h.

References set(), and sym().

◆ set() [15/16]

template<bool Ow = false>

Def * mim::Def::set ( Sym s )

inline

Definition at line 623 of file def.h.

References dbg(), and set_dbg().

◆ set() [16/16]

template<bool Ow = false>

const Def * mim::Def::set ( Sym s ) const

inline

Definition at line 622 of file def.h.

References dbg(), and set_dbg().

◆ set_dbg()

void mim::Def::set_dbg ( Dbg d ) const

protected

Interns dbg via Driver::dbg and stores the key in Def::dbg_.

Definition at line 431 of file def.cpp.

References mim::World::driver(), and world().

Referenced by debug_prefix(), debug_suffix(), set(), set(), set(), set(), and set_dbg_().

◆ set_dbg_()

void mim::Def::set_dbg_	(	Dbg	d,
		bool	ow ) const

protected

Backs Def::set(Dbg).

Definition at line 435 of file def.cpp.

References dbg(), and set_dbg().

Referenced by set(), set(), and set_dbg_key_().

◆ set_dbg_key_()

void mim::Def::set_dbg_key_	(	DbgKey	key,
		bool	ow ) const

protected

Backs Def::set(DbgKey).

Definition at line 444 of file def.cpp.

References dbg(), driver(), set_dbg_(), and world().

Referenced by set(), and set().

◆ set_type()

Def * mim::Def::set_type ( const Def * type )

Update type.

Warning: Only make type-preserving updates such as removing Holes. Do this even before updating all other ops()!

Definition at line 207 of file def.cpp.

References type().

Referenced by mim::Zonker::rewire_mut().

◆ stream()

std::ostream & mim::Def::stream	(	std::ostream &	os,
		int	max ) const

Definition at line 557 of file dump.cpp.

References world().

Referenced by dump(), and write().

◆ sym() [1/4]

Sym mim::Def::sym ( ) const

inline

Definition at line 612 of file def.h.

References dbg().

Referenced by mim::plug::autodiff::phase::Eval::augment_lam(), debug_prefix(), debug_suffix(), mim::World::Externals::externalize(), mim::World::Externals::internalize(), mim::plug::gpu::phase::SplitOffKernels::rewrite_mut_Lam(), set(), set(), set(), set(), transfer_external(), and unique_name().

◆ sym() [2/4]

Sym mim::Def::sym ( const char * s ) const

protected

Definition at line 426 of file def.cpp.

References mim::World::sym(), and world().

◆ sym() [3/4]

Sym mim::Def::sym ( std::string s ) const

protected

Definition at line 428 of file def.cpp.

References mim::World::sym(), and world().

◆ sym() [4/4]

Sym mim::Def::sym ( std::string_view s ) const

protected

Definition at line 427 of file def.cpp.

References mim::World::sym(), and world().

◆ tproj()

const Def * mim::Def::tproj ( nat_t i ) const

inline

As above but takes Def::num_tprojs.

Definition at line 435 of file def.h.

References num_tprojs(), and proj().

◆ tprojs() [1/2]

auto mim::Def::tprojs ( ) const

inline

Definition at line 466 of file def.h.

References tprojs().

◆ tprojs() [2/2]

template<class F>

auto mim::Def::tprojs ( F f ) const

inline

Definition at line 453 of file def.h.

References num_tprojs(), and projs().

Referenced by tprojs().

◆ transfer_external()

void mim::Def::transfer_external ( Def * to )

Definition at line 610 of file def.cpp.

References externalize(), internalize(), and sym().

◆ tvar()

const Def * mim::Def::tvar ( nat_t i )

inlinenoexcept

Definition at line 479 of file def.h.

Referenced by mim::Scalarize::rewrite_mut_Lam().

◆ tvars() [1/2]

auto mim::Def::tvars ( )

inlinenoexcept

Definition at line 479 of file def.h.

◆ tvars() [2/2]

template<class F>

auto mim::Def::tvars ( F f )

inlinenoexcept

Definition at line 479 of file def.h.

◆ type()

const Def * mim::Def::type ( ) const

inlinenoexcept

Yields the "raw" type of this Def (maybe nullptr).

See also: Def::unfold_type.

Definition at line 1111 of file def.h.

References var(), and var_type().

Referenced by arity(), mim::Checker::assignable(), mim::plug::autodiff::phase::Eval::augment_(), mim::plug::autodiff::phase::Eval::augment_app(), mim::plug::autodiff::phase::Eval::augment_extract(), mim::plug::autodiff::phase::Eval::augment_pack(), mim::plug::autodiff::phase::Eval::augment_tuple(), mim::Bound::Bound(), mim::App::callee_type(), check(), mim::plug::clos::clos_pack(), cmp(), mim::plug::tensor::phase::compose_map(), Def(), Def(), Def(), mim::sexpr::Emitter::emit_bb(), mim::sexpr::Emitter::emit_node(), mim::plug::clos::ClosLit::env_type(), mim::Idx::expect_bitwidth(), mim::Ext::Ext(), mim::Bound::find(), mim::plug::ll::find_common_simd_src(), mim::plug::clos::ClosLit::fnc_type(), mim::Bound::get(), mim::Axm::infer_curry_and_trip(), is_term(), mim::is_unit(), mim::plug::clos::ClosLit::isa_clos_lit, mim::plug::mem::isa_mem(), mim::plug::ll_nvptx::HostEmitter::isa_targetspecific_intrinsic(), mim::plug::mem::mem_def(), mim::plug::core::normalize_bitcast(), mim::plug::core::normalize_conv(), mim::plug::math::normalize_conv(), mim::plug::tensor::normalize_set(), mim::plug::tensor::normalize_shape(), mim::plug::refly::normalize_type(), mim::plug::core::op(), mim::plug::cps::op_cps2ds_dep(), mim::plug::mem::op_lea(), mim::plug::mem::op_lea_unsafe(), mim::optimize(), mim::Pi::Pi(), mim::Pi::Pi(), mim::plug::mem::pointee(), mim::plug::ll_nvptx::DeviceEmitter::prepare(), proj(), mim::plug::tensor::phase::read_through(), mim::Reform::Reform(), mim::Zonker::rewire_mut(), mim::plug::clos::phase::LowerTypedClos::rewrite(), mim::plug::clos::phase::LowerTypedClos::rewrite_imm(), mim::plug::affine::phase::LowerFor::rewrite_imm_App(), mim::plug::clos::phase::ClosConvPrep::rewrite_imm_App(), mim::plug::mem::phase::AddMem::rewrite_imm_App(), mim::plug::tensor::phase::LowerToMem::rewrite_imm_App(), mim::InplaceRWPhase::rewrite_mut(), mim::Scalarize::rewrite_mut_Lam(), mim::Rewriter::rewrite_mut_Seq(), mim::Scheduler::Scheduler(), set_type(), mim::plug::mem::strip_mem(), mim::Hole::tuplefy(), mim::Global::type(), mim::Lam::type(), mim::plug::clos::ClosLit::type(), mim::Rule::type(), and unfold_type().

◆ unfold_type()

const Def * mim::Def::unfold_type ( ) const

Yields the type of this Def and builds a new Type (UInc n) if necessary.

Definition at line 451 of file def.cpp.

References type(), and world().

Referenced by mim::World::extract(), mim::World::implicit_app(), mim::Pi::infer(), mim::Reform::infer(), mim::World::insert(), mim::isa_dim(), isa_type(), mim::World::match(), mim::World::seq(), mim::type_of(), and mim::World::uniq().

◆ unique_name()

std::string mim::Def::unique_name ( ) const

name + "_" + Def::gid

Definition at line 616 of file def.cpp.

References gid(), and sym().

Referenced by mim::plug::ll::Emitter::emit_slot(), mim::plug::ll_nvptx::DeviceEmitter::emit_slot(), mim::plug::ll::Emitter::id(), mim::Nest::Node::name(), mim::plug::ll::Emitter::prepare(), mim::sexpr::Emitter::prepare(), and mim::plug::gpu::phase::SplitOffKernels::rewrite_mut_Lam().

◆ unset()

Def * mim::Def::unset ( )

Unsets all Def::ops; works even, if not set at all or only partially set.

Definition at line 213 of file def.cpp.

References num_ops().

Referenced by mim::Zonker::rewire_mut(), mim::InplaceRWPhase::rewrite_mut(), mim::Arr::unset(), mim::Hole::unset(), mim::Lam::unset(), mim::Pack::unset(), mim::Pi::unset(), mim::Reform::unset(), mim::Rule::unset(), mim::Seq::unset(), and mim::Sigma::unset().

◆ users()

Muts mim::Def::users ( )

inline

Set of mutables where this mutable is locally referenced.

Definition at line 520 of file def.h.

◆ var() [1/3]

const Def * mim::Def::var ( )

Not necessarily a Var: E.g., if the return type is [], this will yield ().

Definition at line 226 of file def.cpp.

References mim::World::var(), and world().

◆ var() [2/3]

const Def * mim::Def::var	(	nat_t	a,
		nat_t	i )

inlinenoexcept

Definition at line 479 of file def.h.

Referenced by mim::plug::autodiff::phase::Eval::augment_app(), mim::plug::autodiff::phase::Eval::augment_lam(), mim::plug::tensor::phase::compose_map(), mim::plug::autodiff::phase::Eval::derive_(), mim::ast::TuplePtrn::emit_body(), mim::sexpr::Emitter::emit_head(), mim::sexpr::Emitter::emit_imported(), mim::plug::ll::Emitter::emit_phi_args(), mim::Lam::eta_reduce(), has_free_var(), mim::Rule::is_in_rule(), isa_binder(), mim::plug::clos::isa_clos_type(), mim::plug::mem::mem_var(), num_tvars(), mim::plug::tensor::phase::reads_injectively(), mim::Rule::replace(), mim::Lam::ret_var(), mim::plug::affine::phase::LowerFor::rewrite_imm_App(), mim::plug::affine::phase::LowerIndex::rewrite_imm_App(), mim::plug::mem::phase::AddMem::rewrite_mut_Lam(), mim::RetWrap::rewrite_mut_Lam(), mim::Scalarize::rewrite_mut_Lam(), mim::StaticArgOpt::rewrite_mut_Lam(), mim::Hole::tuplefy(), and type().

◆ var() [3/3]

const Def * mim::Def::var ( nat_t i )

inlinenoexcept

Definition at line 479 of file def.h.

◆ var_type()

const Def * mim::Def::var_type ( )

If this is a binder, compute the type of its Variable.

Definition at line 232 of file def.cpp.

References arity(), mim::Arr, mim::Global, mim::Hole, mim::Join, mim::Lam, mim::Meet, node(), mim::Pack, mim::Pi, mim::Rule, mim::Sigma, mim::World::type_idx(), and world().

Referenced by num_tvars(), type(), mim::Var::type(), and mim::World::var().

◆ vars() [1/4]

template<nat_t A = std::dynamic_extent>

auto mim::Def::vars ( )

inlinenoexcept

Definition at line 479 of file def.h.

◆ vars() [2/4]

template<nat_t A = std::dynamic_extent, class F>

auto mim::Def::vars ( F f )

inlinenoexcept

Definition at line 479 of file def.h.

Referenced by Def(), mim::Prod::Def(), free_vars(), has_free_vars_in(), mim::plug::ll::Emitter::prepare(), mim::plug::affine::phase::LowerFor::rewrite_imm_App(), and mim::RetWrap::rewrite_mut_Lam().

◆ vars() [3/4]

auto mim::Def::vars ( nat_t a )

inlinenoexcept

Definition at line 479 of file def.h.

◆ vars() [4/4]

template<class F>

auto mim::Def::vars	(	nat_t	a,
		F	f )

inlinenoexcept

Definition at line 479 of file def.h.

◆ world()

World & mim::Def::world ( ) const

inlinenoexcept

Definition at line 1097 of file def.h.

References mim::Type, mim::Univ, mim::Var, and World.

Referenced by mim::Checker::alpha(), mim::Lam::app(), mim::Lam::app(), mim::plug::clos::apply_closure(), arity(), mim::plug::autodiff::autodiff_type_fun(), mim::plug::autodiff::autodiff_type_fun_pi(), mim::Lam::branch(), mim::cat_sigma(), mim::cat_tuple(), check(), check(), mim::plug::clos::clos_apply(), mim::plug::clos::clos_insert_env(), mim::plug::clos::clos_pack(), mim::plug::clos::clos_remove_env(), mim::plug::clos::clos_sub_env(), mim::plug::clos::clos_type(), mim::plug::clos::clos_type_to_pi(), mim::plug::core::convert(), mim::Phase::create(), dbg(), Def(), Def(), Def(), mim::Prod::Def(), driver(), mim::drop_self(), err_loc(), externalize(), free_vars(), mim::plug::autodiff::id_pullback(), immutabilize(), mim::Pi::infer(), internalize(), mim::is_unit(), mim::plug::clos::isa_clos_type(), mim::plug::mem::mut_con(), mim::plug::compile::normalize_aggr(), mim::plug::core::normalize_bitcast(), mim::plug::buffer::normalize_Buf(), mim::plug::compile::normalize_cond(), mim::plug::ord::normalize_contains(), mim::plug::core::normalize_conv(), mim::plug::math::normalize_conv(), mim::plug::core::normalize_div(), mim::plug::tensor::normalize_fastest_axis(), mim::plug::ord::normalize_get(), mim::plug::refly::normalize_gid(), mim::plug::core::normalize_idx_unsafe(), mim::plug::compile::normalize_is_loaded(), num_tprojs(), mim::plug::core::op(), mim::plug::cps::op_cps2ds_dep(), mim::plug::tensor::op_get(), mim::plug::mem::op_lea(), mim::plug::mem::op_lea_unsafe(), mim::plug::mem::op_lea_unsafe(), mim::plug::tensor::op_set(), mim::plug::autodiff::op_sum(), operator<<, proj(), mim::plug::autodiff::pullback_type(), mim::Lam::reduce(), mim::Rule::replace(), mim::plug::clos::phase::Clos2SJLJ::rewrite(), mim::Lam::set(), set_dbg(), set_dbg_key_(), mim::Pi::set_dom(), mim::Lam::set_filter(), stream(), mim::plug::mem::strip_mem(), mim::plug::mem::strip_mem_ty(), sym(), sym(), sym(), mim::tuple2str(), mim::Hole::tuplefy(), unfold_type(), var(), var_type(), mim::plug::autodiff::zero_def(), mim::plug::autodiff::zero_pullback(), zonk(), and zonk_mut().

◆ write() [1/2]

void mim::Def::write ( int max ) const

Definition at line 585 of file dump.cpp.

References write().

Referenced by write().

◆ write() [2/2]

void mim::Def::write	(	int	max,
		const char *	file ) const

Definition at line 580 of file dump.cpp.

References stream().

◆ zonk() [1/2]

const Def * mim::Def::zonk ( ) const

If Holes have been filled, reconstruct the program without them.

Only goes up to but excluding other mutables.

See also: https://stackoverflow.com/questions/31889048/what-does-the-ghc-source-mean-by-zonk

Definition at line 21 of file check.cpp.

References isa_mut(), needs_zonk(), mim::Zonker::rewrite(), world(), zonk_mut(), and mim::World::zonker().

Referenced by mim::World::app(), mim::World::extract(), mim::World::inj(), mim::World::insert(), mim::World::match(), mim::World::merge(), mim::World::merge(), mim::World::raw_app(), mim::Rewriter::rewrite_mut_Seq(), mim::World::seq(), mim::World::sigma(), mim::World::split(), mim::World::tuple(), mim::World::tuple(), mim::World::type(), mim::World::uniq(), zonk(), and zonk_mut().

◆ zonk() [2/2]

DefVec mim::Def::zonk ( Defs defs )

static

zonks all defs and returns a new DefVec.

Definition at line 51 of file check.cpp.

References zonk().

◆ zonk_mut()

const Def * mim::Def::zonk_mut ( ) const

If mutable, zonk()s all ops and tries to immutabilize it; otherwise just zonk.

Definition at line 27 of file check.cpp.

References deps(), is_set(), isa_mut(), op(), mim::Zonker::rewire_mut(), world(), zonk(), and mim::World::zonker().

Referenced by mim::isa_dim(), and zonk().

◆ DefKey

friend struct DefKey

friend

Definition at line 795 of file def.h.

References DefKey.

Referenced by DefKey.

◆ operator<<

std::ostream & operator<<	(	std::ostream &	os,
		const Def *	def )

friend

This will stream def as an operand.

This is usually id(def) unless it can be displayed Inline.

Definition at line 548 of file dump.cpp.

References mim::World::freeze(), operator<<, and world().

Referenced by operator<<, and swap.

◆ swap

void swap	(	World &	,
		World &	)

friend

References operator<<, and World.

◆ World

friend class World

friend

Definition at line 796 of file def.h.

References World.

Referenced by Def(), swap, World, and world().

Member Data Documentation

◆ @243221335266362140133133112245024260072253327200

union { ... } mim::Def

◆ curry_

u8 mim::Def::curry_ = 0

protected

Definition at line 771 of file def.h.

Referenced by mim::App::curry(), and mim::Axm::curry().

◆ flags_

flags_t mim::Def::flags_

protected

Definition at line 770 of file def.h.

Referenced by Def(), Def(), Def(), flags(), mim::Lit::get(), mim::Pi::make_explicit(), mim::Pi::make_implicit(), and mim::Proxy::tag().

◆ trip_

u8 mim::Def::trip_ = 0

protected

Definition at line 772 of file def.h.

Referenced by mim::App::trip(), and mim::Axm::trip().

The documentation for this class was generated from the following files:

include/mim/def.h
src/mim/check.cpp
src/mim/def.cpp
src/mim/dot.cpp
src/mim/dump.cpp

Public Member Functions

Static Public Member Functions

Protected Member Functions

Protected Attributes

Friends

Syntactic Comparison

Detailed Description

Member Enumeration Documentation

◆ Cmp

Constructor & Destructor Documentation

◆ Def() [1/4]

◆ Def() [2/4]

◆ Def() [3/4]

◆ Def() [4/4]

Member Function Documentation

◆ arity()

◆ as_imm()

◆ as_mut()

◆ blame()

◆ check() [1/2]

◆ check() [2/2]

◆ cmp()

◆ dbg()

◆ dbg_key()

◆ debug_prefix()

◆ debug_suffix()

◆ dep() [1/2]

◆ dep() [2/2]

◆ deps()

◆ dirty()

◆ dot() [1/3]

◆ dot() [2/3]

◆ dot() [3/3]

◆ driver()

◆ dump() [1/2]

◆ dump() [2/2]

◆ err_loc()

◆ error()

◆ expect_mut()

◆ externalize()

◆ flags()

◆ free_vars() [1/2]

◆ free_vars() [2/2]

◆ gid()

◆ greater()

◆ has_dep() [1/2]

◆ has_dep() [2/2]

◆ has_free_var()

◆ has_free_vars()

◆ has_free_vars_in()

◆ has_var() [1/2]

◆ has_var() [2/2]

◆ hash()

◆ immutabilize()

◆ internalize()

◆ is_annex()

◆ is_closed()

◆ is_dirty()

◆ is_elim()

◆ is_external()

◆ is_form()

◆ is_ground()

◆ is_immutabilizable()

◆ is_intro()

◆ is_meta()

◆ is_mutable()

◆ is_open()

◆ is_set()

◆ is_term()

◆ isa_binder()

◆ isa_imm()

◆ isa_mut()

◆ isa_type()

◆ judge()

◆ less()

◆ loc()

◆ local_muts()

◆ local_vars()

◆ mark()

◆ needs_zonk()