diff --git a/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py b/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py index 2e152f4dba3a78ce7c933dd438d832bf7286699e..de025784bd3007fe47746791de231b22bb65fd59 100644 --- a/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py +++ b/debug/accuracy_tools/msprobe/mindspore/dump/cell_dump_process.py @@ -564,6 +564,11 @@ def is_download_finished(directory, interval=3): def process(dump_path): + if not os.path.exists(dump_path): + logger.warning('No grap cell data is dumped.') + create_directory(dump_path) + return + rank_id = os.environ.get('RANK_ID') rank_dir = DEFAULT_RANK_DIR if rank_id is not None: @@ -653,6 +658,11 @@ def merge_file(dump_path, rank_dir, file_dict): def process_statistics(dump_path): + if not os.path.exists(dump_path): + logger.warning('No grap cell data is dumped.') + create_directory(dump_path) + return + rank_id = os.environ.get('RANK_ID') rank_dir_kbk = "rank_0" if rank_id is not None: