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.
Files changed (278) hide show
  1. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/LICENSE +201 -201
  2. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/METADATA +36 -34
  3. mindstudio_probe-1.1.0.dist-info/RECORD +287 -0
  4. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/WHEEL +1 -1
  5. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/entry_points.txt +1 -0
  6. msprobe/README.md +131 -237
  7. msprobe/__init__.py +16 -1
  8. msprobe/{config/config.json → config.json} +47 -49
  9. msprobe/core/advisor/advisor.py +124 -124
  10. msprobe/core/advisor/advisor_const.py +58 -59
  11. msprobe/core/advisor/advisor_result.py +58 -58
  12. msprobe/core/common/const.py +402 -318
  13. msprobe/core/common/exceptions.py +99 -99
  14. msprobe/core/common/{file_check.py → file_utils.py} +523 -283
  15. msprobe/core/common/inplace_op_checker.py +38 -0
  16. msprobe/core/common/inplace_ops.yaml +251 -0
  17. msprobe/core/common/log.py +86 -69
  18. msprobe/core/common/utils.py +371 -616
  19. msprobe/core/common_config.py +78 -71
  20. msprobe/core/compare/acc_compare.py +472 -298
  21. msprobe/core/compare/check.py +180 -95
  22. msprobe/core/compare/compare_cli.py +69 -49
  23. msprobe/core/compare/highlight.py +259 -222
  24. msprobe/core/compare/multiprocessing_compute.py +174 -149
  25. msprobe/core/compare/npy_compare.py +310 -295
  26. msprobe/core/compare/utils.py +464 -429
  27. msprobe/core/data_dump/data_collector.py +153 -144
  28. msprobe/core/data_dump/data_processor/base.py +337 -293
  29. msprobe/core/data_dump/data_processor/factory.py +76 -59
  30. msprobe/core/data_dump/data_processor/mindspore_processor.py +192 -198
  31. msprobe/core/data_dump/data_processor/pytorch_processor.py +383 -389
  32. msprobe/core/data_dump/json_writer.py +117 -116
  33. msprobe/core/data_dump/scope.py +194 -178
  34. msprobe/core/grad_probe/constant.py +74 -70
  35. msprobe/core/grad_probe/grad_compare.py +170 -175
  36. msprobe/core/grad_probe/utils.py +77 -52
  37. msprobe/docs/01.installation.md +99 -0
  38. msprobe/docs/02.config_introduction.md +137 -0
  39. msprobe/docs/03.config_examples.md +237 -0
  40. msprobe/docs/04.acl_config_examples.md +78 -0
  41. msprobe/docs/05.data_dump_PyTorch.md +326 -0
  42. msprobe/docs/06.data_dump_MindSpore.md +285 -0
  43. msprobe/docs/07.accuracy_checker_PyTorch.md +297 -0
  44. msprobe/docs/08.accuracy_checker_online_PyTorch.md +238 -0
  45. msprobe/docs/09.accuracy_checker_MindSpore.md +68 -0
  46. msprobe/docs/10.accuracy_compare_PyTorch.md +327 -0
  47. msprobe/docs/11.accuracy_compare_MindSpore.md +333 -0
  48. msprobe/docs/12.overflow_check_PyTorch.md +79 -0
  49. msprobe/docs/13.overflow_check_MindSpore.md +31 -0
  50. msprobe/{pytorch/doc/parse_tool.md → docs/14.data_parse_PyTorch.md} +283 -286
  51. msprobe/docs/15.free_benchmarking_PyTorch.md +170 -0
  52. msprobe/docs/16.free_benchmarking_MindSpore.md +140 -0
  53. msprobe/{doc/grad_probe/grad_probe.md → docs/17.grad_probe.md} +205 -207
  54. 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
  55. msprobe/docs/FAQ.md +189 -0
  56. msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +146 -0
  57. msprobe/docs/img/free_benchmark_framework.png +0 -0
  58. msprobe/docs/img/ms_dump.png +0 -0
  59. msprobe/docs/img/ms_layer.png +0 -0
  60. msprobe/docs/img/pt_dump.png +0 -0
  61. msprobe/mindspore/__init__.py +2 -1
  62. msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +278 -245
  63. msprobe/mindspore/api_accuracy_checker/api_info.py +76 -69
  64. msprobe/mindspore/api_accuracy_checker/api_runner.py +155 -151
  65. msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +196 -196
  66. msprobe/mindspore/api_accuracy_checker/cmd_parser.py +6 -0
  67. msprobe/mindspore/api_accuracy_checker/compute_element.py +238 -223
  68. msprobe/mindspore/api_accuracy_checker/main.py +8 -15
  69. msprobe/mindspore/api_accuracy_checker/type_mapping.py +113 -113
  70. msprobe/mindspore/api_accuracy_checker/utils.py +79 -62
  71. msprobe/mindspore/cell_processor.py +58 -34
  72. msprobe/mindspore/common/const.py +108 -87
  73. msprobe/mindspore/common/log.py +37 -37
  74. msprobe/mindspore/common/utils.py +97 -57
  75. msprobe/mindspore/compare/distributed_compare.py +62 -75
  76. msprobe/mindspore/compare/layer_mapping.py +146 -0
  77. msprobe/mindspore/compare/modify_mapping.py +107 -0
  78. msprobe/mindspore/compare/ms_compare.py +357 -117
  79. msprobe/mindspore/compare/ms_graph_compare.py +364 -317
  80. msprobe/mindspore/compare/ms_to_pt_api.yaml +399 -399
  81. msprobe/mindspore/debugger/debugger_config.py +69 -74
  82. msprobe/mindspore/debugger/precision_debugger.py +150 -107
  83. msprobe/mindspore/dump/dump_tool_factory.py +50 -35
  84. msprobe/mindspore/dump/hook_cell/api_registry.py +128 -104
  85. msprobe/mindspore/dump/hook_cell/hook_cell.py +55 -53
  86. msprobe/mindspore/dump/hook_cell/primitive_hooks.py +206 -0
  87. msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +994 -925
  88. msprobe/mindspore/dump/hook_cell/wrap_api.py +121 -0
  89. msprobe/mindspore/dump/jit_dump.py +96 -56
  90. msprobe/mindspore/dump/kernel_graph_dump.py +75 -60
  91. msprobe/mindspore/dump/kernel_kbyk_dump.py +79 -65
  92. msprobe/mindspore/free_benchmark/api_pynative_self_check.py +131 -116
  93. msprobe/mindspore/free_benchmark/common/config.py +27 -12
  94. msprobe/mindspore/free_benchmark/common/handler_params.py +32 -17
  95. msprobe/mindspore/free_benchmark/common/utils.py +85 -71
  96. msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +842 -842
  97. msprobe/mindspore/free_benchmark/decorator/dec_forward.py +57 -42
  98. msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +122 -107
  99. msprobe/mindspore/free_benchmark/handler/base_handler.py +105 -90
  100. msprobe/mindspore/free_benchmark/handler/check_handler.py +56 -41
  101. msprobe/mindspore/free_benchmark/handler/fix_handler.py +51 -36
  102. msprobe/mindspore/free_benchmark/handler/handler_factory.py +36 -21
  103. msprobe/mindspore/free_benchmark/perturbation/add_noise.py +82 -67
  104. msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +36 -21
  105. msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +78 -63
  106. msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +77 -0
  107. msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +49 -34
  108. msprobe/mindspore/free_benchmark/perturbation/no_change.py +27 -12
  109. msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +44 -27
  110. msprobe/mindspore/free_benchmark/self_check_tool_factory.py +48 -33
  111. msprobe/mindspore/grad_probe/global_context.py +100 -91
  112. msprobe/mindspore/grad_probe/grad_analyzer.py +231 -231
  113. msprobe/mindspore/grad_probe/grad_monitor.py +27 -27
  114. msprobe/mindspore/grad_probe/grad_stat_csv.py +131 -131
  115. msprobe/mindspore/grad_probe/hook.py +94 -92
  116. msprobe/mindspore/grad_probe/utils.py +29 -28
  117. msprobe/mindspore/ms_config.py +128 -126
  118. msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +60 -45
  119. msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +49 -34
  120. msprobe/mindspore/runtime.py +4 -4
  121. msprobe/mindspore/service.py +297 -354
  122. msprobe/mindspore/task_handler_factory.py +24 -24
  123. msprobe/msprobe.py +105 -107
  124. msprobe/pytorch/__init__.py +23 -4
  125. msprobe/pytorch/api_accuracy_checker/common/config.py +70 -55
  126. msprobe/pytorch/api_accuracy_checker/common/utils.py +246 -165
  127. msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +230 -213
  128. msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +632 -581
  129. msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +132 -132
  130. msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +390 -390
  131. msprobe/pytorch/api_accuracy_checker/compare/compare.py +416 -381
  132. msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +90 -73
  133. msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +265 -244
  134. msprobe/pytorch/api_accuracy_checker/config.yaml +10 -10
  135. msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +370 -332
  136. msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +221 -199
  137. msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +150 -134
  138. msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +518 -581
  139. msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +213 -74
  140. msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +7 -4
  141. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +218 -202
  142. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +370 -324
  143. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +227 -204
  144. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py +110 -0
  145. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +244 -218
  146. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml +63 -0
  147. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py +44 -0
  148. msprobe/pytorch/bench_functions/__init__.py +30 -15
  149. msprobe/pytorch/bench_functions/apply_adam_w.py +43 -28
  150. msprobe/pytorch/bench_functions/confusion_transpose.py +34 -19
  151. msprobe/pytorch/bench_functions/fast_gelu.py +70 -55
  152. msprobe/pytorch/bench_functions/layer_norm_eval.py +21 -6
  153. msprobe/pytorch/bench_functions/linear.py +27 -12
  154. msprobe/pytorch/bench_functions/matmul_backward.py +63 -48
  155. msprobe/pytorch/bench_functions/npu_fusion_attention.py +538 -421
  156. msprobe/pytorch/bench_functions/rms_norm.py +30 -15
  157. msprobe/pytorch/bench_functions/rotary_mul.py +71 -52
  158. msprobe/pytorch/bench_functions/scaled_mask_softmax.py +41 -26
  159. msprobe/pytorch/bench_functions/swiglu.py +70 -55
  160. msprobe/pytorch/common/__init__.py +17 -2
  161. msprobe/pytorch/common/compare_script.template +14 -14
  162. msprobe/pytorch/common/log.py +33 -32
  163. msprobe/pytorch/common/parse_json.py +54 -39
  164. msprobe/pytorch/common/utils.py +310 -300
  165. msprobe/pytorch/compare/distributed_compare.py +66 -66
  166. msprobe/pytorch/compare/mapping.yaml +607 -607
  167. msprobe/pytorch/compare/match.py +49 -33
  168. msprobe/pytorch/compare/pt_compare.py +82 -40
  169. msprobe/pytorch/debugger/debugger_config.py +108 -95
  170. msprobe/pytorch/debugger/precision_debugger.py +173 -125
  171. msprobe/pytorch/free_benchmark/__init__.py +23 -8
  172. msprobe/pytorch/free_benchmark/common/constant.py +70 -70
  173. msprobe/pytorch/free_benchmark/common/counter.py +71 -71
  174. msprobe/pytorch/free_benchmark/common/enums.py +65 -37
  175. msprobe/pytorch/free_benchmark/common/params.py +144 -129
  176. msprobe/pytorch/free_benchmark/common/utils.py +118 -102
  177. msprobe/pytorch/free_benchmark/compare/grad_saver.py +200 -179
  178. msprobe/pytorch/free_benchmark/compare/single_benchmark.py +119 -104
  179. msprobe/pytorch/free_benchmark/main.py +120 -105
  180. msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +28 -13
  181. msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +56 -41
  182. msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +105 -90
  183. msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +119 -104
  184. msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +87 -63
  185. msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +83 -68
  186. msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +43 -28
  187. msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +60 -45
  188. msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +34 -19
  189. msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +256 -217
  190. msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +54 -39
  191. msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +38 -23
  192. msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +45 -30
  193. msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +185 -170
  194. msprobe/pytorch/function_factory.py +91 -75
  195. msprobe/pytorch/functional/module_dump.py +84 -0
  196. msprobe/pytorch/grad_probe/grad_monitor.py +91 -90
  197. msprobe/pytorch/grad_probe/grad_stat_csv.py +128 -128
  198. msprobe/pytorch/hook_module/__init__.py +16 -1
  199. msprobe/pytorch/hook_module/api_registry.py +166 -161
  200. msprobe/pytorch/hook_module/hook_module.py +118 -120
  201. msprobe/pytorch/hook_module/support_wrap_ops.yaml +1879 -1877
  202. msprobe/pytorch/hook_module/utils.py +28 -29
  203. msprobe/pytorch/hook_module/wrap_aten.py +111 -110
  204. msprobe/pytorch/hook_module/wrap_distributed.py +77 -78
  205. msprobe/pytorch/hook_module/wrap_functional.py +104 -105
  206. msprobe/pytorch/hook_module/wrap_npu_custom.py +85 -84
  207. msprobe/pytorch/hook_module/wrap_tensor.py +69 -71
  208. msprobe/pytorch/hook_module/wrap_torch.py +84 -86
  209. msprobe/pytorch/hook_module/wrap_vf.py +60 -62
  210. msprobe/pytorch/module_processer.py +153 -138
  211. msprobe/pytorch/online_dispatch/__init__.py +20 -20
  212. msprobe/pytorch/online_dispatch/compare.py +235 -236
  213. msprobe/pytorch/online_dispatch/dispatch.py +271 -271
  214. msprobe/pytorch/online_dispatch/dump_compare.py +155 -156
  215. msprobe/pytorch/online_dispatch/single_compare.py +391 -391
  216. msprobe/pytorch/online_dispatch/torch_ops_config.yaml +57 -49
  217. msprobe/pytorch/online_dispatch/utils.py +127 -146
  218. msprobe/pytorch/parse.py +19 -4
  219. msprobe/pytorch/parse_tool/cli.py +31 -32
  220. msprobe/pytorch/parse_tool/lib/compare.py +259 -271
  221. msprobe/pytorch/parse_tool/lib/config.py +52 -52
  222. msprobe/pytorch/parse_tool/lib/file_desc.py +31 -31
  223. msprobe/pytorch/parse_tool/lib/interactive_cli.py +102 -102
  224. msprobe/pytorch/parse_tool/lib/parse_exception.py +54 -54
  225. msprobe/pytorch/parse_tool/lib/parse_tool.py +161 -158
  226. msprobe/pytorch/parse_tool/lib/utils.py +320 -321
  227. msprobe/pytorch/parse_tool/lib/visualization.py +85 -91
  228. msprobe/pytorch/pt_config.py +317 -187
  229. msprobe/pytorch/service.py +311 -252
  230. mindstudio_probe-1.0.3.dist-info/RECORD +0 -272
  231. msprobe/config/README.md +0 -539
  232. msprobe/mindspore/doc/compare.md +0 -58
  233. msprobe/mindspore/doc/dump.md +0 -217
  234. msprobe/mindspore/dump/hook_cell/wrap_functional.py +0 -91
  235. msprobe/mindspore/dump/hook_cell/wrap_tensor.py +0 -63
  236. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/ssl_config.py +0 -10
  237. msprobe/pytorch/doc/FAQ.md +0 -193
  238. msprobe/pytorch/doc/api_accuracy_checker.md +0 -313
  239. msprobe/pytorch/doc/api_accuracy_checker_online.md +0 -187
  240. msprobe/pytorch/doc/dump.md +0 -260
  241. 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
  242. msprobe/pytorch/doc/ptdbg_ascend_compare.md +0 -240
  243. msprobe/pytorch/doc/ptdbg_ascend_overview.md +0 -68
  244. msprobe/pytorch/doc/ptdbg_ascend_quickstart.md +0 -381
  245. msprobe/pytorch/doc/run_overflow_check.md +0 -25
  246. 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
  247. msprobe/pytorch/functional/data_processor.py +0 -0
  248. msprobe/pytorch/functional/dump_module.py +0 -39
  249. {mindstudio_probe-1.0.3.dist-info → mindstudio_probe-1.1.0.dist-info}/top_level.txt +0 -0
  250. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_1.png +0 -0
  251. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_2.png +0 -0
  252. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_3.png +0 -0
  253. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_4.png +0 -0
  254. /msprobe/{pytorch/doc → docs}/img/GPT-3_1.png +0 -0
  255. /msprobe/{pytorch/doc → docs}/img/GPT-3_2.png +0 -0
  256. /msprobe/{pytorch/doc → docs}/img/GPT-3_3.png +0 -0
  257. /msprobe/{pytorch/doc → docs}/img/GPT-3_4.png +0 -0
  258. /msprobe/{pytorch/doc → docs}/img/GPT-3_5.png +0 -0
  259. /msprobe/{pytorch/doc → docs}/img/GPT-3_6.png +0 -0
  260. /msprobe/{pytorch/doc → docs}/img/GPT-3_7.png +0 -0
  261. /msprobe/{pytorch/doc → docs}/img/GPT-3_8.png +0 -0
  262. /msprobe/{pytorch/doc → docs}/img/YOLOV5S_1.png +0 -0
  263. /msprobe/{pytorch/doc → docs}/img/YOLOV5S_2.png +0 -0
  264. /msprobe/{pytorch/doc → docs}/img/accuracy_checking_details.png +0 -0
  265. /msprobe/{pytorch/doc → docs}/img/accuracy_checking_result.png +0 -0
  266. /msprobe/{pytorch/doc → docs}/img/api_precision_compare_details.png +0 -0
  267. /msprobe/{pytorch/doc → docs}/img/api_precision_compare_result.png +0 -0
  268. /msprobe/{pytorch/doc → docs}/img/auto_analyze_log.png +0 -0
  269. /msprobe/{pytorch/doc → docs}/img/compare_result_pkl.png +0 -0
  270. /msprobe/{pytorch/doc → docs}/img/compare_result_pkl_md5.png.png +0 -0
  271. /msprobe/{pytorch/doc → docs}/img/cpu_info.png +0 -0
  272. /msprobe/{config → docs}/img/free_benchmark.png +0 -0
  273. /msprobe/{doc/grad_probe/img/image-1.png → docs/img/grad_probe_image-1.png} +0 -0
  274. /msprobe/{doc/grad_probe/img/image-2.png → docs/img/grad_probe_image-2.png} +0 -0
  275. /msprobe/{doc/grad_probe/img/image-3.png → docs/img/grad_probe_image-3.png} +0 -0
  276. /msprobe/{doc/grad_probe/img/image-4.png → docs/img/grad_probe_image-4.png} +0 -0
  277. /msprobe/{doc/grad_probe/img/image.png → docs/img/grad_probe_image.png} +0 -0
  278. /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.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 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
+ _[:] = []