!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
commit 08ff9099ed
2 changed files with 4 additions and 2 deletions

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_dump_path=str, enable_reduce_precision=bool, variable_memory_max_size=str,
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):
"""
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_graphs_path (str): Path to save graphs. Default: "."
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.
enable_reduce_precision (bool): Whether to enable precision reduction. Default: True.
enable_dump (bool): Whether to enable dump. Default: False.

View File

@ -822,7 +822,7 @@ class GraphKernel(Cell):
"""
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).
Examples: