megengine.config

async_level

Get or set config whether raise error exactly when invoking op. The default level is 2, which means both device and user side errors are async.

Examples

import megengine as mge
mge.config.async_level = 2

Return type

int

benchmark_kernel

Whether or not run possible algorithms on real device to find the best one. The default option is false, which means use heuristic to choose the fastest algorithm.

Examples

import megengine as mge
mge.config.benchmark_kernel = True

benchmark_with_subprocess

Whether or not run possible algorithms on real device to find the best one. The default option is false, which means use heuristic to choose the fastest algorithm.

Examples

import megengine as mge
mge.config.benchmark_with_subprocess = True

deterministic_kernel

Whether or not the fastest algorithm choosed is reproducible. The default option is false, which means the algorithm is not reproducible.

Examples

import megengine as mge
mge.config.deterministic_kernel = True

disable_memory_forwarding

Get or set config whether to disable memory forwarding. The default option is false, which means storage may be shared among tensors.

Examples

import megengine as mge
mge.config.disable_memory_forwarding = False

Return type

bool

previous

megengine.random.poisson

next

lite/tensor.h