diff --git a/mindspore/context.py b/mindspore/context.py index 3fef630eaf..aa4ee135a4 100644 --- a/mindspore/context.py +++ b/mindspore/context.py @@ -461,6 +461,7 @@ def _check_target_specific_cfgs(device, arg_key): 'enable_auto_mixed_precision': ['Ascend'], 'enable_dump': ['Ascend'], 'enable_profiling': ['Ascend'], + 'print_file_path': ['Ascend'], 'variable_memory_max_size': ['Ascend'], 'max_device_memory': ['GPU'] } @@ -509,10 +510,10 @@ def set_context(**kwargs): device_id enable_dump device_target enable_profiling enable_graph_kernel variable_memory_max_size - enable_reduce_precision + enable_reduce_precision print_file_path enable_sparse + max_call_depth mode - print_file_path profiling_options reserve_class_name_in_scope save_dump_path