mindstudio-probe 8.1.0__py3-none-any.whl → 8.1.1__py3-none-any.whl

{mindstudio_probe-8.1.0.dist-info → mindstudio_probe-8.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mindstudio-probe
-Version: 8.1.0
+Version: 8.1.1
 Summary: Ascend Probe Utils
 Home-page: https://gitee.com/ascend/mstt/tree/master/debug/accuracy_tools/msprobe
 Author: Ascend Team
@@ -28,11 +28,12 @@ Requires-Dist: pyyaml
 Requires-Dist: rich
 Requires-Dist: tqdm
 Requires-Dist: openpyxl>=3.0.6
-Requires-Dist: pyopenssl
+Requires-Dist: pyopenssl==24.2.1
 Requires-Dist: twisted
 Requires-Dist: matplotlib
 Requires-Dist: tensorboard
 Requires-Dist: tabulate
 Requires-Dist: pwinput
+Requires-Dist: psutil
 
 MindStudio-Probe is a set of tools for diagnosing and improving model accuracy on Ascend NPU, including API acc checker, ptdbg, grad tool etc.

{mindstudio_probe-8.1.0.dist-info → mindstudio_probe-8.1.1.dist-info}/RECORD

@@ -6,22 +6,22 @@ msprobe/msprobe.py,sha256=pWQYbbADJ-uydeyIqga2tznOCC-spY5r_gr5m75XBC0,8146
 msprobe/core/__init__.py,sha256=X_OkpIz42vnqhGy6WCeuEvYK4IFx4BwCT5bQj7tuuwI,757
 msprobe/core/common_config.py,sha256=j-crSu1_ipmaU2Cs8ciLpUc33e8P_GlBjhGqsN3kZ3U,6385
 msprobe/core/hook_manager.py,sha256=nEEP1G6G8KMHBGPVWaxQfdjgBPqZmMgpCVaFBS_i79E,10574
-msprobe/core/service.py,sha256=4xXg4XbQ-IYbEPwAsEiVsljTTDxaD4vDmjEuskL4LVE,13702
+msprobe/core/service.py,sha256=Nx9dbVJMkEiMCUDQusehzl04AZ94Dl_997O3zO99xlg,13801
 msprobe/core/advisor/advisor.py,sha256=WP8rBF2F9pcSphZONJDgNfz9c6GiRxbkWwPS9wRU4Tc,5900
 msprobe/core/advisor/advisor_const.py,sha256=thEEVYKPimv9WGxxaC4vBXQ-Coq6yBlZ394T_b49Ibw,2624
 msprobe/core/advisor/advisor_result.py,sha256=yM78n6NXLP1bjTsGeSddwb1bWQwtg1ZqjnPHmTzVxHE,2359
-msprobe/core/common/const.py,sha256=iKL3FMunl49vS_dFo1n6ABu6Fmi3SEbL_HxsaKmwaDk,25623
+msprobe/core/common/const.py,sha256=j01Q5KdN7WeHQan9UyommLLKj6sFuOb7LCntGyNqN0E,25680
 msprobe/core/common/decorator.py,sha256=lj1upkAkGYeIV3tnH0bFqp7VDIKqTkFVaE5GUOFUNUE,1785
 msprobe/core/common/exceptions.py,sha256=PJB4UTxT5VxyUE9aTCgwxiEExcuQZCcbu5V7tFdLVrY,4045
-msprobe/core/common/file_utils.py,sha256=MII3PgCfqxJldyu3Z769P7p-Z2cjBTT4Kl-QPDzaWtI,35024
+msprobe/core/common/file_utils.py,sha256=7bEbLA89yeY6fa9OEzcvSdl4K7VFjvxSelCl6g1UxCI,35850
 msprobe/core/common/framework_adapter.py,sha256=xwOOc0cbDt_nIqHsHYT8G6uqhvZywLcOnB0w5UzATFk,6255
 msprobe/core/common/global_lock.py,sha256=MQedbRCzJTLAYGeNeeD1T7Vmn349r7ymGKt8d-U3m2Y,2721
 msprobe/core/common/inplace_op_checker.py,sha256=gwxpddOtPS2dRGbI2JB5ZixgoPLCYyDICb_07N9XVaM,1864
 msprobe/core/common/inplace_ops.yaml,sha256=Ta1zh6eF0A_Trd345gts2xlhFsU0-ZTXKEQOqUHT-54,3178
 msprobe/core/common/log.py,sha256=6mvE2DQtIZ3X66y3pIi8RtMGlznXJaIlHWwKHxhMKGk,3173
 msprobe/core/common/runtime.py,sha256=KpPoWIvIWSGjD-xnldxtZQXcEB6Ip2qj1hVIcZG4tWg,844
-msprobe/core/common/utils.py,sha256=m2voSC8fv7jmKv0tvQtVeATdeAHiDe_y2UXtALICWh8,23263
-msprobe/core/compare/acc_compare.py,sha256=Xh3yj7hlK_FATXX7PIXIPfAViJuPN5LaJoVkh08xVe0,34852
+msprobe/core/common/utils.py,sha256=4gvRuJ4oNFhZEiJGxPkhUIrrP_KwUHrJ8w_RWCHF880,23789
+msprobe/core/compare/acc_compare.py,sha256=SddSF82z4HI5ontlJ9wCoxZkMf21w1oeMs3RUlOxWTE,35230
 msprobe/core/compare/check.py,sha256=elHN9T4RG7ixfmvRCyDxY5xXg9MnVFUx3a_U-sfI8UM,3777
 msprobe/core/compare/compare_cli.py,sha256=PTnym0z3JKW6v-JgoGbNgxtJ36ZUQ6Go1QDQX20Qvp0,4271
 msprobe/core/compare/config.py,sha256=IoliSaGxpaRKvsWOiXbBAZ97lJFAwO_k4gRn0dezZyM,2806
@@ -29,7 +29,7 @@ msprobe/core/compare/highlight.py,sha256=pKJQ_vBZM2JpqXItt--8jfmSrBx6ImjkreOkiud
 msprobe/core/compare/ms_to_pt_api.yaml,sha256=NGzy6_yIArM6V0zYsW3sg3KLMJe0sr9ljKiIcHujJus,13203
 msprobe/core/compare/multiprocessing_compute.py,sha256=fdWNtXoqXyNjrunDx482Og6b7zNy0xYlRKD2IibWBMc,13563
 msprobe/core/compare/npy_compare.py,sha256=U0VNtZMaeyomA-7D8fGWn3GYAwYDSCNU7AABvKRLFSI,12811
-msprobe/core/compare/utils.py,sha256=w1bqk3BzRVgdGnnOalyW_lUSIZjjDAAMe-1i5k4X9nA,25677
+msprobe/core/compare/utils.py,sha256=ClblaATqsczJ3ye6depVSXaCqaqAvEGNmmXxL8Pz7LU,26078
 msprobe/core/compare/layer_mapping/__init__.py,sha256=MLMlb-zcaZCmG7Ykh6cY28kBTw-3_waNBRKbajwQc74,776
 msprobe/core/compare/layer_mapping/data_scope_parser.py,sha256=WU9nelUpbkPa7RH_G3QAHRUhiWSHQ7Bt7TTMGFy51eM,9429
 msprobe/core/compare/layer_mapping/layer_mapping.py,sha256=VOD8lWVDp6JExvPiee4YOkayEYPkT6EVqgjG7Z4owmk,10237
@@ -58,14 +58,14 @@ msprobe/core/config_check/resource/hyperparameter.yaml,sha256=365VVSmZbDZiRrXiCx
 msprobe/core/config_check/utils/hyperparameter_parser.py,sha256=bfnyaA7IuRzyw3BfuawjNabWgbydKEFpPc0Kr9R8UIk,4336
 msprobe/core/config_check/utils/utils.py,sha256=ktRV40AIsFUAeP6Qt2JFM4dH-i2Xt7JTBW7rEn3hwnc,3543
 msprobe/core/data_dump/api_registry.py,sha256=dPzQykReQofCOUTxQbCiOfgYOeUltvMy4HC_5NzGXoc,11654
-msprobe/core/data_dump/data_collector.py,sha256=IZWA2rIK2c8qfzCR--313YANffd52QvBJas9_dUPR0g,11545
-msprobe/core/data_dump/json_writer.py,sha256=ZoP-6064kYWnCEz4AduDXuovr3sTR9G8OtROGLO6D-Q,10098
+msprobe/core/data_dump/data_collector.py,sha256=DM5bBFnzOFaTw6-Xjp17TN534OWWo_AKB-TVHdfTTDs,13659
+msprobe/core/data_dump/json_writer.py,sha256=XSQj-UsHtR-wtZgiFhinw37wRez7muJ4pimA01FU9Oc,11485
 msprobe/core/data_dump/scope.py,sha256=VWn2G8m5f_zVraqTHshDDaRHD0eJhR-5hRoebOoJLF8,9897
 msprobe/core/data_dump/data_processor/base.py,sha256=wP2DjKLIuhPPy-2KDg6alqe6utRB99wc8SXvpN5zv5k,19998
 msprobe/core/data_dump/data_processor/factory.py,sha256=onl0M8URU8Tjuw6mMdAHEwg2G-kL1fWXb8vPP5lmH_Y,4381
 msprobe/core/data_dump/data_processor/mindspore_processor.py,sha256=MC0pYFyAfnUYauka8lyBWHGLUsikWeKoyqDxCEmmBpE,15572
 msprobe/core/data_dump/data_processor/pytorch_processor.py,sha256=aoOjpOOnNzLKpLB2Wow6GV8t16VBtgDMJz2azfjxQds,25572
-msprobe/core/debugger/precision_debugger.py,sha256=DR1WcE04YdAXF6SQw36kZN6dGS0WwoQ0yJBCqEc0Rik,5719
+msprobe/core/debugger/precision_debugger.py,sha256=I49cTW0hrtvmLijfztsVzBuKjiWEVJpkuq0EnrU-OCw,5730
 msprobe/core/grad_probe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/core/grad_probe/constant.py,sha256=VjKIJU5NuVWB9K63_LaND2LeQgDxLNg_kg8w63K9Ero,2323
 msprobe/core/grad_probe/grad_compare.py,sha256=B3HLYy_O8lg5vOjypjEIBBC9e_u9VYbaUoH6Xp_IA_4,8270
@@ -81,16 +81,16 @@ msprobe/core/overflow_check/ignore_rules.yaml,sha256=3k5HEKgF7_LtTexRk19AT1UZQ99
 msprobe/core/overflow_check/level.py,sha256=0w-X2zVshf4I_2qECXm4uz6LqkhY5NFyCdZYU0pC62U,741
 msprobe/core/overflow_check/utils.py,sha256=1j5p2qWBOmOoat1v_Zi8krLOWz2NMCdUjrnD03AWTK8,991
 msprobe/core/single_save/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-msprobe/core/single_save/single_comparator.py,sha256=Rw3Yety0ORJOzSF9LYziUvYefcvD3sHqK7JK8ThmCsA,10062
-msprobe/core/single_save/single_saver.py,sha256=_V5mH3W8WNOD_RDUWfBg9x2JAVHgwBhigZZUensT240,6015
-msprobe/docs/01.installation.md,sha256=Qxw0wHmc9OOwXdB6a3dUGRuUoTuNrtYIpYJF69hYRZY,8277
+msprobe/core/single_save/single_comparator.py,sha256=EqwBiRXRDj_6vHm9Xhwf0VJtVQew7ObZbKNRAbQ9boE,9959
+msprobe/core/single_save/single_saver.py,sha256=IVS7fsG5CWmZNCeUenNwtVxhAmpIWSYHK7az2NpZ678,5562
+msprobe/docs/01.installation.md,sha256=CFITuPYQx30L-ZUWqlkrnkWLe85_U1-9d6IRf7xOTvg,8526
 msprobe/docs/02.config_introduction.md,sha256=-0ikWNlWqlArbZ3bN9GGUvEn9Q211iQ2cOCqm6S13FI,32807
 msprobe/docs/03.config_examples.md,sha256=b8xgis20TDV0dS-a6w91lR68SZiT-OE5QYjo2iEqRr8,4014
 msprobe/docs/04.kernel_dump_PyTorch.md,sha256=Hlc0JFUiVIPgG7bmVkLvJBkoThGE8hGXhh3VzAP09GY,3032
-msprobe/docs/05.data_dump_PyTorch.md,sha256=f2N5VLD6bENgWgtJsSDEv0NV8-mPXu1Vh0B8M40qxig,21741
-msprobe/docs/06.data_dump_MindSpore.md,sha256=Cap_YR2YVRtWJpd4xP_rT7XUhDDmCok25Txg6Rp8OJo,27661
+msprobe/docs/05.data_dump_PyTorch.md,sha256=wK_S-o-TsQ7pDKhGkAQ9kpz3YQSg08uE24Q79JkZfJU,21935
+msprobe/docs/06.data_dump_MindSpore.md,sha256=_dxUyl04wjQi0THwQ7O2rZK2Mm5XhT4JiNX5cTee3zA,27855
 msprobe/docs/07.accuracy_checker_PyTorch.md,sha256=a4WydbrVtbBuSA_rbQaO4yy7uS7yXmo8v83tmTkx2P8,29428
-msprobe/docs/08.accuracy_checker_online_PyTorch.md,sha256=byWgtFHUcO7UjX8OakYO46717hURex4boXfqJndC4xQ,14585
+msprobe/docs/08.accuracy_checker_online_PyTorch.md,sha256=u3kDOo3Y1YMSu2rd2QiIToYxqtpi4epfKk_ngdSc_Sw,14717
 msprobe/docs/09.accuracy_checker_MindSpore.md,sha256=HhegcH2g0FteIObWtIw1ukBt0KAbuB-6N1TNlKgBeDs,10768
 msprobe/docs/10.accuracy_compare_PyTorch.md,sha256=07RjmlzD_m9Fndmk6fH3Hu68RQJMwOORVDo0xb69dfQ,35798
 msprobe/docs/11.accuracy_compare_MindSpore.md,sha256=qqBv0E-AFok7EtylH2Z9JxGu1n14SLctid9hVjOh9WQ,36530
@@ -107,7 +107,7 @@ msprobe/docs/21.visualization_PyTorch.md,sha256=fJYTPJtOuHU9t6h2gBM3nISg0-2u6Pid
 msprobe/docs/22.visualization_MindSpore.md,sha256=fSTV86d7X1GSKIYrzb5GznVYUy4heNn10fExWjHo8Ow,28320
 msprobe/docs/23.generate_operator_PyTorch.md,sha256=XKYf5VZ_PVqq07k4eeIETCJvgnaCTCzbe0Xda5fswUs,10287
 msprobe/docs/24.code_mapping_Mindspore.md,sha256=FB4akTJce685-F9gO_ubKJf64sjsuhAzxQvYwtncFiA,1874
-msprobe/docs/25.tool_function_introduction.md,sha256=-_YFZNLQBMm3cdJQEwQCrTxXJP55sbeiT6v9KsPwM3c,16503
+msprobe/docs/25.tool_function_introduction.md,sha256=I1SFsz9RVdpjyNxLTyPPuKcLZIRDDWuwiVx2O2cQezc,16391
 msprobe/docs/26.data_dump_PyTorch_baseline.md,sha256=w4vVI7q0UVWEI3cJ45GQKxv6uYLGB7Fy2UWXe2mWS28,2391
 msprobe/docs/27.dump_json_instruction.md,sha256=YW6q4U0ccoFwax_fZ5Gqr1CQo5PFlKaxr9S-4ocycZs,21025
 msprobe/docs/28.debugger_save_instruction.md,sha256=bSyStQFEqqMe8ygtjXxAciILYm1BRQ6tskB1B4z_l0A,7278
@@ -116,7 +116,7 @@ msprobe/docs/29.data_dump_MSAdapter.md,sha256=cXPxHH-arqMBH1kLTiMB29W584czt-bB4X
 msprobe/docs/30.overflow_check_MSAdapter.md,sha256=leiJqnUR3L93K3-kDDeEPTcG5UeIl-Gjbq9GCzFjXNw,2208
 msprobe/docs/31.config_check.md,sha256=76PmtMvxb2lnHkxb4S44uRPByK8zum6UpgYW6UJRl7c,3310
 msprobe/docs/32.ckpt_compare.md,sha256=iPCUAC17GChf2_h4e2x4PaponRQYKcqZ-0jMoi2dS6c,2519
-msprobe/docs/33.generate_operator_MindSpore.md,sha256=pDfvYBbZUA8PiWX3zta6spyMpSfhLF4Siw7-kGd96QA,9819
+msprobe/docs/33.generate_operator_MindSpore.md,sha256=A3Zs2eUNKUaFSAHBRUuHVcgTRy0UFO0mpmntYYJITgw,9960
 msprobe/docs/34.RL_collect.md,sha256=P6JltGBHYwRnwfcepdhyzIe9RLLAne0D9PBSN8hPzYU,3396
 msprobe/docs/35.nan_analyze.md,sha256=Jhq5eGdh_ENa7gvzGO90PcDTIHXJMqQxuyccDO-It24,4855
 msprobe/docs/FAQ.md,sha256=4h0IUcDGZXLTm9D1eImOoDuHu6zuZh0ku3XFYmU78wo,10960
@@ -188,28 +188,27 @@ msprobe/docs/visualization/no_mapping_analyze.png,sha256=GOqHERkQZQBzKbxTyYqQG7Z
 msprobe/docs/visualization/top_layer.png,sha256=an5TuWPvM_o5eLSUAmrGby3wcrS04pXfMX4fvydKaVI,13196
 msprobe/mindspore/__init__.py,sha256=EnwZ59AkDwQ_EgHyxqbO3aoc_DiNwUQ9gSZ_PQccwzY,1169
 msprobe/mindspore/cell_processor.py,sha256=6Svn2hsLmZ760vcqAFfdZ2ITv1RY-1f6oaQ81IpMpmI,11643
-msprobe/mindspore/mindspore_service.py,sha256=XRAwYdeh-o41_FgiAVknhkOELV76fkYVD-6MXq2dgmQ,4600
+msprobe/mindspore/mindspore_service.py,sha256=3cmw33a0Hy0--N829-_iEzZbucZNm3pqIldkUgK9j_U,4658
 msprobe/mindspore/ms_config.py,sha256=e8QJcy6lGpv6V2u6UQIp5hwEzlyCG29-LDPB1unzbOg,6116
 msprobe/mindspore/task_handler_factory.py,sha256=UzBzUZNa4mUfk3yQnzkWr-oNep4he4HbRu42OztSp6o,1679
 msprobe/mindspore/api_accuracy_checker/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py,sha256=8lP_ThhnhT5ngaYAIPdgdvvoiwl2pLe_J5Uto2WdR6c,20400
+msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py,sha256=-lddP7qbdQoEtXfnZue6CgXmKPax0aJU0YbrqEw-JQw,20499
 msprobe/mindspore/api_accuracy_checker/api_info.py,sha256=XrHmU-22BOpYObX999NOOPzW7h0wCM5KW89-bZpVBYc,4858
 msprobe/mindspore/api_accuracy_checker/api_runner.py,sha256=a-K7gTf9v2vZgQi6fZ_bhZE5H68Xz_WD2tKtj5jn3fw,12404
 msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py,sha256=MMe1vEpHO1PWS7Wiz4Zze__CFsPwxGPrLSbCU7KZo_o,8470
 msprobe/mindspore/api_accuracy_checker/checker_support_api.yaml,sha256=xs4auF9UUv7LSQS2fDfWCbh6CJeaJ_tKsXedt5hS0G4,1325
 msprobe/mindspore/api_accuracy_checker/cmd_parser.py,sha256=g0Ji8y4r-C1z27XBv96dISj2FJCT1IaAEg2nn180X4g,3715
-msprobe/mindspore/api_accuracy_checker/compute_element.py,sha256=99MTomO7-u0eLZnM4pUWy33Wi-o_S0D6qQbzmwY9oX0,15067
+msprobe/mindspore/api_accuracy_checker/compute_element.py,sha256=18Up8_WLfZPnFn0Wszy7pkUeVL7PAc66dJd9vlJ2o7U,15054
 msprobe/mindspore/api_accuracy_checker/data_manager.py,sha256=Hl8m8vohMV4MvnCp2tFgKP0pHgirALE3r1j8bniiaps,13466
 msprobe/mindspore/api_accuracy_checker/main.py,sha256=ECA3HrNJHgEbOxOANNNl_c0EVq77IgiE45atfMICCF8,1295
 msprobe/mindspore/api_accuracy_checker/multi_api_accuracy_checker.py,sha256=yT66kZ3piC_707E9bGpGQbMH7vpu2NCA9Je7NLmkZn8,9747
 msprobe/mindspore/api_accuracy_checker/multi_data_manager.py,sha256=ry_pFW9dWMTsaMbCkA2g-8dNa3yd-5LX2FbShEzZ1Bs,2684
-msprobe/mindspore/api_accuracy_checker/torch_mindtorch_importer.py,sha256=E2Dx_vstjsAU9AXHRu6ydfWgsvIIDzIqMJ4K6NQcDQY,3487
+msprobe/mindspore/api_accuracy_checker/torch_mindtorch_importer.py,sha256=OzVL6E8PBo_xjsq2FVJqO6uibanyjp9rcVZgi5PszeU,3528
 msprobe/mindspore/api_accuracy_checker/type_mapping.py,sha256=FL5SIQ3dJRKVmicvOOUzZ0zDrLzqXGCb8-fgMCizd-A,4071
 msprobe/mindspore/api_accuracy_checker/utils.py,sha256=uREcuxnIOy0Zeuqeo8wP4g33dcyqND_VtUIzKQBY2kQ,3951
 msprobe/mindspore/api_accuracy_checker/bench_functions/flash_attention_score.py,sha256=QJAveBVaeYN0Tmc75I-x4iA6nKpmk1Td_X3M76ClczA,23323
 msprobe/mindspore/api_accuracy_checker/bench_functions/fusion_operator.py,sha256=NXMgBWA8GW7NK7utftc5X-6aPp0c4uf_99kOUY1fwgA,1464
-msprobe/mindspore/api_accuracy_checker/generate_op_script/config_op.json,sha256=fX4Zg1ddolCAQypm7Eo0IcIUflyS9J1c7hoOcv08PL8,225
-msprobe/mindspore/api_accuracy_checker/generate_op_script/op_generator.py,sha256=tyXhm1imODnBV1MMCrCwYtL09jTZSIHw1iAsAID29ow,19255
+msprobe/mindspore/api_accuracy_checker/generate_op_script/op_generator.py,sha256=lpHr1c1XesCNOfwFqTr97zqyPaE3H1cYY1SurYkzp_E,19490
 msprobe/mindspore/api_accuracy_checker/generate_op_script/operator_replication.template,sha256=ni56NrjDH0Qj-w13pDNw2XCqE-MOOmnq3jW2AuGm--I,81379
 msprobe/mindspore/code_mapping/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/mindspore/code_mapping/bind.py,sha256=NX_8SvfSa7PAFy_PVgR39CgkFz-OPqtbWQYCGBUFMek,8814
@@ -220,7 +219,7 @@ msprobe/mindspore/code_mapping/main.py,sha256=VjxdvroXa5D58QUg56jQUXlJ-YNWmTMeri
 msprobe/mindspore/code_mapping/processor.py,sha256=o-YQyl90NPCpoex3iWbNMGHiWTuqLx8C0UfY2RSZ2wg,1164
 msprobe/mindspore/common/const.py,sha256=6FGrfTASq4yi-SyA3MYE3dOmy-U_GkfmylWfWMIG9-4,5984
 msprobe/mindspore/common/log.py,sha256=b-XZiXylAOK85DH3leCGC0zbxTBRIVTokLOYQAyHCII,1127
-msprobe/mindspore/common/utils.py,sha256=sIvEtic4epzHRqukETAPUVJIJft16py5C47Ckz6nTZs,11710
+msprobe/mindspore/common/utils.py,sha256=QgQc_oQbRSAEficHhAioQxcNTvd8Fl0qgpRxqdYrbd0,11754
 msprobe/mindspore/compare/common_dir_compare.py,sha256=YuLbvL9Hd8nDgG1r-gf3ehnODigSR7CdMm-8pkBppIg,15315
 msprobe/mindspore/compare/distributed_compare.py,sha256=H8zdjBBFyifyl3Nx21jtdoma6qN1duclPsJaArpAW1s,1556
 msprobe/mindspore/compare/ms_compare.py,sha256=eipkoEEgx_duqqnualQJzlmB4dkk1-1ZAry3vIlUOdQ,2066
@@ -228,22 +227,22 @@ msprobe/mindspore/compare/ms_graph_compare.py,sha256=duJxj44Va6hWYsNiEvTxlkN-OMj
 msprobe/mindspore/compare/utils.py,sha256=iPuQoZ4b_hhFI6-NfG9_NdECBymhOQVhz44DdfdyFqA,1374
 msprobe/mindspore/debugger/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/mindspore/debugger/debugger_config.py,sha256=_2B7U7oeEFhM4S4d6JK-pVf5EBfS4fDx3j1sQQM0dE8,7595
-msprobe/mindspore/debugger/precision_debugger.py,sha256=ebI03m2_CqyM-fAh20UTH3TJkH_xV3yyh1h68fjdSxo,8947
+msprobe/mindspore/debugger/precision_debugger.py,sha256=vDe7zuMcUzPsYJTHXFwCZ3RwEAInjdsP47hti3_0r7E,8951
 msprobe/mindspore/dump/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-msprobe/mindspore/dump/cell_dump_process.py,sha256=5y5rvsAshFvSDSzCVgI2fEfIJ6HFj7jmwiIXUyGVk5c,35467
-msprobe/mindspore/dump/cell_dump_with_insert_gradient.py,sha256=7UVocI8KWchCWwEQ8dxew4awbj2CnJ1KwFOx1blKdrk,34880
+msprobe/mindspore/dump/cell_dump_process.py,sha256=wZHlLeTk6IuzcVKZ1Gh1Z6k-ELev2Lpq5B7HUhNBbg4,34446
+msprobe/mindspore/dump/cell_dump_with_insert_gradient.py,sha256=52ef-kU1dGu-mcI757AOYbGU1t3efdCY1ncrbrXo5-c,33829
 msprobe/mindspore/dump/dump_tool_factory.py,sha256=cwmO5jXCae5KXa1XGPAhYtID3cYFSEv8otFnsviww4w,2708
 msprobe/mindspore/dump/graph_mode_cell_dump.py,sha256=Ne-tAFvHuC5UXC_CPgtuR7WINihChOFJG7pRllqkJ7E,5583
 msprobe/mindspore/dump/graph_tensor_dump.py,sha256=mGrWU0LqMvlyUyVPsZiUR7j823gjMt1-SJK4mLi6Yl8,3696
 msprobe/mindspore/dump/jit_dump.py,sha256=iAIPqp3j_LK2KcGTmtI6PnHCsy-2NWSY_nRkhqMQ3jM,4978
 msprobe/mindspore/dump/kernel_graph_dump.py,sha256=DcQMwdGZeLU0WQ-ppOBzssZn8XFcpmRmY323vkjDGr0,3905
 msprobe/mindspore/dump/kernel_kbyk_dump.py,sha256=2SPEoPORnwN0YsNcyVOF-ih9WZiyrKVdrOOVM_ESecI,3094
-msprobe/mindspore/dump/hook_cell/api_register.py,sha256=Q2P8dchHQV8boC7Bb3a4yp_5en8VH1TR0kqtZ5H9_3o,6643
+msprobe/mindspore/dump/hook_cell/api_register.py,sha256=3le1SSMhEAQ2l93hqTcYcik26spTl1bH2OmoeHNGot4,6647
 msprobe/mindspore/dump/hook_cell/hook_cell.py,sha256=3iZmuNhcf-70VrZTa19-kvKpfexgMLP6D3l83jUQgus,2873
 msprobe/mindspore/dump/hook_cell/ms_hook_manager.py,sha256=uuM9wVKszQaFKYePd4nSElCqk82PCFvUK6ZPHod1KNk,3655
 msprobe/mindspore/dump/hook_cell/primitive_hooks.py,sha256=RCbF9pav8fAPlt1ax8Ie_THtaaKlt62B9tAwcWDrTKk,10800
 msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml,sha256=1We_gQvBk66-WEkijwAOJRj8NGUyA8Kg4X6T4PJdgjo,13834
-msprobe/mindspore/dym_loader/hook_dynamic_loader.cpp,sha256=wQvuRMx83G6l53XwXqUJE93iZYVu_qJpOK5z5vyLf74,3250
+msprobe/mindspore/dym_loader/hook_dynamic_loader.cpp,sha256=9sTyCsZMm4NcwPyqloOsSzUa-cPz9w1aheeFz40N6vs,3279
 msprobe/mindspore/dym_loader/hook_dynamic_loader.h,sha256=elDkg54akDKL4kyxtNreI9DJ0npvi-tSgV3z4OGRtYw,1507
 msprobe/mindspore/free_benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/mindspore/free_benchmark/api_pynative_self_check.py,sha256=ggwnqjE6-YHSeZP4YN2C-8whq286zDHdMv0tuvfKgC4,9089
@@ -276,9 +275,9 @@ msprobe/mindspore/mindtorch/__init__.py,sha256=ReXDwvtuuKdJYec0YvcxTLL2bA9RdPuko
 msprobe/mindspore/mindtorch/mindtorch_adaptor.py,sha256=A8hssylam5fdwSz8dO5JN08OdtZurbCooHaVH-iVpn4,11291
 msprobe/mindspore/monitor/common_func.py,sha256=-veG0RHnS5AAXzW7Fzv-biqaVwOG3fIV7KvB9KNzgjc,1557
 msprobe/mindspore/monitor/data_writers.py,sha256=xBj_kv1MpFH3DpB1mdl39vH-1_M81uz4rMJiz4ptKkw,8782
-msprobe/mindspore/monitor/features.py,sha256=mFbLfEYtPifVU73RCkFsfjMuZQmGNL7QzOfn7qIci_s,1744
+msprobe/mindspore/monitor/features.py,sha256=EozOLm3k8YCwFPZ0SOhALr2ZDbk34wDDdjrfU6LHt0M,1676
 msprobe/mindspore/monitor/module_hook.py,sha256=FxLER2Vx9Da_YWRIbLr6ViV-M4jaC79Dw-uCJSkNl6g,40762
-msprobe/mindspore/monitor/optimizer_collect.py,sha256=nUxZdGuv9GeMWlfbm9Ox8WPG7ChRRLMIoeZVdJIaCT8,14221
+msprobe/mindspore/monitor/optimizer_collect.py,sha256=B_WsSnuX5LDAY2LTwuwwC8mMtVcR3X1iautlNzaxCSE,14419
 msprobe/mindspore/monitor/utils.py,sha256=l59gLDoCTxkUyvvrUt7lmd2KFzJH5ZuzmHLiMScsAT4,11584
 msprobe/mindspore/monitor/distributed/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/mindspore/monitor/distributed/distributed_ops.yaml,sha256=3mfRKCnhSio6TYsWQkNpr_Vae2a8oO5TGdFobNAdd0c,292
@@ -295,7 +294,7 @@ msprobe/pytorch/__init__.py,sha256=qIvhnAk61oSpvPU0QI0YAC4zyLKYyOcfNzdXJxN7Klo,1
 msprobe/pytorch/attl_manager.py,sha256=Fe_vnjgmqZPpSHzEV7ayKweFjmBUuQV60MLSuw-fj2A,2706
 msprobe/pytorch/function_factory.py,sha256=Fi4w0zfO64Sd2IU9z45mBbDgS8k_CQEiZ9vRpC_TkVk,4031
 msprobe/pytorch/pt_config.py,sha256=Ciq_IEFxt64vNO2OeM9X5T0CrOqGLOeJCxocyNcpMR0,14485
-msprobe/pytorch/pytorch_service.py,sha256=nxu26e1QgvbAmAbn4VovUtAgX1rKEvn8ZXhca8yG1VY,2975
+msprobe/pytorch/pytorch_service.py,sha256=4VzY3qLj8wgr9K5JLN7INeLDIAriNsInh5hL75ShOQo,3037
 msprobe/pytorch/api_accuracy_checker/.keep,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/pytorch/api_accuracy_checker/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/pytorch/api_accuracy_checker/config.yaml,sha256=2kktin5NoWkSp2MOGDoxlz3MPv0lYXXD5r5SPYArEYs,144
@@ -311,7 +310,7 @@ msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml,sha256
 msprobe/pytorch/api_accuracy_checker/compare/compare.py,sha256=x_xM69LJNZfdO6AtafVEHDD0xetdAPnFQryC0XAwP-8,22125
 msprobe/pytorch/api_accuracy_checker/compare/compare_column.py,sha256=2bQBWg8qVoG3i-HhcUat310kivy7B2de4mQgJrRxWsk,7028
 msprobe/pytorch/api_accuracy_checker/compare/compare_input.py,sha256=OG83KTsDpc-NzUCjtMj5SnzuBmOHaQaJYPoyPaaxKfU,2059
-msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py,sha256=K_YhER9znzATNCgCPBddbSfW5PJcFrKfjbRVyYIw4W8,9839
+msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py,sha256=o64RKRmaMe-8E3qsASZJOxI_SBT15O9SDB_Lz59iW28,9817
 msprobe/pytorch/api_accuracy_checker/generate_op_script/config_op.json,sha256=qIwrmgYSnO08FFrfIPhTWwB5XssHfWWsoRFzKIQYHxc,195
 msprobe/pytorch/api_accuracy_checker/generate_op_script/op_generator.py,sha256=zD8QPEDEXmZOUfpU_5wpXd4f60mZg6grN4H7favRNBM,24049
 msprobe/pytorch/api_accuracy_checker/generate_op_script/operator_replication.template,sha256=Qbv1_maNROMVcJAZgwemw42TZPfgkLP5CAWmjBJDAlk,25286
@@ -327,7 +326,7 @@ msprobe/pytorch/api_accuracy_checker/precision_standard/ulp_compare.py,sha256=pE
 msprobe/pytorch/api_accuracy_checker/run_ut/.keep,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/pytorch/api_accuracy_checker/run_ut/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py,sha256=t8DbR13Sna4d8ovyVmrrMw0xMEI9gueEHexLPKwrv8g,17599
-msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py,sha256=mTj-ukomtUhsDbdv5d0BIy31Jx4rGmsw0qOEPORvGpE,9970
+msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py,sha256=shAtwVsYsr3nZzop0A7OGH1nKQG4sD2BJrx_AqX2mJk,9949
 msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py,sha256=cl9Avx5-_ihF4KPuofGrhPw6dzQqHJLbZslluVPC-LQ,8060
 msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py,sha256=eVKIkMNjGOCs3GaTemyyrEFpWDwQzYUqe93LYaOtHq4,28293
 msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py,sha256=iZMGFpzQfK94Zxj7IJn0ldvcLHS5XVW63SCyK93doAo,11428
@@ -339,7 +338,7 @@ msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py,s
 msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py,sha256=LllAmk0_PqUKDltUukKoZtcVMu9wuEqQKBW2VJyoA4I,3986
 msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py,sha256=DqenaweA_kULje8J-qwiQEMadGGf4ZxlHw9I3gNhjD8,10384
 msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml,sha256=AAEAsbIK8R1O9c9kDb6svfeklqt2W8GHf19ewKpEROo,1057
-msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py,sha256=DV4j3gNXl-vI1rK4QCSDsjFBbX5ldLem5s17CcSQIr4,7272
+msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py,sha256=LmMVRO6loxmZRuPlHw9BUd7TYypgamv-3um8RfKvDWA,7307
 msprobe/pytorch/bench_functions/__init__.py,sha256=BoY2ZOpo_mj2gQ8pcl8oLwFxfHpZidOflJ1aGYDmPrM,1157
 msprobe/pytorch/bench_functions/apply_adam.py,sha256=CNL19fQjwcMMDl1SHv3E_LKUoawkhz8_Kac5iXACsRc,6583
 msprobe/pytorch/bench_functions/apply_adam_w.py,sha256=tUkQEmmUf__kIkBTagtjphR7HaiZRpDjk_q7j6AGl00,1690
@@ -361,7 +360,7 @@ msprobe/pytorch/common/__init__.py,sha256=87H-l6y1gbEL2EQiSbPceeJwKJdcsS6zv-9Nmr
 msprobe/pytorch/common/compare_script.template,sha256=CsfmJ5IuPycAYa0-IfVPXrXVu-o6g8BeEF9v17UK_1w,286
 msprobe/pytorch/common/log.py,sha256=yGLpZ6mGfB-s9sHj_gcL5_PMq_Ai4BJxumA0F0knAso,1121
 msprobe/pytorch/common/parse_json.py,sha256=7HbrBdVCX8umYqa4u8JPHETAxUO1246jl7qhcE4FG1s,2286
-msprobe/pytorch/common/utils.py,sha256=Lag14F-QNgnW5VWQ10ndZjwFfA9xLiekHiyiXyki-YQ,18913
+msprobe/pytorch/common/utils.py,sha256=lSr-W1296ta-mQP_chzo18eDjT63FLuLf--ZLASY6UA,18933
 msprobe/pytorch/compare/distributed_compare.py,sha256=uUMMB4cTsnxHUG8e8TyG5rcrqvip8-lo0DyMA0wB0zQ,916
 msprobe/pytorch/compare/mapping.yaml,sha256=Udlbf1dWDAdnTs8VcDTiSkqsMNj93o2qFUU2ocFgt2k,11697
 msprobe/pytorch/compare/match.py,sha256=j6ydfhs9zNMu6vzxsp_qMei-7-zDMuPO8Uj9i_OIhSA,1886
@@ -369,7 +368,7 @@ msprobe/pytorch/compare/pt_compare.py,sha256=FxRSalLYPd_RlngTmGwJX43HCFX7M7kGc_o
 msprobe/pytorch/compare/utils.py,sha256=8wr-n_x-x_Qh2kRCKN9B3Upj2y4crUj1qKMu3-gtMoE,1891
 msprobe/pytorch/debugger/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/pytorch/debugger/debugger_config.py,sha256=YiIoKNw8UtNoy8Ib8Gvr0ui1swZC5MhUlu0VTI44s0o,7844
-msprobe/pytorch/debugger/precision_debugger.py,sha256=XHEodOd1uLWAKg2eIytrbUMh2mDNaYM0NkN3D6_dJxY,6239
+msprobe/pytorch/debugger/precision_debugger.py,sha256=VeWcVfAasq0zF9NKEjUhTT0jbRGMh0mmlkqirGuls44,6511
 msprobe/pytorch/dump/module_dump/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 msprobe/pytorch/dump/module_dump/hook_wrapper.py,sha256=QjFF7cGFl8NiCGBjS3lD2SxDPt8q6_ANZA6-R_AhN-E,3635
 msprobe/pytorch/dump/module_dump/module_dump.py,sha256=yHKVkDaxayrbanUkxc8Wo7crHc1KNPQAXpLFpbcybjY,1767
@@ -405,7 +404,7 @@ msprobe/pytorch/grad_probe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 msprobe/pytorch/grad_probe/grad_monitor.py,sha256=xb3M1XguMxVsSgZuYwyB08tz_jno1-43cURs2pG1FYU,4910
 msprobe/pytorch/grad_probe/grad_stat_csv.py,sha256=k0YesT0Jy-2LsUlSYEDt8eRuhhfFhpSGv4ImtE74NH0,4940
 msprobe/pytorch/hook_module/__init__.py,sha256=ktS1W-6hBa1q8wF-DGz8kFsfVhiIPWBOj6C3kSatgno,679
-msprobe/pytorch/hook_module/api_register.py,sha256=mPDkCR4XjGZG6ixe7V27MXaSp4a5yyZdg6G6PoUbLoU,5819
+msprobe/pytorch/hook_module/api_register.py,sha256=rrV_LBLy7BpZB3Aox44OyBw1_I9sKF43KYVs2KJZecs,5823
 msprobe/pytorch/hook_module/hook_module.py,sha256=Var0BVu_14K5KxTRSYFCPDgvkXD9wYw6sDFxEjC8_Hc,4548
 msprobe/pytorch/hook_module/jit_script_wrapper.py,sha256=09d1_G_CLd8fd7Q-I4AAr1VVAMu_lMHwubjZwzm3J6k,1177
 msprobe/pytorch/hook_module/pt_hook_manager.py,sha256=YNVYev8o48RpoFHX47yWSRXme2ApdwKPXYWcLFl-iyk,2623
@@ -419,7 +418,7 @@ msprobe/pytorch/monitor/data_writers.py,sha256=ieJTuN7hSzusUqq-Pi251Xwtq1B6iwnIk
 msprobe/pytorch/monitor/features.py,sha256=4C9I7nOKtMl5xYWtdu5bk5cRVtocJE88lYvtTOkcJ-Q,2946
 msprobe/pytorch/monitor/module_hook.py,sha256=yN9GnxTShkM340XL5i0UYZhDTGQ3Z29XsycQHS3JEWo,53013
 msprobe/pytorch/monitor/module_metric.py,sha256=MsRs5mfOz5RT5KCuxUabTKBZZfi9Z1rfM1K_azGwmpo,5615
-msprobe/pytorch/monitor/optimizer_collect.py,sha256=USyqjGEkwZTr3ZA7_BIeo3ayw4_-MyoexDvFxU_T_8s,14743
+msprobe/pytorch/monitor/optimizer_collect.py,sha256=F-MEAm9udWbWWOgU_Ilk9ZnqGZm5TU2n5ABScClV9kU,14941
 msprobe/pytorch/monitor/utils.py,sha256=iRnyKm06mDWEnOkRP3zwh5-k28O9iLDWyOktSIlm6RI,13358
 msprobe/pytorch/monitor/visualizer.py,sha256=HMjBMUUP96aYh2MFmqMD4eKM4QI_YD8STfU0ZdZhcb4,2661
 msprobe/pytorch/monitor/distributed/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -446,22 +445,22 @@ msprobe/pytorch/parse_tool/lib/utils.py,sha256=n1PvqAlyPrvd56SC4V7wSlPyDTzBnrmAK
 msprobe/pytorch/parse_tool/lib/visualization.py,sha256=8PAzoBGL8UWuF9Xryw-v6POBM4SczqBT3fcIBIugYWg,4102
 msprobe/visualization/__init__.py,sha256=rEvJTAx-jTGHBM7-bB6VZ7fwfWwYA961M1eZJj4sRcY,622
 msprobe/visualization/graph_service.py,sha256=10K4vjN3b8q2sIJbJ_FEMM_7tH5UpTyLKRNJQImgbDQ,21631
-msprobe/visualization/utils.py,sha256=GGDDZ0KM71b-65cW_XHq4QHoh_aQT8bQmqo_HUOBDaE,6848
+msprobe/visualization/utils.py,sha256=sq8h_lGUoBybHDWet5oLJgKNFxNDR1zvu3v5JE3f_QA,7008
 msprobe/visualization/builder/__init__.py,sha256=rEvJTAx-jTGHBM7-bB6VZ7fwfWwYA961M1eZJj4sRcY,622
 msprobe/visualization/builder/graph_builder.py,sha256=nz6w5i3pprsRFrx-aRBx12Pp9_pX0L_f7KfzSapNFsM,14086
 msprobe/visualization/builder/msprobe_adapter.py,sha256=-vtIjokZ3fwIGKeugCUxPgb3sg8fmUKPDalKX_Mx8L4,9429
 msprobe/visualization/compare/__init__.py,sha256=vd0u6IEpGE1r-HC4CEhs3csiMaa0bxuFcHtyRlAX0kU,618
 msprobe/visualization/compare/graph_comparator.py,sha256=R8bncXDc1elUQSpxhOYbPCR9tMXDt97YAvaA98kFEug,9606
-msprobe/visualization/compare/mode_adapter.py,sha256=1aGM70hyYQhm4bAKyGatcvoNJwF0T3n7yycQsiixpKU,9344
+msprobe/visualization/compare/mode_adapter.py,sha256=d7yjHzL1rwAg0E8RJRU0B0s7S90q4ZCxTex9EiHoyho,9839
 msprobe/visualization/graph/__init__.py,sha256=rEvJTAx-jTGHBM7-bB6VZ7fwfWwYA961M1eZJj4sRcY,622
 msprobe/visualization/graph/base_node.py,sha256=ItSWwjBsiGkOsDmiFhy_Ct5NsddbOdsFOQCDw0s3EJA,4561
 msprobe/visualization/graph/distributed_analyzer.py,sha256=obQvIwBudtvRcqJrE5DlHDU1Rjp4HBtShcORVVLpkFA,19070
 msprobe/visualization/graph/graph.py,sha256=HI0xp4ripm-8gtn3sxaJG7Vu4I4V7zb3lb3OkhG5N2E,7574
 msprobe/visualization/graph/node_colors.py,sha256=7LpurTuE3edKGilwLVsXD7Ue4bMT7Maz63Udq_6CADM,4504
 msprobe/visualization/graph/node_op.py,sha256=qkbw3ZJkKGQH071C1CLIXi4kavvsLxBTzt0KXncz6fY,1368
-mindstudio_probe-8.1.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mindstudio_probe-8.1.0.dist-info/METADATA,sha256=1fyfUN_Qty5osZuqKTBtoJIlONs790h3FKBFF45bSSM,1405
-mindstudio_probe-8.1.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-mindstudio_probe-8.1.0.dist-info/entry_points.txt,sha256=4ob3a9L018EBZFdlfgMW1lbgeIOhc4F-HCR8gBksaCQ,49
-mindstudio_probe-8.1.0.dist-info/top_level.txt,sha256=LxFEFqelENSyWmRtocCiEUF04IE8aZvwTl7ADB598Tk,8
-mindstudio_probe-8.1.0.dist-info/RECORD,,
+mindstudio_probe-8.1.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mindstudio_probe-8.1.1.dist-info/METADATA,sha256=Y3WQDXenq6-ugQspiz9_lJ1fDcf2sAB2-Ow-0h8GPdc,1435
+mindstudio_probe-8.1.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+mindstudio_probe-8.1.1.dist-info/entry_points.txt,sha256=4ob3a9L018EBZFdlfgMW1lbgeIOhc4F-HCR8gBksaCQ,49
+mindstudio_probe-8.1.1.dist-info/top_level.txt,sha256=LxFEFqelENSyWmRtocCiEUF04IE8aZvwTl7ADB598Tk,8
+mindstudio_probe-8.1.1.dist-info/RECORD,,

msprobe/core/common/const.py

@@ -681,6 +681,7 @@ class FileCheckConst:
     MAX_IR_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
     MAX_ZIP_SIZE = 10737418240  # 10 * 1024 * 1024 * 1024
     MAX_FILE_IN_ZIP_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
+    MAX_FILE_SIZE = 1073741824  # 1 * 1024 * 1024 * 1024
     COMMOM_FILE_SIZE = 1048576  # 1 * 1024 * 1024
     DIR = "dir"
     FILE = "file"

msprobe/core/common/file_utils.py

@@ -33,7 +33,7 @@ import pandas as pd
 from msprobe.core.common.decorator import recursion_depth_decorator
 from msprobe.core.common.log import logger
 from msprobe.core.common.exceptions import FileCheckException
-from msprobe.core.common.const import FileCheckConst
+from msprobe.core.common.const import FileCheckConst, CompareConst
 from msprobe.core.common.global_lock import global_lock, is_main_process
 
 proc_lock = multiprocessing.Lock()
@@ -460,6 +460,17 @@ def save_excel(path, data):
                 return "list"
         raise ValueError("Data must be a DataFrame or a list of (DataFrame, sheet_name) pairs.")
 
+    def save_in_slice(df, base_name):
+        df_length = len(df)
+        if df_length < CompareConst.MAX_EXCEL_LENGTH:
+            df.to_excel(writer, sheet_name=base_name if base_name else 'Sheet1', index=False)
+        else:
+            slice_num = (df_length + CompareConst.MAX_EXCEL_LENGTH - 1) // CompareConst.MAX_EXCEL_LENGTH
+            slice_size = (df_length + slice_num - 1) // slice_num
+            for i in range(slice_num):
+                df.iloc[i * slice_size: min((i + 1) * slice_size, df_length)] \
+                    .to_excel(writer, sheet_name=f'{base_name}_part_{i}' if base_name else f'part_{i}', index=False)
+
     check_path_before_create(path)
     path = os.path.realpath(path)
 
@@ -467,12 +478,12 @@ def save_excel(path, data):
     data_type = validate_data(data)
 
     try:
-        if data_type == "single":
-            data.to_excel(path, index=False)
-        elif data_type == "list":
-            with pd.ExcelWriter(path) as writer:
+        with pd.ExcelWriter(path) as writer:
+            if data_type == "single":
+                save_in_slice(data, None)
+            elif data_type == "list":
                 for data_df, sheet_name in data:
-                    data_df.to_excel(writer, sheet_name=sheet_name, index=False)
+                    save_in_slice(data_df, sheet_name)
     except Exception as e:
         logger.error(f'Save excel file "{os.path.basename(path)}" failed.')
         raise RuntimeError(f"Save excel file {path} failed.") from e
@@ -677,8 +688,23 @@ def os_walk_for_files(path, depth):
     return res
 
 
+def check_zip_file(zip_file_path):
+    with zipfile.ZipFile(zip_file_path, 'r') as zip_file:
+        total_size = 0
+        if len(zip_file.infolist()) > FileCheckConst.MAX_FILE_IN_ZIP_SIZE:
+            raise ValueError(f"Too many files in {os.path.basename(zip_file_path)}")
+        for file_info in zip_file.infolist():
+            if file_info.file_size > FileCheckConst.MAX_FILE_SIZE:
+                raise ValueError(f"File {file_info.filename} is too large to extract")
+
+            total_size += file_info.file_size
+            if total_size > FileCheckConst.MAX_ZIP_SIZE:
+                raise ValueError(f"Total extracted size exceeds the limit of {FileCheckConst.MAX_ZIP_SIZE} bytes")
+
+
 def read_xlsx(file_path, sheet_name=None):
     check_file_or_directory_path(file_path)
+    check_zip_file(file_path)
     try:
         if sheet_name:
             result_df = pd.read_excel(file_path, keep_default_na=False, sheet_name=sheet_name)
@@ -783,17 +809,7 @@ def extract_zip(zip_file_path, extract_dir):
     check_file_suffix(zip_file_path, FileCheckConst.ZIP_SUFFIX)
     try:
         proc_lock.acquire()
-        with zipfile.ZipFile(zip_file_path, 'r') as zip_file:
-            total_size = 0
-            if len(zip_file.infolist()) > FileCheckConst.MAX_FILE_IN_ZIP_SIZE:
-                raise ValueError(f"Too many files in {os.path.basename(zip_file_path)}")
-            for file_info in zip_file.infolist():
-                if file_info.file_size > FileCheckConst.MAX_FILE_IN_ZIP_SIZE:
-                    raise ValueError(f"File {file_info.filename} is too large to extract")
-
-                total_size += file_info.file_size
-                if total_size > FileCheckConst.MAX_ZIP_SIZE:
-                    raise ValueError(f"Total extracted size exceeds the limit of {FileCheckConst.MAX_ZIP_SIZE} bytes")
+        check_zip_file(zip_file_path)
     except Exception as e:
         logger.error(f'Save content to file "{os.path.basename(zip_file_path)}" failed.')
         raise RuntimeError(f"Save content to file {os.path.basename(zip_file_path)} failed.") from e
@@ -877,7 +893,8 @@ class SharedDict:
             self._shm = shared_memory.SharedMemory(create=False, name=name)
         except FileNotFoundError:
             try:
-                self._shm = shared_memory.SharedMemory(create=True, name=name, size=1024 * 1024)
+                # 共享内存空间增加至5M
+                self._shm = shared_memory.SharedMemory(create=True, name=name, size=1024 * 1024 * 5)
                 data = pickle.dumps({})
                 self._shm.buf[0:len(data)] = bytearray(data)
                 logger.debug(f'create shared memory, name: {name}')
@@ -892,6 +909,7 @@ class SharedDict:
             except Exception as e:
                 logger.debug(f'shared dict is unreadable, reason: {e}, create new dict.')
                 self._dict = {}
+                self._shm.buf[:] = bytearray(b'\x00' * len(self._shm.buf))  # 清空内存
                 self._changed = True
 
 

msprobe/core/common/utils.py

@@ -303,6 +303,21 @@ def get_file_type(file_path):
     return file_type
 
 
+def check_dump_json_key(json_data, device_type):
+    task = json_data.get('task', None)
+    if not task:
+        logger.error(f"Task for {device_type} is empty, please check.")
+        raise CompareException(CompareException.INVALID_TASK_ERROR)
+    if 'data' not in json_data:
+        logger.error(f"Missing 'data' in dump.json, please check dump.json of {device_type}.")
+        raise CompareException(CompareException.INVALID_DATA_ERROR)
+    api_data = json_data.get('data')
+    if not isinstance(api_data, dict):
+        logger.error(f"Invalid type for 'data': expected a dict. Please check dump.json of {device_type}.")
+        raise CompareException(CompareException.INVALID_DATA_ERROR)
+    return task, api_data
+
+
 def get_dump_mode(input_param):
     npu_path = input_param.get("npu_json_path", None)
     bench_path = input_param.get("bench_json_path", None)
@@ -310,12 +325,8 @@ def get_dump_mode(input_param):
     bench_json_data = load_json(bench_path)
     json_type = get_file_type(file_path=npu_path)
 
-    npu_task = npu_json_data.get('task', None)
-    bench_task = bench_json_data.get('task', None)
-
-    if not npu_task or not bench_task:
-        logger.error(f"Please check the dump task is correct, npu's task is {npu_task}, bench's task is {bench_task}.")
-        raise CompareException(CompareException.INVALID_TASK_ERROR)
+    npu_task, npu_api_data = check_dump_json_key(npu_json_data, 'npu')
+    bench_task, bench_api_data = check_dump_json_key(bench_json_data, 'bench')
 
     if npu_task != bench_task:
         logger.error(f"Please check the dump task is consistent.")
@@ -328,8 +339,8 @@ def get_dump_mode(input_param):
         return Const.STRUCTURE
 
     if npu_task == Const.STATISTICS:
-        npu_md5_compare = md5_find(npu_json_data['data'], json_type)
-        bench_md5_compare = md5_find(bench_json_data['data'], json_type)
+        npu_md5_compare = md5_find(npu_api_data, json_type)
+        bench_md5_compare = md5_find(bench_api_data, json_type)
         if npu_md5_compare == bench_md5_compare:
             return Const.MD5 if npu_md5_compare else Const.SUMMARY
         else:

msprobe/core/compare/acc_compare.py

@@ -105,7 +105,7 @@ class Comparator:
         stack_json = input_param.get("stack_json_path")
         result_df = self.compare_statistics([npu_json, bench_json, stack_json])
         if not result_df.values.tolist():
-            logger.warning("Can`t match any op.")
+            logger.warning("Can`t match any op. No compare result file generated.")
             return
 
         # compare real data
@@ -186,11 +186,16 @@ class ParseData:
         elif self.mode_config.dump_mode == Const.MD5:
             result[Const.MD5] = []
 
-        api_nums = len(data_json['data'])
+        apis_data = data_json.get('data', None)
+        if not apis_data:
+            logger.warning('No APIs found in dump.json.')
+            return pd.DataFrame(result)
+
+        api_nums = len(apis_data)
         progress_bar = tqdm(total=api_nums, desc="API/Module Read Progress", unit="api/module", ncols=100)
 
         # 从json中循环解析API数据，遍历所有API
-        for data_name in data_json['data']:
+        for data_name in apis_data:
             check_op_str_pattern_valid(data_name)
             merge_list = self.gen_merge_list(data_json, data_name, stack_json_data)
             if not merge_list:
@@ -599,8 +604,12 @@ class CreateTable:
         npu_summary = [CompareConst.NPU_MAX, CompareConst.NPU_MIN, CompareConst.NPU_MEAN, CompareConst.NPU_NORM]
         bench_summary = [CompareConst.BENCH_MAX, CompareConst.BENCH_MIN, CompareConst.BENCH_MEAN,
                          CompareConst.BENCH_NORM]
-        result[npu_summary] = result['summary_x'].apply(self.set_summary).tolist()
-        result[bench_summary] = result['summary_y'].apply(self.set_summary).tolist()
+        if result.empty:
+            result[npu_summary] = pd.DataFrame(columns=npu_summary)
+            result[bench_summary] = pd.DataFrame(columns=bench_summary)
+        else:
+            result[npu_summary] = result['summary_x'].apply(self.set_summary).tolist()
+            result[bench_summary] = result['summary_y'].apply(self.set_summary).tolist()
 
         result_df = pd.DataFrame(columns=header)
         for h in header:

msprobe/core/compare/utils.py

@@ -590,4 +590,10 @@ def compare_distributed_inner(npu_dump_dir, bench_dump_dir, output_path, compare
                 'bench_json_path': bench_path,
                 'is_print_compare_log': is_print_compare_log
             }
-            compare_func(input_param=dump_result_param, output_path=output_path, suffix=f'_{nr}', **kwargs)
+            try:
+                compare_func(input_param=dump_result_param, output_path=output_path, suffix=f'_{nr}', **kwargs)
+            except CompareException as e:
+                if e.code == CompareException.INVALID_DATA_ERROR:
+                    logger.error(f"Invalid or missing 'data' in dump.json. Skipping {nr} comparison.")
+                if e.code == CompareException.INVALID_TASK_ERROR:
+                    logger.error(f"Invalid or missing 'task' in dump.json. Skipping {nr} comparison.")