From d7d36c4328fa0ea67304a707998d37980e51b886 Mon Sep 17 00:00:00 2001 From: fary86 Date: Sat, 19 Sep 2020 17:39:29 +0800 Subject: [PATCH] Fix print_file_path check --- mindspore/context.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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