Mila::Dnn::Extensibility::IModulePlugin Class Reference

Public Member Functions
virtual	~IModulePlugin ()=default
virtual bool	canHandle (const std::string &module_type) const =0
	Check if this plugin handles the given module type.
virtual PluginInfo	getInfo () const =0
virtual void	registerOperations (Compute::OperationRegistry &registry)=0
	Create module instance from archive.

Constructor & Destructor Documentation

~IModulePlugin()

virtual Mila::Dnn::Extensibility::IModulePlugin::~IModulePlugin ( )

virtualdefault

Member Function Documentation

canHandle()

virtual bool Mila::Dnn::Extensibility::IModulePlugin::canHandle ( const std::string & module_type ) const

pure virtual

Check if this plugin handles the given module type.

getInfo()

virtual PluginInfo Mila::Dnn::Extensibility::IModulePlugin::getInfo ( ) const

pure virtual

registerOperations()

virtual void Mila::Dnn::Extensibility::IModulePlugin::registerOperations ( Compute::OperationRegistry & registry )

pure virtual

Create module instance from archive.

Parameters

device_type	Device as string (Device::Cpu(), "Cuda")
precision	Precision as string ("float32", etc.)
archive	Archive to read from
module_name	Module name in archive
exec_context	Execution context (type-erased)

Returns

Type-erased module pointer

Register backend compute operations

Called when plugin is loaded to register operations with the compute backend (similar to how you register operations now).

---

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

• /__w/Mila/Mila/Mila/Src/Dnn/Extensibility/IModulePlugin.ixx