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