megengine.functional.debug_param.set_execution_strategy¶

set_execution_strategy(option)[源代码]¶

设置用于 Conv2d and matmul 等算子的执行策略算法

参数

option –

Decides how matmul algorithms are chosen. Available value Strategy

HEURISTIC：使用启发式方式来选择最快的算法。

PROFILE：在当前真实设备上运行所有可能的算法来确定最优算法。

REPRODUCIBLE：使用具有可复现特性的算法。

OPTIMIZED：使用被优化过的算法。

默认策略是 HEURISTIC，一个维度的选项可与其它维度的选项相结合，比如： PROFILE | REPRODUCIBLE 可以合并得到一个新策略，使用根据测速结果最快的，并且可复现的算法。

可选的字符串值为：

默认策略是 HEURISTIC。

该选项也可以通过设置环境变量 MEGENGINE_EXECUTION_STRATEGY 来进行修改。

megengine.functional.debug_param.get_execution_strategy

megengine.module