Scalarization (= argument simplification): More...

#include <mim/phase/scalarize.h>

Inheritance diagram for mim::Scalarize:

Public Member Functions
	Scalarize (World &world, flags_t annex)
Public Member Functions inherited from mim::RWPhase
	RWPhase (World &world, std::string name, Analysis *analysis=nullptr)
	RWPhase (World &world, flags_t annex, Analysis *analysis=nullptr)
Analysis *	analysis ()
const Analysis *	analysis () const
const Def *	lattice (const Def *old_def) const
	Returns the abstract value computed by the associated Analysis for the given old-world Def, or nullptr if no value is available.
const Def *	abstracted (const Def *old_def) const
	Returns lattice(`old_def`) if it differs from `old_def` (i.e. we learned something), otherwise nullptr.
bool	is_top (const Def *old_def) const
virtual bool	analyze ()
	Runs the optional pre-analysis on old_world(), typically to a fixed point, before rewriting begins.
virtual void	rewrite_external (Def *)
const Def *	lookup (const Def *) override
	Lookup old_def by searching in reverse through the stack of maps.
World &	world ()=delete
	Hides both and forbids direct access.
World &	old_world ()
	Get old Defs from here.
World &	new_world ()
	Create new Defs into this.
Public Member Functions inherited from mim::Phase
	Phase (World &world, std::string name)
	Phase (World &world, flags_t annex)
virtual	~Phase ()=default
virtual std::unique_ptr< Phase >	recreate ()
	Creates a new instance; needed by a fixed-point PhaseMan.
virtual void	apply (const App *)
	Invoked if your Phase has additional args.
virtual void	apply (Phase &)
	Dito, but invoked by Phase::recreate.
virtual bool	redirects () const
	If true, Phase::create uses take_resolved().
virtual std::unique_ptr< Phase >	take_resolved ()
	The Phase to use instead; nullptr means elide.
World &	world ()
Driver &	driver ()
const fe::Log &	log () const
std::string_view	name () const
flags_t	annex () const
const fe::Vector< std::string > &	args ()
	Command-line arguments passed to this Phase's plugin via -X <plugin>:<arg>.
bool	todo () const
void	invalidate (bool todo=true)
	Signals that another round of fixed-point iteration is required, either as part of.
virtual void	run ()
	Entry point and generates some debug output; invokes Phase::start.
void	profile_count (std::string_view key, uint64_t n=1)
	Adds `n` to the custom fe::Profiler counter `key` of the current run; no-op unless profiling is enabled.
Public Member Functions inherited from mim::Rewriter
template<class D = Def>
D *	curr_mut () const
	Rewriter (World &world)
virtual	~Rewriter ()
void	reset ()
World &	world ()
void	push (Vars old_vars={}, Vars new_vars={})
void	pop ()
void	bind_old (const Var *var)
	Adds the old `var` - if any - to the innermost scope.
void	bind_new (const Var *var)
	Adds the new `var` - if any - to the innermost scope.
void	bind (const Var old_var, const Var new_var)
const Def *	map (const Def old_def, const Def new_def)
const Def *	map (const Def *old_def, Defs new_defs)
const Def *	map (Defs old_defs, const Def *new_def)
const Def *	map (Defs old_defs, Defs new_defs)
virtual const Def *	rewrite (const Def *)
virtual const Def *	rewrite_imm (const Def *)
virtual const Def *	rewrite_mut (Def *)
const Def *	rewrite_stub (Def old_mut, Def new_mut, fe::View< size_t > new2old)
	Like above, but `new_mut` may have a different number of ops than `old_mut:` new_mut->op(i) is rewritten from old_mut->op(new2old[i]).
virtual DefVec	rewrite (Defs)
virtual const Def *	rewrite_imm_Lit (const Lit *)
virtual const Def *	rewrite_imm_Axm (const Axm *)
virtual const Def *	rewrite_imm_Var (const Var *)
virtual const Def *	rewrite_imm_Proxy (const Proxy *)
virtual const Def *	rewrite_imm_Type (const Type *)
virtual const Def *	rewrite_imm_Level (const Level *)
virtual const Def *	rewrite_imm_LMax (const LMax *)
virtual const Def *	rewrite_imm_LSucc (const LSucc *)
virtual const Def *	rewrite_imm_Lam (const Lam *)
virtual const Def *	rewrite_imm_Sigma (const Sigma *)
virtual const Def *	rewrite_imm_Extract (const Extract *)
virtual const Def *	rewrite_imm_Insert (const Insert *)
virtual const Def *	rewrite_imm_Arr (const Arr *)
virtual const Def *	rewrite_imm_Pack (const Pack *)
virtual const Def *	rewrite_imm_Join (const Join *)
virtual const Def *	rewrite_imm_Inj (const Inj *)
virtual const Def *	rewrite_imm_Match (const Match *)
virtual const Def *	rewrite_imm_Variant (const Variant *)
virtual const Def *	rewrite_imm_Top (const Top *)
virtual const Def *	rewrite_imm_Bot (const Bot *)
virtual const Def *	rewrite_imm_Reform (const Reform *)
virtual const Def *	rewrite_imm_Rule (const Rule *)
virtual const Def *	rewrite_imm_Single (const Single *)
virtual const Def *	rewrite_imm_Wrap (const Wrap *)
virtual const Def *	rewrite_imm_Nat (const Nat *)
virtual const Def *	rewrite_imm_Idx (const Idx *)
virtual const Def *	rewrite_imm_Nom (const Nom *)
virtual const Def *	rewrite_imm_Name (const Name *)
virtual const Def *	rewrite_imm_Struc (const Struc *)
virtual const Def *	rewrite_mut_Hole (Hole *)
virtual const Def *	rewrite_mut_Pi (Pi *)
virtual const Def *	rewrite_mut_Sigma (Sigma *)
virtual const Def *	rewrite_mut_Arr (Arr *)
virtual const Def *	rewrite_mut_Pack (Pack *)
virtual const Def *	rewrite_mut_Variant (Variant *)
virtual const Def *	rewrite_mut_Rule (Rule *)
virtual const Def *	rewrite_imm_Seq (const Seq *seq)
virtual const Def *	rewrite_mut_Seq (Seq *seq)

