megengine.traced_module.wrap¶

调用这个函数将 func 注册为一个内置函数。

这个函数可以在模块级作用域被调用，来将 func 注册为一个内置函数。在 trace 时，内置函数将被转换为 CallFunction Expr.

def my_func(x, y):
    return x + y

import megengine.traced_module as tm
tm.wrap(my_func)

这个函数也可以用作装饰器：

@tm.wrap
def my_func(x, y):
    return x + y

megengine.traced_module.register_as_builtin

ModuleNode