Browse Source

Fix print_file_path check

tags/v1.0.0
fary86 5 years ago
parent
commit
d7d36c4328
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      mindspore/context.py

+ 3
- 2
mindspore/context.py View File

@@ -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


Loading…
Cancel
Save