megengine.traced_module.register_as_builtin

register_as_builtin(mod_cls)

Registers class mod_cls (subclass of Module) as builtin module.

Parameters

mod_cls (Type[Module]) – the module class which will be treated as builtin module in tracing.

Return type

None

previous

megengine.traced_module.trace_module

next

megengine.traced_module.wrap