Private Member Functions
const Def *	rewrite_imm_Pi (const Pi *) final
const Def *	rewrite_imm_App (const App *) final
const Def *	rewrite_mut_Lam (Lam *) final

Additional Inherited Members
Static Public Member Functions inherited from mim::Phase
static std::unique_ptr< Phase >	create (const Flags2Phases &phases, const Def *def)
template<class A, class P>
static void	hook (Flags2Phases &phases)
template<class P, class... Args>
static void	run (Args &&... args)
	Runs a single Phase.
Protected Member Functions inherited from mim::RWPhase
void	start () override
	Actual entry.
virtual void	finalize ()
	Run after all roots have been walked but still before the two worlds are swapped.
const Def *	rewrite_imm_Tuple (const Tuple *) override
const Def *	rewrite_stub (Def , Def ) override
const Def *	coerce (const Def new_type, const Def new_val)
	Converts the new `new_val` into a value of the new `new_type`, where a checking position demands it.
const Def *	rewrite_stub (Def old_mut, Def new_mut, fe::View< size_t > new2old)
	Like above, but `new_mut` may have a different number of ops than `old_mut:` new_mut->op(i) is rewritten from old_mut->op(new2old[i]).
Protected Member Functions inherited from mim::Rewriter
Def2Def &	old2new ()
	The map of the innermost push() scope.
bool	has_intersection (const Def *old_def)
	Does `old_def` depend on the old Vars of any scope?
auto	enter (Def *new_mut)
	Updates curr_mut() to `new_mut` and restores it at the end of the scope.
const Def *	seal (Def old_mut, Def new_mut)
	The tail of rewrite_stub: immutabilizes `new_mut` in hindsight, as rewriting may have made it vacuous.
static bool	splittable (const Def *type)
	Does coerce() explode and reassemble values of `type`?
Protected Attributes inherited from mim::Phase
std::string	name_
Protected Attributes inherited from mim::Rewriter
fe::Vector< Scope >	scopes_

Detailed Description

Scalarization (= argument simplification):

f := λ (x_1: [T_1, T_2], .., x_n: T_n).E

becomes

f' := λ (y_1: T_1, y_2: T_2, .. y_n: T_n).E[x_1 \ (y_1, y_2); ..; x_n \ y_n]

It is type-directed: an immutable Cn is flattened along with all its Lams and Apps, even escaping ones. A flattened value is coerce()d where it meets an unflattened shape, e.g. an Axm's signature. An Axm handed a bare function (autodiff.ad f) is instantiated with the old shape, as its type may rely on it. A Pi is pinned where a coercion would be observable:

a Cn an interface Lam's signature reaches through a type coerce() cannot take apart (mem.Ptr0 K), or
a parameter indexed non-constantly or consumed whole by an Axm - only this parameter is kept.

Each run flattens one level; the fixed-point pipeline repeats it. Flags::scalarize_threshold bounds which Arrs are expanded.

Definition at line 28 of file scalarize.h.

Constructor & Destructor Documentation

◆ Scalarize()

mim::Scalarize::Scalarize	(	World &	world,
		flags_t	annex )

inline

Definition at line 63 of file scalarize.h.

References mim::Phase::annex(), mim::RWPhase::RWPhase(), and mim::RWPhase::world().

Member Function Documentation

◆ rewrite_imm_App()

const Def * mim::Scalarize::rewrite_imm_App ( const App * app )

finalprivatevirtual

Reimplemented from mim::RWPhase.

Definition at line 156 of file scalarize.cpp.

References mim::World::app(), mim::App::arg(), mim::App::axm(), mim::App::callee(), mim::RWPhase::coerce(), mim::Pi::dom(), mim::RWPhase::new_world(), mim::Rewriter::rewrite(), mim::RWPhase::rewrite_imm_App(), and mim::Def::type().

◆ rewrite_imm_Pi()

const Def * mim::Scalarize::rewrite_imm_Pi ( const Pi * pi )

finalprivatevirtual

Reimplemented from mim::Rewriter.

Definition at line 154 of file scalarize.cpp.

◆ rewrite_mut_Lam()

const Def * mim::Scalarize::rewrite_mut_Lam ( Lam * old )

finalprivatevirtual

Reimplemented from mim::Rewriter.

Definition at line 246 of file scalarize.cpp.

References mim::Lam::is_rewritable(), mim::Rewriter::rewrite_mut_Lam(), mim::RWPhase::rewrite_stub(), and mim::Lam::type().

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

include/mim/phase/scalarize.h
src/mim/phase/scalarize.cpp

Public Member Functions

Private Member Functions

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ Scalarize()

Member Function Documentation

◆ rewrite_imm_App()

◆ rewrite_imm_Pi()

◆ rewrite_mut_Lam()