mindstudio-probe 1.0.3__py3-none-any.whl → 1.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/LICENSE +201 -201
- {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/METADATA +36 -34
- mindstudio_probe-1.1.0.dist-info/RECORD +287 -0
- {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/WHEEL +1 -1
- {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/entry_points.txt +1 -0
- msprobe/README.md +131 -237
- msprobe/__init__.py +16 -1
- msprobe/{config/config.json → config.json} +47 -49
- msprobe/core/advisor/advisor.py +124 -124
- msprobe/core/advisor/advisor_const.py +58 -59
- msprobe/core/advisor/advisor_result.py +58 -58
- msprobe/core/common/const.py +402 -318
- msprobe/core/common/exceptions.py +99 -99
- msprobe/core/common/{file_check.py → file_utils.py} +523 -283
- msprobe/core/common/inplace_op_checker.py +38 -0
- msprobe/core/common/inplace_ops.yaml +251 -0
- msprobe/core/common/log.py +86 -69
- msprobe/core/common/utils.py +371 -616
- msprobe/core/common_config.py +78 -71
- msprobe/core/compare/acc_compare.py +472 -298
- msprobe/core/compare/check.py +180 -95
- msprobe/core/compare/compare_cli.py +69 -49
- msprobe/core/compare/highlight.py +259 -222
- msprobe/core/compare/multiprocessing_compute.py +174 -149
- msprobe/core/compare/npy_compare.py +310 -295
- msprobe/core/compare/utils.py +464 -429
- msprobe/core/data_dump/data_collector.py +153 -144
- msprobe/core/data_dump/data_processor/base.py +337 -293
- msprobe/core/data_dump/data_processor/factory.py +76 -59
- msprobe/core/data_dump/data_processor/mindspore_processor.py +192 -198
- msprobe/core/data_dump/data_processor/pytorch_processor.py +383 -389
- msprobe/core/data_dump/json_writer.py +117 -116
- msprobe/core/data_dump/scope.py +194 -178
- msprobe/core/grad_probe/constant.py +74 -70
- msprobe/core/grad_probe/grad_compare.py +170 -175
- msprobe/core/grad_probe/utils.py +77 -52
- msprobe/docs/01.installation.md +99 -0
- msprobe/docs/02.config_introduction.md +137 -0
- msprobe/docs/03.config_examples.md +237 -0
- msprobe/docs/04.acl_config_examples.md +78 -0
- msprobe/docs/05.data_dump_PyTorch.md +326 -0
- msprobe/docs/06.data_dump_MindSpore.md +285 -0
- msprobe/docs/07.accuracy_checker_PyTorch.md +297 -0
- msprobe/docs/08.accuracy_checker_online_PyTorch.md +238 -0
- msprobe/docs/09.accuracy_checker_MindSpore.md +68 -0
- msprobe/docs/10.accuracy_compare_PyTorch.md +327 -0
- msprobe/docs/11.accuracy_compare_MindSpore.md +333 -0
- msprobe/docs/12.overflow_check_PyTorch.md +79 -0
- msprobe/docs/13.overflow_check_MindSpore.md +31 -0
- msprobe/{pytorch/doc/parse_tool.md → docs/14.data_parse_PyTorch.md} +283 -286
- msprobe/docs/15.free_benchmarking_PyTorch.md +170 -0
- msprobe/docs/16.free_benchmarking_MindSpore.md +140 -0
- msprobe/{doc/grad_probe/grad_probe.md → docs/17.grad_probe.md} +205 -207
- msprobe/{pytorch/doc//321/205/320/254/320/270/321/207/342/225/221/342/224/220/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/206/320/277/320/244/321/205/320/277/342/225/243.md → docs/18.online_dispatch.md} +89 -90
- msprobe/docs/FAQ.md +189 -0
- msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +146 -0
- msprobe/docs/img/free_benchmark_framework.png +0 -0
- msprobe/docs/img/ms_dump.png +0 -0
- msprobe/docs/img/ms_layer.png +0 -0
- msprobe/docs/img/pt_dump.png +0 -0
- msprobe/mindspore/__init__.py +2 -1
- msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +278 -245
- msprobe/mindspore/api_accuracy_checker/api_info.py +76 -69
- msprobe/mindspore/api_accuracy_checker/api_runner.py +155 -151
- msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +196 -196
- msprobe/mindspore/api_accuracy_checker/cmd_parser.py +6 -0
- msprobe/mindspore/api_accuracy_checker/compute_element.py +238 -223
- msprobe/mindspore/api_accuracy_checker/main.py +8 -15
- msprobe/mindspore/api_accuracy_checker/type_mapping.py +113 -113
- msprobe/mindspore/api_accuracy_checker/utils.py +79 -62
- msprobe/mindspore/cell_processor.py +58 -34
- msprobe/mindspore/common/const.py +108 -87
- msprobe/mindspore/common/log.py +37 -37
- msprobe/mindspore/common/utils.py +97 -57
- msprobe/mindspore/compare/distributed_compare.py +62 -75
- msprobe/mindspore/compare/layer_mapping.py +146 -0
- msprobe/mindspore/compare/modify_mapping.py +107 -0
- msprobe/mindspore/compare/ms_compare.py +357 -117
- msprobe/mindspore/compare/ms_graph_compare.py +364 -317
- msprobe/mindspore/compare/ms_to_pt_api.yaml +399 -399
- msprobe/mindspore/debugger/debugger_config.py +69 -74
- msprobe/mindspore/debugger/precision_debugger.py +150 -107
- msprobe/mindspore/dump/dump_tool_factory.py +50 -35
- msprobe/mindspore/dump/hook_cell/api_registry.py +128 -104
- msprobe/mindspore/dump/hook_cell/hook_cell.py +55 -53
- msprobe/mindspore/dump/hook_cell/primitive_hooks.py +206 -0
- msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +994 -925
- msprobe/mindspore/dump/hook_cell/wrap_api.py +121 -0
- msprobe/mindspore/dump/jit_dump.py +96 -56
- msprobe/mindspore/dump/kernel_graph_dump.py +75 -60
- msprobe/mindspore/dump/kernel_kbyk_dump.py +79 -65
- msprobe/mindspore/free_benchmark/api_pynative_self_check.py +131 -116
- msprobe/mindspore/free_benchmark/common/config.py +27 -12
- msprobe/mindspore/free_benchmark/common/handler_params.py +32 -17
- msprobe/mindspore/free_benchmark/common/utils.py +85 -71
- msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +842 -842
- msprobe/mindspore/free_benchmark/decorator/dec_forward.py +57 -42
- msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +122 -107
- msprobe/mindspore/free_benchmark/handler/base_handler.py +105 -90
- msprobe/mindspore/free_benchmark/handler/check_handler.py +56 -41
- msprobe/mindspore/free_benchmark/handler/fix_handler.py +51 -36
- msprobe/mindspore/free_benchmark/handler/handler_factory.py +36 -21
- msprobe/mindspore/free_benchmark/perturbation/add_noise.py +82 -67
- msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +36 -21
- msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +78 -63
- msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +77 -0
- msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +49 -34
- msprobe/mindspore/free_benchmark/perturbation/no_change.py +27 -12
- msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +44 -27
- msprobe/mindspore/free_benchmark/self_check_tool_factory.py +48 -33
- msprobe/mindspore/grad_probe/global_context.py +100 -91
- msprobe/mindspore/grad_probe/grad_analyzer.py +231 -231
- msprobe/mindspore/grad_probe/grad_monitor.py +27 -27
- msprobe/mindspore/grad_probe/grad_stat_csv.py +131 -131
- msprobe/mindspore/grad_probe/hook.py +94 -92
- msprobe/mindspore/grad_probe/utils.py +29 -28
- msprobe/mindspore/ms_config.py +128 -126
- msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +60 -45
- msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +49 -34
- msprobe/mindspore/runtime.py +4 -4
- msprobe/mindspore/service.py +297 -354
- msprobe/mindspore/task_handler_factory.py +24 -24
- msprobe/msprobe.py +105 -107
- msprobe/pytorch/__init__.py +23 -4
- msprobe/pytorch/api_accuracy_checker/common/config.py +70 -55
- msprobe/pytorch/api_accuracy_checker/common/utils.py +246 -165
- msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +230 -213
- msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +632 -581
- msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +132 -132
- msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +390 -390
- msprobe/pytorch/api_accuracy_checker/compare/compare.py +416 -381
- msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +90 -73
- msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +265 -244
- msprobe/pytorch/api_accuracy_checker/config.yaml +10 -10
- msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +370 -332
- msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +221 -199
- msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +150 -134
- msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +518 -581
- msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +213 -74
- msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +7 -4
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +218 -202
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +370 -324
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +227 -204
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py +110 -0
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +244 -218
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml +63 -0
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py +44 -0
- msprobe/pytorch/bench_functions/__init__.py +30 -15
- msprobe/pytorch/bench_functions/apply_adam_w.py +43 -28
- msprobe/pytorch/bench_functions/confusion_transpose.py +34 -19
- msprobe/pytorch/bench_functions/fast_gelu.py +70 -55
- msprobe/pytorch/bench_functions/layer_norm_eval.py +21 -6
- msprobe/pytorch/bench_functions/linear.py +27 -12
- msprobe/pytorch/bench_functions/matmul_backward.py +63 -48
- msprobe/pytorch/bench_functions/npu_fusion_attention.py +538 -421
- msprobe/pytorch/bench_functions/rms_norm.py +30 -15
- msprobe/pytorch/bench_functions/rotary_mul.py +71 -52
- msprobe/pytorch/bench_functions/scaled_mask_softmax.py +41 -26
- msprobe/pytorch/bench_functions/swiglu.py +70 -55
- msprobe/pytorch/common/__init__.py +17 -2
- msprobe/pytorch/common/compare_script.template +14 -14
- msprobe/pytorch/common/log.py +33 -32
- msprobe/pytorch/common/parse_json.py +54 -39
- msprobe/pytorch/common/utils.py +310 -300
- msprobe/pytorch/compare/distributed_compare.py +66 -66
- msprobe/pytorch/compare/mapping.yaml +607 -607
- msprobe/pytorch/compare/match.py +49 -33
- msprobe/pytorch/compare/pt_compare.py +82 -40
- msprobe/pytorch/debugger/debugger_config.py +108 -95
- msprobe/pytorch/debugger/precision_debugger.py +173 -125
- msprobe/pytorch/free_benchmark/__init__.py +23 -8
- msprobe/pytorch/free_benchmark/common/constant.py +70 -70
- msprobe/pytorch/free_benchmark/common/counter.py +71 -71
- msprobe/pytorch/free_benchmark/common/enums.py +65 -37
- msprobe/pytorch/free_benchmark/common/params.py +144 -129
- msprobe/pytorch/free_benchmark/common/utils.py +118 -102
- msprobe/pytorch/free_benchmark/compare/grad_saver.py +200 -179
- msprobe/pytorch/free_benchmark/compare/single_benchmark.py +119 -104
- msprobe/pytorch/free_benchmark/main.py +120 -105
- msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +28 -13
- msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +56 -41
- msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +105 -90
- msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +119 -104
- msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +87 -63
- msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +83 -68
- msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +43 -28
- msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +60 -45
- msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +34 -19
- msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +256 -217
- msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +54 -39
- msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +38 -23
- msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +45 -30
- msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +185 -170
- msprobe/pytorch/function_factory.py +91 -75
- msprobe/pytorch/functional/module_dump.py +84 -0
- msprobe/pytorch/grad_probe/grad_monitor.py +91 -90
- msprobe/pytorch/grad_probe/grad_stat_csv.py +128 -128
- msprobe/pytorch/hook_module/__init__.py +16 -1
- msprobe/pytorch/hook_module/api_registry.py +166 -161
- msprobe/pytorch/hook_module/hook_module.py +118 -120
- msprobe/pytorch/hook_module/support_wrap_ops.yaml +1879 -1877
- msprobe/pytorch/hook_module/utils.py +28 -29
- msprobe/pytorch/hook_module/wrap_aten.py +111 -110
- msprobe/pytorch/hook_module/wrap_distributed.py +77 -78
- msprobe/pytorch/hook_module/wrap_functional.py +104 -105
- msprobe/pytorch/hook_module/wrap_npu_custom.py +85 -84
- msprobe/pytorch/hook_module/wrap_tensor.py +69 -71
- msprobe/pytorch/hook_module/wrap_torch.py +84 -86
- msprobe/pytorch/hook_module/wrap_vf.py +60 -62
- msprobe/pytorch/module_processer.py +153 -138
- msprobe/pytorch/online_dispatch/__init__.py +20 -20
- msprobe/pytorch/online_dispatch/compare.py +235 -236
- msprobe/pytorch/online_dispatch/dispatch.py +271 -271
- msprobe/pytorch/online_dispatch/dump_compare.py +155 -156
- msprobe/pytorch/online_dispatch/single_compare.py +391 -391
- msprobe/pytorch/online_dispatch/torch_ops_config.yaml +57 -49
- msprobe/pytorch/online_dispatch/utils.py +127 -146
- msprobe/pytorch/parse.py +19 -4
- msprobe/pytorch/parse_tool/cli.py +31 -32
- msprobe/pytorch/parse_tool/lib/compare.py +259 -271
- msprobe/pytorch/parse_tool/lib/config.py +52 -52
- msprobe/pytorch/parse_tool/lib/file_desc.py +31 -31
- msprobe/pytorch/parse_tool/lib/interactive_cli.py +102 -102
- msprobe/pytorch/parse_tool/lib/parse_exception.py +54 -54
- msprobe/pytorch/parse_tool/lib/parse_tool.py +161 -158
- msprobe/pytorch/parse_tool/lib/utils.py +320 -321
- msprobe/pytorch/parse_tool/lib/visualization.py +85 -91
- msprobe/pytorch/pt_config.py +317 -187
- msprobe/pytorch/service.py +311 -252
- mindstudio_probe-1.0.3.dist-info/RECORD +0 -272
- msprobe/config/README.md +0 -539
- msprobe/mindspore/doc/compare.md +0 -58
- msprobe/mindspore/doc/dump.md +0 -217
- msprobe/mindspore/dump/hook_cell/wrap_functional.py +0 -91
- msprobe/mindspore/dump/hook_cell/wrap_tensor.py +0 -63
- msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/ssl_config.py +0 -10
- msprobe/pytorch/doc/FAQ.md +0 -193
- msprobe/pytorch/doc/api_accuracy_checker.md +0 -313
- msprobe/pytorch/doc/api_accuracy_checker_online.md +0 -187
- msprobe/pytorch/doc/dump.md +0 -260
- msprobe/pytorch/doc/msprobe/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/205/342/225/226/320/265/321/205/320/225/342/225/226/321/206/320/245/342/226/221/321/206/320/235/320/276dump/321/206/320/260/320/227/321/205/320/227/320/226/321/206/320/220/320/267/321/210/320/223/342/225/234/321/205/320/257/342/225/221/321/207/342/225/221/342/224/220/321/206/320/232/320/265/321/205/320/241/320/232.md +0 -182
- msprobe/pytorch/doc/ptdbg_ascend_compare.md +0 -240
- msprobe/pytorch/doc/ptdbg_ascend_overview.md +0 -68
- msprobe/pytorch/doc/ptdbg_ascend_quickstart.md +0 -381
- msprobe/pytorch/doc/run_overflow_check.md +0 -25
- msprobe/pytorch/doc//321/206/320/247/320/260/321/206/320/260/320/227/321/206/320/255/320/226/321/205/342/225/226/320/265/321/205/320/225/342/225/226/321/205/320/254/342/225/221/321/206/320/251/320/277/321/211/320/272/320/234/321/210/320/277/320/221/321/205/320/242/320/234/321/206/320/220/320/267/321/210/320/223/342/225/234/321/205/320/257/342/225/221/321/207/342/225/221/342/224/220/321/206/320/232/320/265/321/205/320/241/320/232.md +0 -151
- msprobe/pytorch/functional/data_processor.py +0 -0
- msprobe/pytorch/functional/dump_module.py +0 -39
- {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/top_level.txt +0 -0
- /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_1.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_2.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_3.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_4.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_1.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_2.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_3.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_4.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_5.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_6.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_7.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/GPT-3_8.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/YOLOV5S_1.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/YOLOV5S_2.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/accuracy_checking_details.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/accuracy_checking_result.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/api_precision_compare_details.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/api_precision_compare_result.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/auto_analyze_log.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/compare_result_pkl.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/compare_result_pkl_md5.png.png +0 -0
- /msprobe/{pytorch/doc → docs}/img/cpu_info.png +0 -0
- /msprobe/{config → docs}/img/free_benchmark.png +0 -0
- /msprobe/{doc/grad_probe/img/image-1.png → docs/img/grad_probe_image-1.png} +0 -0
- /msprobe/{doc/grad_probe/img/image-2.png → docs/img/grad_probe_image-2.png} +0 -0
- /msprobe/{doc/grad_probe/img/image-3.png → docs/img/grad_probe_image-3.png} +0 -0
- /msprobe/{doc/grad_probe/img/image-4.png → docs/img/grad_probe_image-4.png} +0 -0
- /msprobe/{doc/grad_probe/img/image.png → docs/img/grad_probe_image.png} +0 -0
- /msprobe/{pytorch/doc → docs}/img/module_compare.png +0 -0
|
@@ -1,271 +1,259 @@
|
|
|
1
|
-
#!/usr/bin/env python3
|
|
2
|
-
# -*- coding: utf-8 -*-
|
|
3
|
-
"""
|
|
4
|
-
# Copyright (C) 2022-2024. Huawei Technologies Co., Ltd. All rights reserved.
|
|
5
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
6
|
-
# you may not use this file except in compliance with the License.
|
|
7
|
-
# You may obtain a copy of the License at
|
|
8
|
-
#
|
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
-
#
|
|
11
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
-
# See the License for the specific language governing permissions and
|
|
15
|
-
# limitations under the License.
|
|
16
|
-
"""
|
|
17
|
-
|
|
18
|
-
import os
|
|
19
|
-
import time
|
|
20
|
-
import numpy as np
|
|
21
|
-
from collections import namedtuple
|
|
22
|
-
from msprobe.pytorch.parse_tool.lib.utils import Util
|
|
23
|
-
from msprobe.pytorch.parse_tool.lib.config import Const
|
|
24
|
-
from msprobe.pytorch.parse_tool.lib.parse_exception import ParseException
|
|
25
|
-
from msprobe.core.common.
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
self.
|
|
32
|
-
self.
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
self.
|
|
37
|
-
self.
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
self.util.
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
)
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
self.
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
self.util.
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
content
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
if
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
self.
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
self.
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
parts = file_name.split(".")
|
|
261
|
-
if len(parts) < 5:
|
|
262
|
-
continue
|
|
263
|
-
op_name = parts[1]
|
|
264
|
-
timestamp = parts[-1]
|
|
265
|
-
output_path = os.path.join(output_dir, op_name, timestamp)
|
|
266
|
-
self.convert_dump_to_npy(file_path, param, output_path, msaccucmp_path)
|
|
267
|
-
path_depth = root.count(os.sep)
|
|
268
|
-
if path_depth <= Const.MAX_TRAVERSAL_DEPTH:
|
|
269
|
-
yield root, _, files
|
|
270
|
-
else:
|
|
271
|
-
_[:] = []
|
|
1
|
+
#!/usr/bin/env python3
|
|
2
|
+
# -*- coding: utf-8 -*-
|
|
3
|
+
"""
|
|
4
|
+
# Copyright (C) 2022-2024. Huawei Technologies Co., Ltd. All rights reserved.
|
|
5
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
6
|
+
# you may not use this file except in compliance with the License.
|
|
7
|
+
# You may obtain a copy of the License at
|
|
8
|
+
#
|
|
9
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
+
#
|
|
11
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
+
# See the License for the specific language governing permissions and
|
|
15
|
+
# limitations under the License.
|
|
16
|
+
"""
|
|
17
|
+
|
|
18
|
+
import os
|
|
19
|
+
import time
|
|
20
|
+
import numpy as np
|
|
21
|
+
from collections import namedtuple
|
|
22
|
+
from msprobe.pytorch.parse_tool.lib.utils import Util
|
|
23
|
+
from msprobe.pytorch.parse_tool.lib.config import Const
|
|
24
|
+
from msprobe.pytorch.parse_tool.lib.parse_exception import ParseException
|
|
25
|
+
from msprobe.core.common.file_utils import create_directory, load_npy, save_npy_to_txt, write_csv
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
class Compare:
|
|
29
|
+
def __init__(self):
|
|
30
|
+
self.util = Util()
|
|
31
|
+
self.log = self.util.log
|
|
32
|
+
self.vector_compare_result = {}
|
|
33
|
+
|
|
34
|
+
def npu_vs_npu_compare(self, my_dump_path, golden_dump_path, result_dir, msaccucmp_path):
|
|
35
|
+
self.log.info("Start Compare ...............")
|
|
36
|
+
self.compare_vector(my_dump_path, golden_dump_path, result_dir, msaccucmp_path)
|
|
37
|
+
self.log.info("Compare finished!!")
|
|
38
|
+
|
|
39
|
+
def compare_vector(self, my_dump_path, golden_dump_path, result_dir, msaccucmp_path):
|
|
40
|
+
create_directory(result_dir)
|
|
41
|
+
self.util.check_path_valid(result_dir)
|
|
42
|
+
call_msaccucmp = self.util.check_msaccucmp(msaccucmp_path)
|
|
43
|
+
cmd = '%s %s compare -m %s -g %s -out %s' % (
|
|
44
|
+
self.util.python, call_msaccucmp, my_dump_path, golden_dump_path, result_dir
|
|
45
|
+
)
|
|
46
|
+
return self.util.execute_command(cmd)
|
|
47
|
+
|
|
48
|
+
def convert_dump_to_npy(self, dump_file, data_format, output, msaccucmp_path):
|
|
49
|
+
dump_file = self.util.path_strip(dump_file)
|
|
50
|
+
file_name = ""
|
|
51
|
+
if os.path.isfile(dump_file):
|
|
52
|
+
self.log.info("Covert file is: %s" % dump_file)
|
|
53
|
+
file_name = os.path.basename(dump_file)
|
|
54
|
+
elif os.path.isdir(dump_file):
|
|
55
|
+
self.log.info("Convert all files in path: %s" % dump_file)
|
|
56
|
+
file_name = ""
|
|
57
|
+
output = output if output else Const.DUMP_CONVERT_DIR
|
|
58
|
+
convert = self.convert(dump_file, data_format, output, msaccucmp_path)
|
|
59
|
+
if convert == 0:
|
|
60
|
+
convert_files = self.util.list_convert_files(output, file_name)
|
|
61
|
+
|
|
62
|
+
summary_txt = ["SrcFile: %s" % dump_file]
|
|
63
|
+
for convert_file in convert_files.values():
|
|
64
|
+
summary_txt.append(" - %s" % convert_file.file_name)
|
|
65
|
+
self.log.info("Transfer result is saved in : %s" % os.path.realpath(output))
|
|
66
|
+
self.util.print_panel("\n".join(summary_txt))
|
|
67
|
+
|
|
68
|
+
def convert(self, dump_file, data_format, output, msaccucmp_path):
|
|
69
|
+
create_directory(output)
|
|
70
|
+
self.util.check_path_valid(output)
|
|
71
|
+
call_msaccucmp = self.util.check_msaccucmp(msaccucmp_path)
|
|
72
|
+
if data_format:
|
|
73
|
+
cmd = '%s %s convert -d %s -out %s -f %s' % (
|
|
74
|
+
self.util.python, call_msaccucmp, dump_file, output, data_format
|
|
75
|
+
)
|
|
76
|
+
else:
|
|
77
|
+
cmd = '%s %s convert -d %s -out %s' % (
|
|
78
|
+
self.util.python, call_msaccucmp, dump_file, output
|
|
79
|
+
)
|
|
80
|
+
return self.util.execute_command(cmd)
|
|
81
|
+
|
|
82
|
+
def compare_data(self, args):
|
|
83
|
+
"""Compare data"""
|
|
84
|
+
(left, right, save_txt, rl, al, diff_count) = args
|
|
85
|
+
if left is None or right is None:
|
|
86
|
+
raise ParseException("invalid input or output")
|
|
87
|
+
if self.util.check_path_valid(left) and self.util.check_path_valid(right):
|
|
88
|
+
left_data = load_npy(left)
|
|
89
|
+
right_data = load_npy(right)
|
|
90
|
+
# save to txt
|
|
91
|
+
if save_txt:
|
|
92
|
+
save_npy_to_txt(left_data, left + ".txt")
|
|
93
|
+
save_npy_to_txt(right_data, right + ".txt")
|
|
94
|
+
# compare data
|
|
95
|
+
(total_cnt, all_close, cos_sim, err_percent) = self.do_compare_data(left_data, right_data, rl, al, diff_count)
|
|
96
|
+
content = ['Left:', ' ├─ NpyFile: %s' % left]
|
|
97
|
+
if save_txt:
|
|
98
|
+
content.append(' ├─ TxtFile: [green]%s.txt[/green]' % left)
|
|
99
|
+
content.append(' └─ NpySpec: [yellow]%s[/yellow]' % self.util.gen_npy_info_txt(left_data))
|
|
100
|
+
content.append('Right:')
|
|
101
|
+
content.append(' ├─ NpyFile: %s' % right)
|
|
102
|
+
if save_txt:
|
|
103
|
+
content.append(' ├─ TxtFile: [green]%s.txt[/green]' % right)
|
|
104
|
+
content.append(' └─ NpySpec: [yellow]%s[/yellow]' % self.util.gen_npy_info_txt(right_data))
|
|
105
|
+
content.append('NumCnt: %s' % total_cnt)
|
|
106
|
+
content.append('AllClose: %s' % all_close)
|
|
107
|
+
content.append('CosSim: %s' % cos_sim)
|
|
108
|
+
content.append('ErrorPer: %s (rl= %s, al= %s)' % (err_percent, rl, al))
|
|
109
|
+
self.util.print_panel("\n".join(content))
|
|
110
|
+
|
|
111
|
+
def do_compare_data(self, left, right, rl=0.001, al=0.001, diff_count=20):
|
|
112
|
+
data_left = left.astype(np.float32)
|
|
113
|
+
data_right = right.astype(np.float32)
|
|
114
|
+
shape_left = data_left.shape
|
|
115
|
+
shape_right = data_right.shape
|
|
116
|
+
if shape_left != shape_right:
|
|
117
|
+
self.log.warning("Data shape not equal: %s vs %s" % (data_left.shape, data_right.shape))
|
|
118
|
+
data_left = data_left.reshape(-1)
|
|
119
|
+
data_right = data_right.reshape(-1)
|
|
120
|
+
if data_left.shape[0] != data_right.shape[0]:
|
|
121
|
+
self.log.warning("Data size not equal: %s vs %s" % (data_left.shape, data_right.shape))
|
|
122
|
+
if data_left.shape[0] < data_right.shape[0]:
|
|
123
|
+
data_left = np.pad(data_left, (0, data_right.shape[0] - data_left.shape[0]), 'constant')
|
|
124
|
+
else:
|
|
125
|
+
data_right = np.pad(data_right, (0, data_left.shape[0] - data_right.shape[0]), 'constant')
|
|
126
|
+
all_close = np.allclose(data_left, data_right, atol=al, rtol=rl)
|
|
127
|
+
np.seterr(divide='raise')
|
|
128
|
+
cos_sim = np.dot(data_left, data_right) / (
|
|
129
|
+
np.sqrt(np.dot(data_left, data_left)) * np.sqrt(np.dot(data_right, data_right)))
|
|
130
|
+
err_cnt = 0
|
|
131
|
+
total_cnt = data_left.shape[0]
|
|
132
|
+
diff_table_columns = ['Index', 'Left', 'Right', 'Diff']
|
|
133
|
+
err_table = self.util.create_table("Error Item Table", diff_table_columns)
|
|
134
|
+
top_table = self.util.create_table("Top Item Table", diff_table_columns)
|
|
135
|
+
for i in range(total_cnt):
|
|
136
|
+
abs_diff = abs(data_left[i] - data_right[i])
|
|
137
|
+
if i < diff_count:
|
|
138
|
+
top_table.add_row(str(i), str(data_left[i]), str(data_right[i]), str(abs_diff))
|
|
139
|
+
if abs_diff > (al + rl * abs(data_right[i])):
|
|
140
|
+
if err_cnt < diff_count:
|
|
141
|
+
err_table.add_row(str(i), str(data_left[i]), str(data_right[i]), str(abs_diff))
|
|
142
|
+
err_cnt += 1
|
|
143
|
+
if total_cnt == 0:
|
|
144
|
+
err_percent = float(0)
|
|
145
|
+
else:
|
|
146
|
+
err_percent = float(err_cnt / total_cnt)
|
|
147
|
+
self.util.print(self.util.create_columns([err_table, top_table]))
|
|
148
|
+
do_compare_data_result = namedtuple('do_compare_data_result', ['cnt', 'close', 'cos', 'err'])
|
|
149
|
+
res = do_compare_data_result(total_cnt, all_close, cos_sim, err_percent)
|
|
150
|
+
return res
|
|
151
|
+
|
|
152
|
+
def compare_npy(self, file, bench_file, output_path):
|
|
153
|
+
if self.util.check_path_valid(file) and self.util.check_path_valid(bench_file):
|
|
154
|
+
data = load_npy(file)
|
|
155
|
+
bench_data = load_npy(bench_file)
|
|
156
|
+
shape, dtype = data.shape, data.dtype
|
|
157
|
+
bench_shape, bench_dtype = bench_data.shape, bench_data.dtype
|
|
158
|
+
filename = os.path.basename(file)
|
|
159
|
+
bench_filename = os.path.basename(bench_file)
|
|
160
|
+
if shape != bench_shape or dtype != bench_dtype:
|
|
161
|
+
self.log.error(
|
|
162
|
+
"Shape or dtype between two npy files is inconsistent. Please check the two files."
|
|
163
|
+
"File 1: %s, file 2: %s" % (file, bench_file))
|
|
164
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
165
|
+
return
|
|
166
|
+
md5_consistency = False
|
|
167
|
+
if self.util.get_md5_for_numpy(data) == self.util.get_md5_for_numpy(bench_data):
|
|
168
|
+
md5_consistency = True
|
|
169
|
+
data_mean = np.mean(data)
|
|
170
|
+
bench_data_mean = np.mean(bench_data)
|
|
171
|
+
abs_error = np.abs(data - bench_data)
|
|
172
|
+
bench_data = self.util.deal_with_value_if_has_zero(bench_data)
|
|
173
|
+
rel_error = np.abs(abs_error / bench_data)
|
|
174
|
+
abs_diff_max = abs_error.max()
|
|
175
|
+
rel_diff_max = np.max(rel_error)
|
|
176
|
+
compare_result = [[filename, bench_filename, data_mean, bench_data_mean, md5_consistency, abs_diff_max,
|
|
177
|
+
rel_diff_max]]
|
|
178
|
+
write_csv(compare_result, output_path)
|
|
179
|
+
|
|
180
|
+
def compare_all_file_in_directory(self, my_dump_dir, golden_dump_dir, output_path):
|
|
181
|
+
if not (self.util.is_subdir_count_equal(my_dump_dir, golden_dump_dir)
|
|
182
|
+
and self.util.check_npy_files_valid_in_dir(my_dump_dir)
|
|
183
|
+
and self.util.check_npy_files_valid_in_dir(golden_dump_dir)):
|
|
184
|
+
self.log.error(
|
|
185
|
+
"Top level(Npy files level) directory structure is inconsistent. Please check the two directory.")
|
|
186
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
187
|
+
return
|
|
188
|
+
my_npy_files = self.util.get_sorted_files_names(my_dump_dir)
|
|
189
|
+
golden_npy_files = self.util.get_sorted_files_names(golden_dump_dir)
|
|
190
|
+
for my_npy_file_name, golden_npy_file_name in zip(my_npy_files, golden_npy_files):
|
|
191
|
+
my_npy_path = os.path.join(my_dump_dir, my_npy_file_name)
|
|
192
|
+
golden_npy_path = os.path.join(golden_dump_dir, golden_npy_file_name)
|
|
193
|
+
self.compare_npy(my_npy_path, golden_npy_path, output_path)
|
|
194
|
+
|
|
195
|
+
def compare_timestamp_directory(self, my_dump_dir, golden_dump_dir, output_path):
|
|
196
|
+
if not self.util.is_subdir_count_equal(my_dump_dir, golden_dump_dir):
|
|
197
|
+
self.log.error(
|
|
198
|
+
"Second level(Timestamp level) directory structure is inconsistent. Please check the two directory.")
|
|
199
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
200
|
+
return
|
|
201
|
+
my_ordered_subdirs = self.util.get_sorted_subdirectories_names(my_dump_dir)
|
|
202
|
+
golden_ordered_subdirs = self.util.get_sorted_subdirectories_names(golden_dump_dir)
|
|
203
|
+
for my_subdir_name, golden_subdir_name in zip(my_ordered_subdirs, golden_ordered_subdirs):
|
|
204
|
+
my_subdir_path = os.path.join(my_dump_dir, my_subdir_name)
|
|
205
|
+
golden_subdir_path = os.path.join(golden_dump_dir, golden_subdir_name)
|
|
206
|
+
self.compare_all_file_in_directory(my_subdir_path, golden_subdir_path, output_path)
|
|
207
|
+
|
|
208
|
+
def compare_converted_dir(self, my_dump_dir, golden_dump_dir, output_dir):
|
|
209
|
+
if not self.util.is_subdir_count_equal(my_dump_dir, golden_dump_dir):
|
|
210
|
+
self.log.error(
|
|
211
|
+
"Top level(Opname level) directory structure is inconsistent. Please check the two directory.")
|
|
212
|
+
return
|
|
213
|
+
timestamp = int(time.time())
|
|
214
|
+
output_file_name = f"batch_compare_{timestamp}.csv"
|
|
215
|
+
output_path = os.path.join(output_dir, output_file_name)
|
|
216
|
+
title_rows = [[
|
|
217
|
+
"NPU File Name",
|
|
218
|
+
"Bench File Name",
|
|
219
|
+
"Mean",
|
|
220
|
+
"Bench Mean",
|
|
221
|
+
"Md5 Consistency",
|
|
222
|
+
"Max Abs Error",
|
|
223
|
+
"Max Relative Error"
|
|
224
|
+
]]
|
|
225
|
+
write_csv(title_rows, output_path)
|
|
226
|
+
|
|
227
|
+
my_ordered_subdirs = self.util.get_sorted_subdirectories_names(my_dump_dir)
|
|
228
|
+
golden_ordered_subdirs = self.util.get_sorted_subdirectories_names(golden_dump_dir)
|
|
229
|
+
for my_subdir_name, golden_subdir_name in zip(my_ordered_subdirs, golden_ordered_subdirs):
|
|
230
|
+
if not my_subdir_name == golden_subdir_name:
|
|
231
|
+
self.log.error(
|
|
232
|
+
"Top level(Opname level) directory structure is inconsistent. Please check the two directory.")
|
|
233
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
234
|
+
return
|
|
235
|
+
my_subdir_path = os.path.join(my_dump_dir, my_subdir_name)
|
|
236
|
+
golden_subdir_path = os.path.join(golden_dump_dir, golden_subdir_name)
|
|
237
|
+
self.compare_timestamp_directory(my_subdir_path, golden_subdir_path, output_path)
|
|
238
|
+
self.util.change_filemode_safe(output_path)
|
|
239
|
+
self.log.info("Compare result is saved in : %s" % (output_path))
|
|
240
|
+
|
|
241
|
+
def convert_api_dir_to_npy(self, dump_dir, param, output_dir, msaccucmp_path):
|
|
242
|
+
dump_dir = self.util.path_strip(dump_dir)
|
|
243
|
+
for root, _, files in os.walk(dump_dir, topdown=True):
|
|
244
|
+
self.util.check_path_valid(root)
|
|
245
|
+
for file in files:
|
|
246
|
+
file_path = os.path.join(root, file)
|
|
247
|
+
file_name = os.path.basename(file_path)
|
|
248
|
+
parts = file_name.split(".")
|
|
249
|
+
if len(parts) < 5:
|
|
250
|
+
continue
|
|
251
|
+
op_name = parts[1]
|
|
252
|
+
timestamp = parts[-1]
|
|
253
|
+
output_path = os.path.join(output_dir, op_name, timestamp)
|
|
254
|
+
self.convert_dump_to_npy(file_path, param, output_path, msaccucmp_path)
|
|
255
|
+
path_depth = root.count(os.sep)
|
|
256
|
+
if path_depth <= Const.MAX_TRAVERSAL_DEPTH:
|
|
257
|
+
yield root, _, files
|
|
258
|
+
else:
|
|
259
|
+
_[:] = []
|