Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
69b89388
MegEngine
项目概览
MegEngine 天元
/
MegEngine
12 个月 前同步成功
通知
393
Star
4704
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
69b89388
编写于
3月 04, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
docs(mge/functional): fix debug_param set_execution_strategy docstring
GitOrigin-RevId: 434929c99894a41fe6a025435156d0f00fe39f97
上级
93c7e451
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
18 addition
and
17 deletion
+18
-17
imperative/python/megengine/functional/debug_param.py
imperative/python/megengine/functional/debug_param.py
+18
-17
未找到文件。
imperative/python/megengine/functional/debug_param.py
浏览文件 @
69b89388
...
...
@@ -28,7 +28,7 @@ _valid_string_option = {
def
get_execution_strategy
()
->
Strategy
:
r
"""Returns the execution strategy of :class:`~
module.
.Conv2d` and :func:`~.matmul`
r
"""Returns the execution strategy of :class:`~
.module
.Conv2d` and :func:`~.matmul`
See :func:`~.set_execution_strategy` for possible return values
"""
...
...
@@ -43,31 +43,32 @@ def get_execution_strategy() -> Strategy:
def
set_execution_strategy
(
option
):
r
"""Sets the execution strategy of :class:`~module.Conv2d` and :func:`~.matmul`
r
"""Sets the execution strategy of :class:`~
.
module.Conv2d` and :func:`~.matmul`
Args:
option: Decides how :class:`~.module.Conv2d`and :func:`~.matmul` algorithms are chosen.
Available
value Strategy
option: Decides how :class:`~.module.Conv2d`
and :func:`~.matmul` algorithms are chosen.
Available
strategy values:
* 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.
* "HEURISTIC":
uses heuristic to choose the fastest algorithm.
* "PROFILE": runs possible algorithms on a
real device to find the best one.
* "REPRODUCIBLE": uses algorithms that are
reproducible.
* "OPTIMIZED": uses algorithms that are
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.
The default strategy is "HEURISTIC", these options can be combined to
form a combination option, e.g. PROFILE_REPRODUCIBLE is a combination
of "PROFILE" and "REPRODUCIBLE", which means using the fastest 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_REPRODUCIBLE' uses the fastest of
profiling result that is also reproducible.
*
'HEURISTIC_REPRODUCIBLE'
uses heuristic to choose the fastest algorithm that is also reproducible.
*
"HEURISTIC"
uses heuristic to choose the fastest algorithm.
*
"PROFILE" runs possible algorithms on a
real device to find the best one.
*
"PROFILE_REPRODUCIBLE" uses the fastest
profiling result that is also reproducible.
*
"HEURISTIC_REPRODUCIBLE"
uses heuristic to choose the fastest algorithm that is also reproducible.
The default strategy is
'HEURISTIC'
.
The default strategy is
"HEURISTIC"
.
It can also be set through the environment variable
'MEGENGINE_EXECUTION_STRATEGY'
.
It can also be set through the environment variable
``MEGENGINE_EXECUTION_STRATEGY``
.
"""
if
isinstance
(
option
,
Strategy
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录