megengine.config

async_level

决定是否在正当执行算子出错的时候, 获取或设置config。 默认等级是2, 代表设备和用户侧的错误都是异步的

实际案例

import megengine as mge
mge.config.async_level = 2
返回类型

int

benchmark_kernel

是否在真实设备上跑算法以找到最佳的算法。 默认选项是false, 代表用启发式算法寻找最佳算法

实际案例

import megengine as mge
mge.config.benchmark_kernel = True
benchmark_with_subprocess

是否在真实设备上跑算法以找到最佳的算法。 默认选项是false, 代表用启发式算法寻找最佳算法

实际案例

import megengine as mge
mge.config.benchmark_with_subprocess = True
deterministic_kernel

最佳算法是否是可复现的。 默认是false,代表算法是不可复现的。

实际案例

import megengine as mge
mge.config.deterministic_kernel = True
disable_memory_forwarding

设置config以决定是否禁用内存传递。 默认选项是false, 代表内存可以在tensor之间共享。

实际案例

import megengine as mge
mge.config.disable_memory_forwarding = False
返回类型

bool