megengine.functional.debug_param.set_execution_strategy¶

set_execution_strategy(option)[源代码]¶

Sets the execution strategy of Conv2d and matmul

参数

option –

Decides how matmul algorithms are chosen. Available value Strategy

HEURISTIC uses heuristic to choose the fastest algorithm.

PROFILE runs possible algorithms on real device to find the best one.

REPRODUCIBLE uses the algorithms that is reproducible.

OPTIMIZED uses the algorithms that is optimized.

The default strategy is HEURISTIC, this options can be combined to form a combination option, e.g. PROFILE | REPRODUCIBLE can combined a option that uses the fastest of profiling result that is also reproducible.

Available values string:

‘HEURISTIC’ uses heuristic to choose the fastest algorithm.
‘PROFILE’ runs possible algorithms on real device to find the best one.
‘PROFILE_HEURISTIC’ uses profiling result and heuristic to choose the fastest algorithm.
‘PROFILE_REPRODUCIBLE’ uses the fastest of profiling result that is also reproducible.
‘HEURISTIC_REPRODUCIBLE’ uses heuristic to choose the fastest algorithm that is also reproducible.

The default strategy is ‘HEURISTIC’.

It can also be set through the environment variable ‘MEGENGINE_EXECUTION_STRATEGY’.

megengine.functional.debug_param.get_execution_strategy

megengine.module