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