MimIR
MimIR is my Intermediate Representation
Loading...
Searching...
No Matches
mim::plug::ll_nvptx::DeviceEmitter Class Reference
Inheritance diagram for mim::plug::ll_nvptx::DeviceEmitter:
[legend]

Public Types

using Super = ll::Emitter
Public Types inherited from mim::plug::ll::Emitter
enum class  Rt {
  embed ,
  ext
}
 How the C runtime wrappers (compiled to a <name>.ll via add_mim_runtime) reach the output. More...
using Super = mim::Emitter<std::string, std::string, BB, Emitter>

Public Member Functions

 DeviceEmitter (World &world, std::ostream &ostream)
void start () final
 Actual entry.
std::string prepare () override
std::optional< std::string > isa_targetspecific_intrinsic (ll::BB &, const Def *) final
bool is_using_libdevice () const
const std::string & get_extra_flags () const
Public Member Functions inherited from mim::plug::ll::Emitter
 Emitter (World &world, std::string name, std::ostream &ostream)
bool is_valid (std::string_view s)
void start () override
 Actual entry.
void emit_imported (Lam *)
virtual void emit_epilogue (Lam *lam)
void finalize ()
std::string emit_bb (BB &bb, const Def *def)
template<class... Args>
void declare (std::format_string< Args... > s, Args &&... args)
void rt_mode (Rt rt)
void rt_module (std::string ll)
 Provides the textual LLVM IR of the runtime module to splice in Rt::embed mode.
bool load_rt_module (std::string_view filename)
 Locates the runtime module rt/<filename> (produced by add_mim_runtime) in the driver's search paths, reads it, and stores it for Rt::embed splicing.
template<class... Args>
void declare_rt (std::format_string< Args... > sig, Args &&... args)
 Declares a runtime wrapper sig (implemented in a C runtime, see add_mim_runtime) and records that the runtime is required by this module.
Public Member Functions inherited from mim::NestPhase< Lam >
 NestPhase (World &world, std::string name, bool elide_empty, bool schedule=false)
const Nestnest () const
Public Member Functions inherited from mim::ClosedMutPhase< Lam >
 ClosedMutPhase (World &world, std::string name, bool elide_empty, bool schedule=false)
bool elide_empty () const
bool schedule () const
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< Phaserecreate ()
 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< Phasetake_resolved ()
 The Phase to use instead; nullptr means elide.
Worldworld ()
Driverdriver ()
Loglog () const
std::string_view name () const
flags_t annex () const
const 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 Profiler counter key of the current run; no-op unless profiling is enabled.

Private Member Functions

std::string convert (const Def *def, bool simd=false) override
std::string emit_slot (ll::BB &, const App *app, const Def *pointee, const Def *addr_space) override
 Device slots live in a module-scope global in their requested address space, not on the stack.

Additional Inherited Members

Static Public Member Functions inherited from mim::Phase
static std::unique_ptr< Phasecreate (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.
Public Attributes inherited from mim::Emitter< std::string, std::string, BB, Emitter >
fe::Tab tab
Protected Member Functions inherited from mim::plug::ll::Emitter
std::string id (const Def *, bool force_bb=false) const
std::string convert_ret_pi (const Pi *)
void emit_phi (Lam *callee, const Def *phi, std::string arg, Lam *pred)
 Registers arg as an incoming phi value for phi in callee, coming from predecessor pred.
void emit_phi_args (Lam *callee, const App *app, Lam *pred)
 Wires all non-mem.M arguments of app into callee's phis, coming from predecessor pred.
Protected Member Functions inherited from mim::Emitter< std::string, std::string, BB, Emitter >
 Emitter (World &world, std::string name, std::ostream &ostream, bool schedule=false)
virtual bool direct_style ()
std::ostream & ostream () const
std::string emit (const Def *def)
 Recursively emits code.
std::string emit_unsafe (const Def *def)
 As above but returning !child().is_valid(value) is permitted.
void visit (const Nest &nest) override
Protected Member Functions inherited from mim::ClosedMutPhase< Lam >
virtual void visit (Lam *)=0
Lamroot () const
Protected Attributes inherited from mim::plug::ll::Emitter
absl::btree_set< std::string > decls_
std::ostringstream type_decls_
std::ostringstream vars_decls_
std::ostringstream func_decls_
std::ostringstream func_impls_
LamMap< const Def * > simd_phi_
Rt rt_ = Rt::embed
bool rt_used_ = false
std::string rt_module_
Protected Attributes inherited from mim::Emitter< std::string, std::string, BB, Emitter >
Lamcurr_lam_
std::ostream & ostream_
Scheduler scheduler_
DefMap< std::string > locals_
DefMap< std::string > globals_
DefMap< std::string > types_
LamMap< BB > lam2bb_
Protected Attributes inherited from mim::Phase
std::string name_

Detailed Description

Definition at line 54 of file ll_nvptx.cpp.

Member Typedef Documentation

◆ Super

Constructor & Destructor Documentation

◆ DeviceEmitter()

mim::plug::ll_nvptx::DeviceEmitter::DeviceEmitter ( World & world,
std::ostream & ostream )
inline

Member Function Documentation

◆ convert()

std::string mim::plug::ll_nvptx::DeviceEmitter::convert ( const Def * def,
bool simd = false )
inlineoverrideprivatevirtual

Reimplemented from mim::plug::ll::Emitter.

Definition at line 71 of file ll_nvptx.cpp.

References mim::plug::ll::Emitter::convert(), and WLOG.

◆ emit_slot()

std::string mim::plug::ll_nvptx::DeviceEmitter::emit_slot ( ll::BB & ,
const App * app,
const Def * pointee,
const Def * addr_space )
inlineoverrideprivatevirtual

Device slots live in a module-scope global in their requested address space, not on the stack.

Reimplemented from mim::plug::ll::Emitter.

Definition at line 77 of file ll_nvptx.cpp.

References mim::plug::ll_nvptx::HostEmitter::convert(), mim::Def::unique_name(), and mim::plug::ll::Emitter::vars_decls_.

◆ get_extra_flags()

const std::string & mim::plug::ll_nvptx::DeviceEmitter::get_extra_flags ( ) const
inline

Definition at line 68 of file ll_nvptx.cpp.

◆ is_using_libdevice()

bool mim::plug::ll_nvptx::DeviceEmitter::is_using_libdevice ( ) const
inline

Definition at line 67 of file ll_nvptx.cpp.

References is_using_libdevice().

Referenced by mim::plug::ll_nvptx::emit_device(), and is_using_libdevice().

◆ isa_targetspecific_intrinsic()

std::optional< std::string > mim::plug::ll_nvptx::DeviceEmitter::isa_targetspecific_intrinsic ( ll::BB & bb,
const Def * def )
finalvirtual

◆ prepare()

◆ start()

void mim::plug::ll_nvptx::DeviceEmitter::start ( )
finalvirtual

Actual entry.

Reimplemented from mim::ClosedMutPhase< Lam >.

Definition at line 452 of file ll_nvptx.cpp.

References mim::plug::ll::Emitter::start(), and mim::Phase::world().


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