p27049618
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!2477 Add descriptin for graph kernel switch 

Merge pull request !2477 from ddwolf/add_desc_for_graph_kernel
This commit is contained in:
mindspore-ci-bot 2020-06-28 21:39:36 +08:00 committed by Gitee
parent eeeffa3622 5977fe7936
commit 08ff9099ed
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/context.py
View File

@ -510,7 +510,7 @@ def reset_auto_parallel_context():
save_graphs_path=str, save_ms_model=bool, save_ms_model_path=str, enable_dump=bool, save_graphs_path=str, save_ms_model=bool, save_ms_model_path=str, enable_dump=bool,
save_dump_path=str, enable_reduce_precision=bool, variable_memory_max_size=str, save_dump_path=str, enable_reduce_precision=bool, variable_memory_max_size=str,
enable_profiling=bool, profiling_options=str, enable_auto_mixed_precision=bool, enable_profiling=bool, profiling_options=str, enable_auto_mixed_precision=bool,
check_bprop=bool, max_device_memory=str, print_file_path=str) enable_graph_kernel=bool, check_bprop=bool, max_device_memory=str, print_file_path=str)
def set_context(**kwargs): def set_context(**kwargs):
""" """
Sets context for running environment. Sets context for running environment.
@ -538,6 +538,8 @@ def set_context(**kwargs):
save_ms_model_path (str): Path to save converted lite model. Default: "." save_ms_model_path (str): Path to save converted lite model. Default: "."
save_graphs_path (str): Path to save graphs. Default: "." save_graphs_path (str): Path to save graphs. Default: "."
enable_auto_mixed_precision (bool): Whether to enable auto mixed precision. Default: True. enable_auto_mixed_precision (bool): Whether to enable auto mixed precision. Default: True.
enable_graph_kernel (bool): Whether to enable composition of basic primitives. These primitives would be
compiled into a fused kernel automatically. Default: False.
reserve_class_name_in_scope (bool) : Whether to save the network class name in the scope. Default: True. reserve_class_name_in_scope (bool) : Whether to save the network class name in the scope. Default: True.
enable_reduce_precision (bool): Whether to enable precision reduction. Default: True. enable_reduce_precision (bool): Whether to enable precision reduction. Default: True.
enable_dump (bool): Whether to enable dump. Default: False. enable_dump (bool): Whether to enable dump. Default: False.

2
mindspore/nn/cell.py
View File

@ -822,7 +822,7 @@ class GraphKernel(Cell):
""" """
Base class for GraphKernel. Base class for GraphKernel.
A `GraphKernel` a composite of basic primitives and can be compiled into a fused kernel automaticly when A `GraphKernel` a composite of basic primitives and can be compiled into a fused kernel automatically when
context.set_context(enable_graph_kernel=True). context.set_context(enable_graph_kernel=True).
Examples: Examples: