From 101e109f7ec0bf18581edcf7edcc1b297d76b760 Mon Sep 17 00:00:00 2001 From: pxp1 <958876660@qq.com> Date: Tue, 1 Jul 2025 14:26:32 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BD=BF=E7=94=A8=E7=A7=81?= =?UTF-8?q?=E6=9C=89=E6=96=87=E4=BB=B6=E6=93=8D=E4=BD=9C=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../msprobe/core/single_save/single_comparator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/debug/accuracy_tools/msprobe/core/single_save/single_comparator.py b/debug/accuracy_tools/msprobe/core/single_save/single_comparator.py index e1cf68dfec..70c4b82697 100644 --- a/debug/accuracy_tools/msprobe/core/single_save/single_comparator.py +++ b/debug/accuracy_tools/msprobe/core/single_save/single_comparator.py @@ -21,7 +21,7 @@ import numpy as np import pandas as pd from tqdm import tqdm -from msprobe.core.common.file_utils import check_file_or_directory_path, create_directory, save_excel +from msprobe.core.common.file_utils import check_file_or_directory_path, create_directory, load_npy, save_excel from msprobe.core.common.log import logger from msprobe.core.common.utils import check_process_num @@ -191,8 +191,8 @@ class SingleComparator: common_keys = set(path_dict1.keys()) & set(path_dict2.keys()) for key in common_keys: try: - array1 = np.load(path_dict1[key]) - array2 = np.load(path_dict2[key]) + array1 = load_npy(path_dict1[key]) + array2 = load_npy(path_dict2[key]) result = cls.compare_arrays(array1, array2) step, rank, micro_step, array_id = key data.append([ -- Gitee