mindstudio-probe 1.1.0__tar.gz → 1.2.1__tar.gz

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 (430) hide show
  1. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/MANIFEST.in +1 -0
  2. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/PKG-INFO +16 -4
  3. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/PKG-INFO +16 -4
  4. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/SOURCES.txt +119 -10
  5. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/entry_points.txt +0 -1
  6. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/requires.txt +2 -0
  7. mindstudio_probe-1.2.1/msprobe/CMakeLists.txt +5 -0
  8. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/README.md +51 -20
  9. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/config.json +2 -3
  10. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/advisor/advisor.py +8 -3
  11. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/const.py +264 -15
  12. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/exceptions.py +27 -3
  13. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/file_utils.py +176 -26
  14. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/inplace_op_checker.py +15 -0
  15. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/inplace_ops.yaml +3 -0
  16. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/log.py +27 -9
  17. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/utils.py +204 -77
  18. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common_config.py +49 -14
  19. mindstudio_probe-1.2.1/msprobe/core/compare/acc_compare.py +549 -0
  20. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/compare/check.py +32 -33
  21. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/compare/compare_cli.py +32 -14
  22. mindstudio_probe-1.2.1/msprobe/core/compare/highlight.py +415 -0
  23. mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/__init__.py +19 -0
  24. mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/data_scope_parser.py +246 -0
  25. mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/layer_mapping.py +249 -0
  26. mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/postprocess_pass.py +95 -0
  27. mindstudio_probe-1.2.1/msprobe/core/compare/merge_result/merge_result.py +380 -0
  28. mindstudio_probe-1.2.1/msprobe/core/compare/merge_result/merge_result_cli.py +31 -0
  29. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/compare/multiprocessing_compute.py +2 -2
  30. mindstudio_probe-1.2.1/msprobe/core/compare/npy_compare.py +301 -0
  31. mindstudio_probe-1.2.1/msprobe/core/compare/utils.py +609 -0
  32. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_collector.py +60 -28
  33. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_processor/base.py +84 -36
  34. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_processor/factory.py +5 -3
  35. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_processor/mindspore_processor.py +152 -18
  36. mindstudio_probe-1.2.1/msprobe/core/data_dump/data_processor/pytorch_processor.py +540 -0
  37. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/json_writer.py +29 -1
  38. mindstudio_probe-1.2.1/msprobe/core/data_dump/scope.py +274 -0
  39. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/constant.py +27 -13
  40. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/grad_compare.py +18 -1
  41. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/utils.py +30 -2
  42. mindstudio_probe-1.2.1/msprobe/core/overflow_check/abnormal_scene.py +189 -0
  43. mindstudio_probe-1.2.1/msprobe/core/overflow_check/api_info.py +55 -0
  44. mindstudio_probe-1.2.1/msprobe/core/overflow_check/checker.py +138 -0
  45. mindstudio_probe-1.2.1/msprobe/core/overflow_check/filter.py +157 -0
  46. mindstudio_probe-1.2.1/msprobe/core/overflow_check/ignore_rules.yaml +55 -0
  47. mindstudio_probe-1.2.1/msprobe/core/overflow_check/level.py +22 -0
  48. mindstudio_probe-1.2.1/msprobe/core/overflow_check/utils.py +28 -0
  49. mindstudio_probe-1.2.1/msprobe/docs/01.installation.md +188 -0
  50. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/02.config_introduction.md +50 -23
  51. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/03.config_examples.md +2 -9
  52. mindstudio_probe-1.2.1/msprobe/docs/04.kernel_dump_PyTorch.md +73 -0
  53. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/05.data_dump_PyTorch.md +93 -61
  54. mindstudio_probe-1.2.1/msprobe/docs/06.data_dump_MindSpore.md +390 -0
  55. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/07.accuracy_checker_PyTorch.md +28 -28
  56. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/08.accuracy_checker_online_PyTorch.md +1 -6
  57. mindstudio_probe-1.2.1/msprobe/docs/09.accuracy_checker_MindSpore.md +104 -0
  58. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/10.accuracy_compare_PyTorch.md +114 -50
  59. mindstudio_probe-1.2.1/msprobe/docs/11.accuracy_compare_MindSpore.md +625 -0
  60. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/12.overflow_check_PyTorch.md +2 -2
  61. mindstudio_probe-1.2.1/msprobe/docs/13.overflow_check_MindSpore.md +31 -0
  62. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/15.free_benchmarking_PyTorch.md +4 -5
  63. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/16.free_benchmarking_MindSpore.md +56 -37
  64. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/17.grad_probe.md +5 -6
  65. mindstudio_probe-1.2.1/msprobe/docs/19.monitor.md +561 -0
  66. mindstudio_probe-1.2.1/msprobe/docs/20.monitor_performance_baseline.md +52 -0
  67. mindstudio_probe-1.2.1/msprobe/docs/21.visualization_PyTorch.md +466 -0
  68. mindstudio_probe-1.2.1/msprobe/docs/22.visualization_MindSpore.md +481 -0
  69. mindstudio_probe-1.2.1/msprobe/docs/23.generate_operator_PyTorch.md +107 -0
  70. mindstudio_probe-1.2.1/msprobe/docs/24.code_mapping_Mindspore.md +28 -0
  71. mindstudio_probe-1.2.1/msprobe/docs/25.tool_function_introduction.md +29 -0
  72. mindstudio_probe-1.2.1/msprobe/docs/26.data_dump_PyTorch_baseline.md +37 -0
  73. mindstudio_probe-1.2.1/msprobe/docs/27.dump_json_instruction.md +521 -0
  74. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/FAQ.md +29 -2
  75. mindstudio_probe-1.2.1/msprobe/docs/accuracy_checker_MindSpore/accuracy_checker_MindSpore_baseline.md +14 -0
  76. mindstudio_probe-1.2.1/msprobe/docs/data_dump_MindSpore/data_dump_MindSpore_baseline.md +22 -0
  77. mindstudio_probe-1.2.1/msprobe/docs/data_dump_MindSpore/dynamic_graph_quick_start_example.md +211 -0
  78. mindstudio_probe-1.2.1/msprobe/docs/img/compare_result.png +0 -0
  79. mindstudio_probe-1.2.1/msprobe/docs/img/merge_result.png +0 -0
  80. mindstudio_probe-1.2.1/msprobe/docs/img/monitor/cpu_info.png +0 -0
  81. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/fuzzy_match_ms.png +0 -0
  82. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/fuzzy_match_pt.png +0 -0
  83. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/tensorboard_1.png +0 -0
  84. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/tensorboard_2.png +0 -0
  85. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_browser_1.png +0 -0
  86. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_browser_2.png +0 -0
  87. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_precision_info.png +0 -0
  88. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_search_info.png +0 -0
  89. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_show_info.png +0 -0
  90. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_showcase.png +0 -0
  91. mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_unmatch_info.png +0 -0
  92. mindstudio_probe-1.2.1/msprobe/docs/visualization/GPTModel.png +0 -0
  93. mindstudio_probe-1.2.1/msprobe/docs/visualization/ParallelMLP.png +0 -0
  94. mindstudio_probe-1.2.1/msprobe/docs/visualization/layer_mapping_example.md +132 -0
  95. mindstudio_probe-1.2.1/msprobe/docs/visualization/mapping.png +0 -0
  96. mindstudio_probe-1.2.1/msprobe/docs/visualization/mapping1.png +0 -0
  97. mindstudio_probe-1.2.1/msprobe/docs/visualization/module_name.png +0 -0
  98. mindstudio_probe-1.2.1/msprobe/docs/visualization/module_name1.png +0 -0
  99. mindstudio_probe-1.2.1/msprobe/docs/visualization/no_mapping.png +0 -0
  100. mindstudio_probe-1.2.1/msprobe/docs/visualization/no_mapping1.png +0 -0
  101. mindstudio_probe-1.2.1/msprobe/docs/visualization/no_mapping_analyze.png +0 -0
  102. mindstudio_probe-1.2.1/msprobe/docs/visualization/top_layer.png +0 -0
  103. mindstudio_probe-1.2.1/msprobe/mindspore/__init__.py +27 -0
  104. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +279 -0
  105. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/api_info.py +21 -6
  106. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/api_runner.py +43 -18
  107. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +21 -7
  108. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/checker_support_api.yaml +77 -0
  109. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/cmd_parser.py +69 -0
  110. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/compute_element.py +64 -31
  111. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/data_manager.py +301 -0
  112. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/main.py +34 -0
  113. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/multi_api_accuracy_checker.py +212 -0
  114. mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/multi_data_manager.py +60 -0
  115. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/type_mapping.py +22 -5
  116. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/utils.py +34 -17
  117. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/cell_processor.py +33 -12
  118. mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/bind.py +264 -0
  119. mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/cmd_parser.py +40 -0
  120. mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/graph.py +49 -0
  121. mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/graph_parser.py +226 -0
  122. mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/main.py +24 -0
  123. mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/processor.py +34 -0
  124. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/common/const.py +35 -13
  125. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/common/log.py +5 -9
  126. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/common/utils.py +88 -4
  127. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/compare/distributed_compare.py +22 -24
  128. mindstudio_probe-1.2.1/msprobe/mindspore/compare/ms_compare.py +422 -0
  129. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/compare/ms_graph_compare.py +95 -52
  130. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/debugger/debugger_config.py +7 -1
  131. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/debugger/precision_debugger.py +87 -12
  132. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/dump_tool_factory.py +3 -1
  133. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/api_registry.py +95 -18
  134. mindstudio_probe-1.2.1/msprobe/mindspore/dump/hook_cell/hook_cell.py +77 -0
  135. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/primitive_hooks.py +45 -30
  136. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +36 -1
  137. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/wrap_api.py +92 -1
  138. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/jit_dump.py +17 -5
  139. mindstudio_probe-1.2.1/msprobe/mindspore/dump/kernel_dump/kernel_config.py +33 -0
  140. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/kernel_graph_dump.py +9 -4
  141. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/kernel_kbyk_dump.py +2 -4
  142. mindstudio_probe-1.2.1/msprobe/mindspore/dym_loader/hook_dynamic_loader.cc +140 -0
  143. mindstudio_probe-1.2.1/msprobe/mindspore/dym_loader/hook_dynamic_loader.h +53 -0
  144. mindstudio_probe-1.2.1/msprobe/mindspore/free_benchmark/api_pynative_self_check.py +246 -0
  145. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/common/handler_params.py +1 -2
  146. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/common/utils.py +19 -4
  147. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +0 -204
  148. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/base_handler.py +3 -3
  149. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/check_handler.py +4 -5
  150. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/fix_handler.py +4 -4
  151. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/handler_factory.py +4 -4
  152. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/add_noise.py +2 -2
  153. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +15 -6
  154. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +2 -2
  155. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +2 -2
  156. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +13 -6
  157. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +2 -2
  158. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/self_check_tool_factory.py +2 -2
  159. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/global_context.py +28 -8
  160. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/grad_analyzer.py +50 -24
  161. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/grad_monitor.py +16 -1
  162. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/grad_stat_csv.py +33 -5
  163. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/hook.py +35 -12
  164. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/utils.py +18 -5
  165. mindstudio_probe-1.2.1/msprobe/mindspore/mindtorch/__init__.py +18 -0
  166. mindstudio_probe-1.2.1/msprobe/mindspore/mindtorch/mindtorch_adaptor.py +255 -0
  167. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/ms_config.py +27 -16
  168. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +9 -4
  169. mindstudio_probe-1.2.1/msprobe/mindspore/runtime.py +19 -0
  170. mindstudio_probe-1.2.1/msprobe/mindspore/service.py +469 -0
  171. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/task_handler_factory.py +15 -0
  172. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/msprobe.py +48 -10
  173. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/__init__.py +8 -6
  174. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/common/config.py +132 -0
  175. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/common/utils.py +31 -16
  176. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +41 -8
  177. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +103 -271
  178. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +4 -1
  179. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/compare.py +69 -68
  180. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +54 -0
  181. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/compare/compare_input.py +51 -0
  182. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +2 -4
  183. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/generate_op_script/config_op.json +9 -0
  184. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/generate_op_script/op_generator.py +478 -0
  185. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/generate_op_script/operator_replication.template +365 -0
  186. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/absolute_threshold.py +106 -0
  187. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/accumulative_error_compare.py +107 -0
  188. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/base_standard.py +151 -0
  189. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/benchmark_compare.py +226 -0
  190. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/binary_consistency.py +68 -0
  191. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/standard_config.py +218 -0
  192. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/standard_register.py +104 -0
  193. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/thousandth_standard.py +63 -0
  194. mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/ulp_compare.py +200 -0
  195. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +63 -2
  196. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +21 -15
  197. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +54 -22
  198. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +140 -71
  199. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +49 -8
  200. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +9 -24
  201. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +4 -12
  202. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +5 -3
  203. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py +9 -4
  204. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +3 -11
  205. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py +2 -2
  206. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/confusion_transpose.py +5 -1
  207. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/matmul_backward.py +12 -0
  208. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/npu_fusion_attention.py +142 -16
  209. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/rotary_mul.py +4 -0
  210. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/swiglu.py +10 -2
  211. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/parse_json.py +7 -6
  212. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/utils.py +101 -7
  213. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/distributed_compare.py +17 -30
  214. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/pt_compare.py +44 -22
  215. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/debugger/debugger_config.py +46 -27
  216. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/debugger/precision_debugger.py +42 -12
  217. mindstudio_probe-1.2.1/msprobe/pytorch/dump/kernel_dump/kernel_config.py +33 -0
  218. mindstudio_probe-1.2.1/msprobe/pytorch/dump/module_dump/module_dump.py +86 -0
  219. {mindstudio-probe-1.1.0/msprobe/pytorch → mindstudio_probe-1.2.1/msprobe/pytorch/dump/module_dump}/module_processer.py +81 -10
  220. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/constant.py +15 -0
  221. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/counter.py +15 -0
  222. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/enums.py +15 -0
  223. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/params.py +10 -2
  224. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/utils.py +29 -4
  225. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/compare/grad_saver.py +20 -5
  226. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/compare/single_benchmark.py +2 -0
  227. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +3 -1
  228. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +6 -4
  229. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +2 -0
  230. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +4 -0
  231. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +41 -47
  232. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +6 -5
  233. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +0 -4
  234. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/grad_probe/grad_monitor.py +23 -6
  235. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/grad_probe/grad_stat_csv.py +40 -10
  236. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/__init__.py +1 -1
  237. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/hook_module.py +14 -11
  238. mindstudio_probe-1.2.1/msprobe/pytorch/hook_module/register_optimizer_hook.py +59 -0
  239. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/support_wrap_ops.yaml +35 -0
  240. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_distributed.py +6 -8
  241. mindstudio_probe-1.2.1/msprobe/pytorch/hook_module/wrap_functional.py +66 -0
  242. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/anomaly_analyse.py +201 -0
  243. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/anomaly_detect.py +425 -0
  244. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/csv2tb.py +166 -0
  245. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed/distributed_ops.yaml +19 -0
  246. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed/stack_blacklist.yaml +5 -0
  247. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed/wrap_distributed.py +283 -0
  248. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/features.py +108 -0
  249. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/module_hook.py +1076 -0
  250. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/module_metric.py +172 -0
  251. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/module_spec_verifier.py +95 -0
  252. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/optimizer_collect.py +333 -0
  253. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/unittest/__init__.py +0 -0
  254. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/unittest/test_monitor.py +160 -0
  255. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/utils.py +321 -0
  256. mindstudio_probe-1.2.1/msprobe/pytorch/monitor/visualizer.py +59 -0
  257. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/__init__.py +2 -3
  258. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/compare.py +29 -38
  259. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/dispatch.py +58 -27
  260. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/dump_compare.py +21 -9
  261. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/single_compare.py +53 -32
  262. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/torch_ops_config.yaml +1 -1
  263. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/utils.py +49 -21
  264. mindstudio_probe-1.2.1/msprobe/pytorch/parse_tool/__init__.py +0 -0
  265. mindstudio_probe-1.2.1/msprobe/pytorch/parse_tool/lib/__init__.py +0 -0
  266. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/compare.py +21 -27
  267. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/config.py +6 -8
  268. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/file_desc.py +15 -1
  269. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/interactive_cli.py +10 -10
  270. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/parse_exception.py +7 -7
  271. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/parse_tool.py +12 -12
  272. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/utils.py +33 -53
  273. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/visualization.py +11 -10
  274. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/pt_config.py +31 -8
  275. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/service.py +188 -108
  276. mindstudio_probe-1.2.1/msprobe/visualization/__init__.py +14 -0
  277. mindstudio_probe-1.2.1/msprobe/visualization/builder/__init__.py +14 -0
  278. mindstudio_probe-1.2.1/msprobe/visualization/builder/graph_builder.py +222 -0
  279. mindstudio_probe-1.2.1/msprobe/visualization/builder/msprobe_adapter.py +227 -0
  280. mindstudio_probe-1.2.1/msprobe/visualization/compare/__init__.py +14 -0
  281. mindstudio_probe-1.2.1/msprobe/visualization/compare/graph_comparator.py +180 -0
  282. mindstudio_probe-1.2.1/msprobe/visualization/compare/mode_adapter.py +197 -0
  283. mindstudio_probe-1.2.1/msprobe/visualization/graph/__init__.py +14 -0
  284. mindstudio_probe-1.2.1/msprobe/visualization/graph/base_node.py +119 -0
  285. mindstudio_probe-1.2.1/msprobe/visualization/graph/distributed_analyzer.py +318 -0
  286. mindstudio_probe-1.2.1/msprobe/visualization/graph/graph.py +209 -0
  287. mindstudio_probe-1.2.1/msprobe/visualization/graph/node_colors.py +95 -0
  288. mindstudio_probe-1.2.1/msprobe/visualization/graph/node_op.py +39 -0
  289. mindstudio_probe-1.2.1/msprobe/visualization/graph_service.py +288 -0
  290. mindstudio_probe-1.2.1/msprobe/visualization/utils.py +217 -0
  291. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/setup.py +48 -7
  292. mindstudio-probe-1.1.0/msprobe/core/compare/acc_compare.py +0 -473
  293. mindstudio-probe-1.1.0/msprobe/core/compare/highlight.py +0 -259
  294. mindstudio-probe-1.1.0/msprobe/core/compare/npy_compare.py +0 -310
  295. mindstudio-probe-1.1.0/msprobe/core/compare/utils.py +0 -464
  296. mindstudio-probe-1.1.0/msprobe/core/data_dump/data_processor/pytorch_processor.py +0 -383
  297. mindstudio-probe-1.1.0/msprobe/core/data_dump/scope.py +0 -194
  298. mindstudio-probe-1.1.0/msprobe/docs/01.installation.md +0 -99
  299. mindstudio-probe-1.1.0/msprobe/docs/04.acl_config_examples.md +0 -78
  300. mindstudio-probe-1.1.0/msprobe/docs/06.data_dump_MindSpore.md +0 -285
  301. mindstudio-probe-1.1.0/msprobe/docs/09.accuracy_checker_MindSpore.md +0 -68
  302. mindstudio-probe-1.1.0/msprobe/docs/11.accuracy_compare_MindSpore.md +0 -333
  303. mindstudio-probe-1.1.0/msprobe/docs/13.overflow_check_MindSpore.md +0 -31
  304. mindstudio-probe-1.1.0/msprobe/mindspore/__init__.py +0 -2
  305. mindstudio-probe-1.1.0/msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +0 -279
  306. mindstudio-probe-1.1.0/msprobe/mindspore/api_accuracy_checker/cmd_parser.py +0 -6
  307. mindstudio-probe-1.1.0/msprobe/mindspore/api_accuracy_checker/main.py +0 -9
  308. mindstudio-probe-1.1.0/msprobe/mindspore/compare/layer_mapping.py +0 -146
  309. mindstudio-probe-1.1.0/msprobe/mindspore/compare/modify_mapping.py +0 -107
  310. mindstudio-probe-1.1.0/msprobe/mindspore/compare/ms_compare.py +0 -357
  311. mindstudio-probe-1.1.0/msprobe/mindspore/dump/hook_cell/hook_cell.py +0 -55
  312. mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/api_pynative_self_check.py +0 -131
  313. mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/decorator/dec_forward.py +0 -57
  314. mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +0 -122
  315. mindstudio-probe-1.1.0/msprobe/mindspore/runtime.py +0 -4
  316. mindstudio-probe-1.1.0/msprobe/mindspore/service.py +0 -297
  317. mindstudio-probe-1.1.0/msprobe/pytorch/api_accuracy_checker/common/config.py +0 -70
  318. mindstudio-probe-1.1.0/msprobe/pytorch/functional/module_dump.py +0 -84
  319. mindstudio-probe-1.1.0/msprobe/pytorch/hook_module/wrap_functional.py +0 -104
  320. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/LICENSE +0 -0
  321. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/dependency_links.txt +0 -0
  322. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/not-zip-safe +0 -0
  323. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/top_level.txt +0 -0
  324. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/__init__.py +0 -0
  325. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/__init__.py +0 -0
  326. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/advisor/advisor_const.py +0 -0
  327. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/advisor/advisor_result.py +0 -0
  328. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/__init__.py +0 -0
  329. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/14.data_parse_PyTorch.md +0 -0
  330. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/18.online_dispatch.md +0 -0
  331. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +0 -0
  332. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_1.png +0 -0
  333. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_2.png +0 -0
  334. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_3.png +0 -0
  335. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_4.png +0 -0
  336. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_1.png +0 -0
  337. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_2.png +0 -0
  338. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_3.png +0 -0
  339. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_4.png +0 -0
  340. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_5.png +0 -0
  341. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_6.png +0 -0
  342. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_7.png +0 -0
  343. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_8.png +0 -0
  344. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/YOLOV5S_1.png +0 -0
  345. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/YOLOV5S_2.png +0 -0
  346. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/accuracy_checking_details.png +0 -0
  347. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/accuracy_checking_result.png +0 -0
  348. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/api_precision_compare_details.png +0 -0
  349. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/api_precision_compare_result.png +0 -0
  350. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/auto_analyze_log.png +0 -0
  351. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/compare_result_pkl.png +0 -0
  352. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/compare_result_pkl_md5.png.png +0 -0
  353. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/cpu_info.png +0 -0
  354. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/free_benchmark.png +0 -0
  355. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/free_benchmark_framework.png +0 -0
  356. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-1.png +0 -0
  357. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-2.png +0 -0
  358. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-3.png +0 -0
  359. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-4.png +0 -0
  360. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image.png +0 -0
  361. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/module_compare.png +0 -0
  362. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/ms_dump.png +0 -0
  363. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/ms_layer.png +0 -0
  364. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/pt_dump.png +0 -0
  365. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/__init__.py +0 -0
  366. {mindstudio-probe-1.1.0/msprobe/mindspore/debugger → mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping}/__init__.py +0 -0
  367. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/compare/ms_to_pt_api.yaml +0 -0
  368. {mindstudio-probe-1.1.0/msprobe/mindspore/dump → mindstudio_probe-1.2.1/msprobe/mindspore/debugger}/__init__.py +0 -0
  369. {mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark → mindstudio_probe-1.2.1/msprobe/mindspore/dump}/__init__.py +0 -0
  370. {mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/common → mindstudio_probe-1.2.1/msprobe/mindspore/free_benchmark}/__init__.py +0 -0
  371. {mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/decorator → mindstudio_probe-1.2.1/msprobe/mindspore/free_benchmark/common}/__init__.py +0 -0
  372. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/common/config.py +0 -0
  373. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/__init__.py +0 -0
  374. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/no_change.py +0 -0
  375. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/__init__.py +0 -0
  376. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/overflow_check/__init__.py +0 -0
  377. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +0 -0
  378. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/.keep +0 -0
  379. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/__init__.py +0 -0
  380. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/common/.keep +0 -0
  381. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/common/__init__.py +0 -0
  382. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/__init__.py +0 -0
  383. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +0 -0
  384. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/config.yaml +0 -0
  385. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/.keep +0 -0
  386. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/__init__.py +0 -0
  387. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +0 -0
  388. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/__init__.py +0 -0
  389. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml +0 -0
  390. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/__init__.py +0 -0
  391. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/apply_adam_w.py +0 -0
  392. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/fast_gelu.py +0 -0
  393. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/layer_norm_eval.py +0 -0
  394. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/linear.py +0 -0
  395. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/rms_norm.py +0 -0
  396. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/scaled_mask_softmax.py +0 -0
  397. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/__init__.py +0 -0
  398. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/compare_script.template +0 -0
  399. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/log.py +0 -0
  400. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/mapping.yaml +0 -0
  401. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/match.py +0 -0
  402. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/debugger/__init__.py +0 -0
  403. {mindstudio-probe-1.1.0/msprobe/pytorch/free_benchmark/common → mindstudio_probe-1.2.1/msprobe/pytorch/dump/module_dump}/__init__.py +0 -0
  404. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/__init__.py +0 -0
  405. {mindstudio-probe-1.1.0/msprobe/pytorch/free_benchmark/perturbed_layers → mindstudio_probe-1.2.1/msprobe/pytorch/free_benchmark/common}/__init__.py +0 -0
  406. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/main.py +0 -0
  407. {mindstudio-probe-1.1.0/msprobe/pytorch/free_benchmark/perturbed_layers/npu → mindstudio_probe-1.2.1/msprobe/pytorch/free_benchmark/perturbed_layers}/__init__.py +0 -0
  408. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +0 -0
  409. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +0 -0
  410. {mindstudio-probe-1.1.0/msprobe/pytorch/free_benchmark/result_handlers → mindstudio_probe-1.2.1/msprobe/pytorch/free_benchmark/perturbed_layers/npu}/__init__.py +0 -0
  411. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +0 -0
  412. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +0 -0
  413. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +0 -0
  414. {mindstudio-probe-1.1.0/msprobe/pytorch/functional → mindstudio_probe-1.2.1/msprobe/pytorch/free_benchmark/result_handlers}/__init__.py +0 -0
  415. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +0 -0
  416. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +0 -0
  417. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/function_factory.py +0 -0
  418. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/grad_probe/__init__.py +0 -0
  419. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/api_registry.py +0 -0
  420. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/utils.py +0 -0
  421. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_aten.py +0 -0
  422. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_npu_custom.py +0 -0
  423. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_tensor.py +0 -0
  424. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_torch.py +0 -0
  425. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_vf.py +0 -0
  426. {mindstudio-probe-1.1.0/msprobe/pytorch/parse_tool → mindstudio_probe-1.2.1/msprobe/pytorch/monitor}/__init__.py +0 -0
  427. {mindstudio-probe-1.1.0/msprobe/pytorch/parse_tool/lib → mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed}/__init__.py +0 -0
  428. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse.py +0 -0
  429. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/cli.py +0 -0
  430. {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/setup.cfg +0 -0
@@ -2,4 +2,5 @@ include README.md
2
2
  include LICENSE
3
3
  recursive-include msprobe *
4
4
  recursive-exclude msprobe/test *
5
+ recursive-exclude msprobe/ccsrc *
5
6
 
@@ -1,17 +1,17 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mindstudio-probe
3
- Version: 1.1.0
4
- Summary: Pytorch Ascend Probe Utils
3
+ Version: 1.2.1
4
+ Summary: Ascend Probe Utils
5
5
  Home-page: https://gitee.com/ascend/mstt/tree/master/debug/accuracy_tools/msprobe
6
6
  Author: Ascend Team
7
7
  Author-email: pmail_mindstudio@huawei.com
8
8
  License: Apache License 2.0
9
9
  Keywords: pytorch msprobe ascend
10
- Platform: UNKNOWN
11
10
  Classifier: Intended Audience :: Developers
12
11
  Classifier: Intended Audience :: Education
13
12
  Classifier: Intended Audience :: Science/Research
14
13
  Classifier: Programming Language :: Python :: 3
14
+ Classifier: Programming Language :: C++
15
15
  Classifier: Topic :: Scientific/Engineering
16
16
  Classifier: Topic :: Scientific/Engineering :: Mathematics
17
17
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -20,6 +20,18 @@ Classifier: Topic :: Software Development :: Libraries
20
20
  Classifier: Topic :: Software Development :: Libraries :: Python Modules
21
21
  Requires-Python: >=3.6.2
22
22
  License-File: LICENSE
23
+ Requires-Dist: wheel
24
+ Requires-Dist: einops
25
+ Requires-Dist: numpy<2.0
26
+ Requires-Dist: pandas<2.1,>=1.3.5
27
+ Requires-Dist: pyyaml
28
+ Requires-Dist: rich
29
+ Requires-Dist: tqdm
30
+ Requires-Dist: openpyxl
31
+ Requires-Dist: pyopenssl
32
+ Requires-Dist: twisted
33
+ Requires-Dist: matplotlib
34
+ Requires-Dist: tensorboard
35
+ Requires-Dist: tabulate
23
36
 
24
37
  MindStudio-Probe is a set of tools for diagnosing and improving model accuracy on Ascend NPU, including API acc checker, ptdbg, grad tool etc.
25
-
@@ -1,17 +1,17 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mindstudio-probe
3
- Version: 1.1.0
4
- Summary: Pytorch Ascend Probe Utils
3
+ Version: 1.2.1
4
+ Summary: Ascend Probe Utils
5
5
  Home-page: https://gitee.com/ascend/mstt/tree/master/debug/accuracy_tools/msprobe
6
6
  Author: Ascend Team
7
7
  Author-email: pmail_mindstudio@huawei.com
8
8
  License: Apache License 2.0
9
9
  Keywords: pytorch msprobe ascend
10
- Platform: UNKNOWN
11
10
  Classifier: Intended Audience :: Developers
12
11
  Classifier: Intended Audience :: Education
13
12
  Classifier: Intended Audience :: Science/Research
14
13
  Classifier: Programming Language :: Python :: 3
14
+ Classifier: Programming Language :: C++
15
15
  Classifier: Topic :: Scientific/Engineering
16
16
  Classifier: Topic :: Scientific/Engineering :: Mathematics
17
17
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -20,6 +20,18 @@ Classifier: Topic :: Software Development :: Libraries
20
20
  Classifier: Topic :: Software Development :: Libraries :: Python Modules
21
21
  Requires-Python: >=3.6.2
22
22
  License-File: LICENSE
23
+ Requires-Dist: wheel
24
+ Requires-Dist: einops
25
+ Requires-Dist: numpy<2.0
26
+ Requires-Dist: pandas<2.1,>=1.3.5
27
+ Requires-Dist: pyyaml
28
+ Requires-Dist: rich
29
+ Requires-Dist: tqdm
30
+ Requires-Dist: openpyxl
31
+ Requires-Dist: pyopenssl
32
+ Requires-Dist: twisted
33
+ Requires-Dist: matplotlib
34
+ Requires-Dist: tensorboard
35
+ Requires-Dist: tabulate
23
36
 
24
37
  MindStudio-Probe is a set of tools for diagnosing and improving model accuracy on Ascend NPU, including API acc checker, ptdbg, grad tool etc.
25
-
@@ -8,6 +8,7 @@ mindstudio_probe.egg-info/entry_points.txt
8
8
  mindstudio_probe.egg-info/not-zip-safe
9
9
  mindstudio_probe.egg-info/requires.txt
10
10
  mindstudio_probe.egg-info/top_level.txt
11
+ msprobe/CMakeLists.txt
11
12
  msprobe/README.md
12
13
  msprobe/__init__.py
13
14
  msprobe/config.json
@@ -31,6 +32,12 @@ msprobe/core/compare/highlight.py
31
32
  msprobe/core/compare/multiprocessing_compute.py
32
33
  msprobe/core/compare/npy_compare.py
33
34
  msprobe/core/compare/utils.py
35
+ msprobe/core/compare/layer_mapping/__init__.py
36
+ msprobe/core/compare/layer_mapping/data_scope_parser.py
37
+ msprobe/core/compare/layer_mapping/layer_mapping.py
38
+ msprobe/core/compare/layer_mapping/postprocess_pass.py
39
+ msprobe/core/compare/merge_result/merge_result.py
40
+ msprobe/core/compare/merge_result/merge_result_cli.py
34
41
  msprobe/core/data_dump/data_collector.py
35
42
  msprobe/core/data_dump/json_writer.py
36
43
  msprobe/core/data_dump/scope.py
@@ -42,10 +49,17 @@ msprobe/core/grad_probe/__init__.py
42
49
  msprobe/core/grad_probe/constant.py
43
50
  msprobe/core/grad_probe/grad_compare.py
44
51
  msprobe/core/grad_probe/utils.py
52
+ msprobe/core/overflow_check/abnormal_scene.py
53
+ msprobe/core/overflow_check/api_info.py
54
+ msprobe/core/overflow_check/checker.py
55
+ msprobe/core/overflow_check/filter.py
56
+ msprobe/core/overflow_check/ignore_rules.yaml
57
+ msprobe/core/overflow_check/level.py
58
+ msprobe/core/overflow_check/utils.py
45
59
  msprobe/docs/01.installation.md
46
60
  msprobe/docs/02.config_introduction.md
47
61
  msprobe/docs/03.config_examples.md
48
- msprobe/docs/04.acl_config_examples.md
62
+ msprobe/docs/04.kernel_dump_PyTorch.md
49
63
  msprobe/docs/05.data_dump_PyTorch.md
50
64
  msprobe/docs/06.data_dump_MindSpore.md
51
65
  msprobe/docs/07.accuracy_checker_PyTorch.md
@@ -60,8 +74,20 @@ msprobe/docs/15.free_benchmarking_PyTorch.md
60
74
  msprobe/docs/16.free_benchmarking_MindSpore.md
61
75
  msprobe/docs/17.grad_probe.md
62
76
  msprobe/docs/18.online_dispatch.md
77
+ msprobe/docs/19.monitor.md
78
+ msprobe/docs/20.monitor_performance_baseline.md
79
+ msprobe/docs/21.visualization_PyTorch.md
80
+ msprobe/docs/22.visualization_MindSpore.md
81
+ msprobe/docs/23.generate_operator_PyTorch.md
82
+ msprobe/docs/24.code_mapping_Mindspore.md
83
+ msprobe/docs/25.tool_function_introduction.md
84
+ msprobe/docs/26.data_dump_PyTorch_baseline.md
85
+ msprobe/docs/27.dump_json_instruction.md
63
86
  msprobe/docs/FAQ.md
64
87
  msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md
88
+ msprobe/docs/accuracy_checker_MindSpore/accuracy_checker_MindSpore_baseline.md
89
+ msprobe/docs/data_dump_MindSpore/data_dump_MindSpore_baseline.md
90
+ msprobe/docs/data_dump_MindSpore/dynamic_graph_quick_start_example.md
65
91
  msprobe/docs/img/BLOOM-7B_1.png
66
92
  msprobe/docs/img/BLOOM-7B_2.png
67
93
  msprobe/docs/img/BLOOM-7B_3.png
@@ -81,6 +107,7 @@ msprobe/docs/img/accuracy_checking_result.png
81
107
  msprobe/docs/img/api_precision_compare_details.png
82
108
  msprobe/docs/img/api_precision_compare_result.png
83
109
  msprobe/docs/img/auto_analyze_log.png
110
+ msprobe/docs/img/compare_result.png
84
111
  msprobe/docs/img/compare_result_pkl.png
85
112
  msprobe/docs/img/compare_result_pkl_md5.png.png
86
113
  msprobe/docs/img/cpu_info.png
@@ -91,10 +118,34 @@ msprobe/docs/img/grad_probe_image-2.png
91
118
  msprobe/docs/img/grad_probe_image-3.png
92
119
  msprobe/docs/img/grad_probe_image-4.png
93
120
  msprobe/docs/img/grad_probe_image.png
121
+ msprobe/docs/img/merge_result.png
94
122
  msprobe/docs/img/module_compare.png
95
123
  msprobe/docs/img/ms_dump.png
96
124
  msprobe/docs/img/ms_layer.png
97
125
  msprobe/docs/img/pt_dump.png
126
+ msprobe/docs/img/monitor/cpu_info.png
127
+ msprobe/docs/img/visualization/fuzzy_match_ms.png
128
+ msprobe/docs/img/visualization/fuzzy_match_pt.png
129
+ msprobe/docs/img/visualization/tensorboard_1.png
130
+ msprobe/docs/img/visualization/tensorboard_2.png
131
+ msprobe/docs/img/visualization/vis_browser_1.png
132
+ msprobe/docs/img/visualization/vis_browser_2.png
133
+ msprobe/docs/img/visualization/vis_precision_info.png
134
+ msprobe/docs/img/visualization/vis_search_info.png
135
+ msprobe/docs/img/visualization/vis_show_info.png
136
+ msprobe/docs/img/visualization/vis_showcase.png
137
+ msprobe/docs/img/visualization/vis_unmatch_info.png
138
+ msprobe/docs/visualization/GPTModel.png
139
+ msprobe/docs/visualization/ParallelMLP.png
140
+ msprobe/docs/visualization/layer_mapping_example.md
141
+ msprobe/docs/visualization/mapping.png
142
+ msprobe/docs/visualization/mapping1.png
143
+ msprobe/docs/visualization/module_name.png
144
+ msprobe/docs/visualization/module_name1.png
145
+ msprobe/docs/visualization/no_mapping.png
146
+ msprobe/docs/visualization/no_mapping1.png
147
+ msprobe/docs/visualization/no_mapping_analyze.png
148
+ msprobe/docs/visualization/top_layer.png
98
149
  msprobe/mindspore/__init__.py
99
150
  msprobe/mindspore/cell_processor.py
100
151
  msprobe/mindspore/ms_config.py
@@ -106,17 +157,26 @@ msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py
106
157
  msprobe/mindspore/api_accuracy_checker/api_info.py
107
158
  msprobe/mindspore/api_accuracy_checker/api_runner.py
108
159
  msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py
160
+ msprobe/mindspore/api_accuracy_checker/checker_support_api.yaml
109
161
  msprobe/mindspore/api_accuracy_checker/cmd_parser.py
110
162
  msprobe/mindspore/api_accuracy_checker/compute_element.py
163
+ msprobe/mindspore/api_accuracy_checker/data_manager.py
111
164
  msprobe/mindspore/api_accuracy_checker/main.py
165
+ msprobe/mindspore/api_accuracy_checker/multi_api_accuracy_checker.py
166
+ msprobe/mindspore/api_accuracy_checker/multi_data_manager.py
112
167
  msprobe/mindspore/api_accuracy_checker/type_mapping.py
113
168
  msprobe/mindspore/api_accuracy_checker/utils.py
169
+ msprobe/mindspore/code_mapping/__init__.py
170
+ msprobe/mindspore/code_mapping/bind.py
171
+ msprobe/mindspore/code_mapping/cmd_parser.py
172
+ msprobe/mindspore/code_mapping/graph.py
173
+ msprobe/mindspore/code_mapping/graph_parser.py
174
+ msprobe/mindspore/code_mapping/main.py
175
+ msprobe/mindspore/code_mapping/processor.py
114
176
  msprobe/mindspore/common/const.py
115
177
  msprobe/mindspore/common/log.py
116
178
  msprobe/mindspore/common/utils.py
117
179
  msprobe/mindspore/compare/distributed_compare.py
118
- msprobe/mindspore/compare/layer_mapping.py
119
- msprobe/mindspore/compare/modify_mapping.py
120
180
  msprobe/mindspore/compare/ms_compare.py
121
181
  msprobe/mindspore/compare/ms_graph_compare.py
122
182
  msprobe/mindspore/compare/ms_to_pt_api.yaml
@@ -133,6 +193,9 @@ msprobe/mindspore/dump/hook_cell/hook_cell.py
133
193
  msprobe/mindspore/dump/hook_cell/primitive_hooks.py
134
194
  msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml
135
195
  msprobe/mindspore/dump/hook_cell/wrap_api.py
196
+ msprobe/mindspore/dump/kernel_dump/kernel_config.py
197
+ msprobe/mindspore/dym_loader/hook_dynamic_loader.cc
198
+ msprobe/mindspore/dym_loader/hook_dynamic_loader.h
136
199
  msprobe/mindspore/free_benchmark/__init__.py
137
200
  msprobe/mindspore/free_benchmark/api_pynative_self_check.py
138
201
  msprobe/mindspore/free_benchmark/self_check_tool_factory.py
@@ -141,9 +204,6 @@ msprobe/mindspore/free_benchmark/common/config.py
141
204
  msprobe/mindspore/free_benchmark/common/handler_params.py
142
205
  msprobe/mindspore/free_benchmark/common/utils.py
143
206
  msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml
144
- msprobe/mindspore/free_benchmark/decorator/__init__.py
145
- msprobe/mindspore/free_benchmark/decorator/dec_forward.py
146
- msprobe/mindspore/free_benchmark/decorator/decorator_factory.py
147
207
  msprobe/mindspore/free_benchmark/handler/__init__.py
148
208
  msprobe/mindspore/free_benchmark/handler/base_handler.py
149
209
  msprobe/mindspore/free_benchmark/handler/check_handler.py
@@ -163,12 +223,13 @@ msprobe/mindspore/grad_probe/grad_monitor.py
163
223
  msprobe/mindspore/grad_probe/grad_stat_csv.py
164
224
  msprobe/mindspore/grad_probe/hook.py
165
225
  msprobe/mindspore/grad_probe/utils.py
226
+ msprobe/mindspore/mindtorch/__init__.py
227
+ msprobe/mindspore/mindtorch/mindtorch_adaptor.py
166
228
  msprobe/mindspore/overflow_check/__init__.py
167
229
  msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py
168
230
  msprobe/mindspore/overflow_check/overflow_check_tool_factory.py
169
231
  msprobe/pytorch/__init__.py
170
232
  msprobe/pytorch/function_factory.py
171
- msprobe/pytorch/module_processer.py
172
233
  msprobe/pytorch/parse.py
173
234
  msprobe/pytorch/pt_config.py
174
235
  msprobe/pytorch/service.py
@@ -186,7 +247,20 @@ msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml
186
247
  msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml
187
248
  msprobe/pytorch/api_accuracy_checker/compare/compare.py
188
249
  msprobe/pytorch/api_accuracy_checker/compare/compare_column.py
250
+ msprobe/pytorch/api_accuracy_checker/compare/compare_input.py
189
251
  msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py
252
+ msprobe/pytorch/api_accuracy_checker/generate_op_script/config_op.json
253
+ msprobe/pytorch/api_accuracy_checker/generate_op_script/op_generator.py
254
+ msprobe/pytorch/api_accuracy_checker/generate_op_script/operator_replication.template
255
+ msprobe/pytorch/api_accuracy_checker/precision_standard/absolute_threshold.py
256
+ msprobe/pytorch/api_accuracy_checker/precision_standard/accumulative_error_compare.py
257
+ msprobe/pytorch/api_accuracy_checker/precision_standard/base_standard.py
258
+ msprobe/pytorch/api_accuracy_checker/precision_standard/benchmark_compare.py
259
+ msprobe/pytorch/api_accuracy_checker/precision_standard/binary_consistency.py
260
+ msprobe/pytorch/api_accuracy_checker/precision_standard/standard_config.py
261
+ msprobe/pytorch/api_accuracy_checker/precision_standard/standard_register.py
262
+ msprobe/pytorch/api_accuracy_checker/precision_standard/thousandth_standard.py
263
+ msprobe/pytorch/api_accuracy_checker/precision_standard/ulp_compare.py
190
264
  msprobe/pytorch/api_accuracy_checker/run_ut/.keep
191
265
  msprobe/pytorch/api_accuracy_checker/run_ut/__init__.py
192
266
  msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py
@@ -227,6 +301,10 @@ msprobe/pytorch/compare/pt_compare.py
227
301
  msprobe/pytorch/debugger/__init__.py
228
302
  msprobe/pytorch/debugger/debugger_config.py
229
303
  msprobe/pytorch/debugger/precision_debugger.py
304
+ msprobe/pytorch/dump/kernel_dump/kernel_config.py
305
+ msprobe/pytorch/dump/module_dump/__init__.py
306
+ msprobe/pytorch/dump/module_dump/module_dump.py
307
+ msprobe/pytorch/dump/module_dump/module_processer.py
230
308
  msprobe/pytorch/free_benchmark/__init__.py
231
309
  msprobe/pytorch/free_benchmark/main.py
232
310
  msprobe/pytorch/free_benchmark/common/__init__.py
@@ -254,14 +332,13 @@ msprobe/pytorch/free_benchmark/result_handlers/check_handler.py
254
332
  msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py
255
333
  msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py
256
334
  msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py
257
- msprobe/pytorch/functional/__init__.py
258
- msprobe/pytorch/functional/module_dump.py
259
335
  msprobe/pytorch/grad_probe/__init__.py
260
336
  msprobe/pytorch/grad_probe/grad_monitor.py
261
337
  msprobe/pytorch/grad_probe/grad_stat_csv.py
262
338
  msprobe/pytorch/hook_module/__init__.py
263
339
  msprobe/pytorch/hook_module/api_registry.py
264
340
  msprobe/pytorch/hook_module/hook_module.py
341
+ msprobe/pytorch/hook_module/register_optimizer_hook.py
265
342
  msprobe/pytorch/hook_module/support_wrap_ops.yaml
266
343
  msprobe/pytorch/hook_module/utils.py
267
344
  msprobe/pytorch/hook_module/wrap_aten.py
@@ -271,6 +348,23 @@ msprobe/pytorch/hook_module/wrap_npu_custom.py
271
348
  msprobe/pytorch/hook_module/wrap_tensor.py
272
349
  msprobe/pytorch/hook_module/wrap_torch.py
273
350
  msprobe/pytorch/hook_module/wrap_vf.py
351
+ msprobe/pytorch/monitor/__init__.py
352
+ msprobe/pytorch/monitor/anomaly_analyse.py
353
+ msprobe/pytorch/monitor/anomaly_detect.py
354
+ msprobe/pytorch/monitor/csv2tb.py
355
+ msprobe/pytorch/monitor/features.py
356
+ msprobe/pytorch/monitor/module_hook.py
357
+ msprobe/pytorch/monitor/module_metric.py
358
+ msprobe/pytorch/monitor/module_spec_verifier.py
359
+ msprobe/pytorch/monitor/optimizer_collect.py
360
+ msprobe/pytorch/monitor/utils.py
361
+ msprobe/pytorch/monitor/visualizer.py
362
+ msprobe/pytorch/monitor/distributed/__init__.py
363
+ msprobe/pytorch/monitor/distributed/distributed_ops.yaml
364
+ msprobe/pytorch/monitor/distributed/stack_blacklist.yaml
365
+ msprobe/pytorch/monitor/distributed/wrap_distributed.py
366
+ msprobe/pytorch/monitor/unittest/__init__.py
367
+ msprobe/pytorch/monitor/unittest/test_monitor.py
274
368
  msprobe/pytorch/online_dispatch/__init__.py
275
369
  msprobe/pytorch/online_dispatch/compare.py
276
370
  msprobe/pytorch/online_dispatch/dispatch.py
@@ -288,4 +382,19 @@ msprobe/pytorch/parse_tool/lib/interactive_cli.py
288
382
  msprobe/pytorch/parse_tool/lib/parse_exception.py
289
383
  msprobe/pytorch/parse_tool/lib/parse_tool.py
290
384
  msprobe/pytorch/parse_tool/lib/utils.py
291
- msprobe/pytorch/parse_tool/lib/visualization.py
385
+ msprobe/pytorch/parse_tool/lib/visualization.py
386
+ msprobe/visualization/__init__.py
387
+ msprobe/visualization/graph_service.py
388
+ msprobe/visualization/utils.py
389
+ msprobe/visualization/builder/__init__.py
390
+ msprobe/visualization/builder/graph_builder.py
391
+ msprobe/visualization/builder/msprobe_adapter.py
392
+ msprobe/visualization/compare/__init__.py
393
+ msprobe/visualization/compare/graph_comparator.py
394
+ msprobe/visualization/compare/mode_adapter.py
395
+ msprobe/visualization/graph/__init__.py
396
+ msprobe/visualization/graph/base_node.py
397
+ msprobe/visualization/graph/distributed_analyzer.py
398
+ msprobe/visualization/graph/graph.py
399
+ msprobe/visualization/graph/node_colors.py
400
+ msprobe/visualization/graph/node_op.py
@@ -1,3 +1,2 @@
1
1
  [console_scripts]
2
2
  msprobe = msprobe.msprobe:main
3
-
@@ -9,3 +9,5 @@ openpyxl
9
9
  pyopenssl
10
10
  twisted
11
11
  matplotlib
12
+ tensorboard
13
+ tabulate
@@ -0,0 +1,5 @@
1
+ add_subdirectory(ccsrc)
2
+
3
+ if (DEFINED BUILD_TEST_CASE AND "${BUILD_TEST_CASE}" STREQUAL "True")
4
+ add_subdirectory(test)
5
+ endif()
@@ -9,7 +9,7 @@
9
9
  为方便使用,本工具提供了统一、简易的程序接口:**PrecisionDebugger**。以 PyTorch 框架为例,通过以下示例模板和 **config.json** 可以轻松使用各种功能。
10
10
 
11
11
  ```python
12
- from msprobe.pytorch import PrecisionDebugger
12
+ from msprobe.pytorch import PrecisionDebugger
13
13
 
14
14
  debugger = PrecisionDebugger(config_path='./config.json')
15
15
  ...
@@ -33,12 +33,40 @@ export MSPROBE_LOG_LEVEL={x}
33
33
  ```
34
34
  **config.json** 的配置要求和各功能具体的使用指导详见后续章节。
35
35
 
36
+ ## 环境和依赖
37
+
38
+ - 硬件环境请参见《[昇腾产品形态说明](https://gitee.com/link?target=https%3A%2F%2Fwww.hiascend.com%2Fdocument%2Fdetail%2Fzh%2Fcanncommercial%2F80RC22%2Fquickstart%2Fquickstart%2Fquickstart_18_0002.html)》。
39
+ - 软件环境请参见《[CANN 软件安装指南](https://gitee.com/link?target=https%3A%2F%2Fwww.hiascend.com%2Fdocument%2Fdetail%2Fzh%2Fcanncommercial%2F80RC22%2Fsoftwareinst%2Finstg%2Finstg_0000.html%3FMode%3DPmIns%26OS%3DUbuntu%26Software%3DcannToolKit)》安装昇腾设备开发或运行环境,即toolkit软件包。
40
+
41
+ 以上环境依赖请根据实际环境选择适配的版本。
42
+
43
+ ## 版本配套说明
44
+
45
+ - msprobe支持AscendPyTorch 1.11.0或更高版本,支持的PyTorch和CANN以及PyTorch和python软件版本配套关系请参见《[Ascend Extension for PyTorch插件](https://gitee.com/ascend/pytorch)》。
46
+ - msprobe支持MindSpore 2.4.0或更高版本,支持的MindSpore和CANN以及MindSpore和python软件版本配套关系请参见《[MindSpore版本发布列表](https://www.mindspore.cn/versions)》。
47
+ - msprobe支持的固件驱动版本与配套CANN软件支持的固件驱动版本相同,开发者可通过“[昇腾社区-固件与驱动](https://gitee.com/link?target=https%3A%2F%2Fwww.hiascend.com%2Fhardware%2Ffirmware-drivers%2Fcommunity%3Fproduct%3D2%26model%3D28%26cann%3D8.0.RC3.alpha003%26driver%3D1.0.25.alpha)”页面根据产品型号与CANN软件版本获取配套的固件与驱动。
48
+
49
+
50
+ ## 🚨 工具限制与注意事项
51
+
52
+ **1. Pytorch 框架下,工具暂不支持 Fully Sharded Data Parallel(FSDP)。**
53
+
54
+ **2. 工具读写的所有路径,如config_path、dump_path等,只允许包含大小写字母、数字、下划线、斜杠、点和短横线。**
55
+
36
56
  ## ⚙️ [安装](./docs/01.installation.md)
37
57
 
58
+ ## 🌟 新版本特性
59
+
60
+ 请参见[特性变更说明](./docs/01.installation.md#特性变更说明)。
61
+
38
62
  ## 🛠️ config.json [介绍](./docs/02.config_introduction.md) 和 [示例](./docs/03.config_examples.md)
39
63
 
40
64
  ## 🧰 主要功能
41
65
 
66
+ ### 0 用前必看
67
+
68
+ 使用工具前,建议先浏览[**工具功能模块简介、适用场景和当前版本局限性**](./docs/25.tool_function_introduction.md),了解功能特性。
69
+
42
70
  ### 1 数据采集
43
71
 
44
72
  msprobe 通过在训练脚本中添加 PrecisionDebugger 接口的方式对 API 执行精度数据 dump 操作,对应 config.json 中的 task 为 statistics 或 tensor。
@@ -95,29 +123,32 @@ MindSpore 动态图场景的[离线预检](./docs/09.accuracy_checker_MindSpore.
95
123
 
96
124
  [PyTorch 场景的在线精度比对](./docs/18.online_dispatch.md)
97
125
 
98
- ## 🌟 新版本特性
126
+ ### 9 训练状态监控
127
+
128
+ 该功能收集和聚合模型训练过程中的网络层,优化器, 通信算子的中间值,帮助诊断模型训练过程中计算, 通信,优化器各部分出现的异常情况。
129
+
130
+ [PyTorch 场景的训练状态监控](./docs/19.monitor.md)
131
+
132
+ ### 10 分级可视化构图比对
133
+
134
+ 该功能将msprobe工具dump的精度数据进行解析,还原模型图结构,实现模型各个层级的精度数据比对,方便用户理解模型结构、分析精度问题。
135
+
136
+ [PyTorch 场景的分级可视化构图比对](./docs/21.visualization_PyTorch.md)
137
+
138
+ [MindSpore 场景的分级可视化构图比对](./docs/22.visualization_MindSpore.md)
139
+
140
+
141
+ ### 11 单算子API自动生成脚本
99
142
 
100
- 若查看历史版本特性,请点击[安装](./docs/01.installation.md)。
143
+ 该功能将msprobe工具dump的精度数据进行解析,自动生成单API脚本,用于复现整网中出现的算子问题,降低用户复现问题的成本,供开发分析算子问题。
101
144
 
102
- 【数据采集】
103
- - 支持 config.json 中的 step 传入范围;
104
- - 优化了指定 step 的机制,指定 step 结束后工具不再采集数据,但训练会继续运行。工具结束运行后,日志提示信息如下:
105
- ```bash
106
- ****************************************
107
- * msprobe ends successfully. *
108
- ****************************************
109
- ```
110
- 注:在多卡场景,每张卡进程训练到指定 step 之后都会打印一次上述信息。
145
+ [PyTorch 单算子API自动生成脚本](./docs/23.generate_operator_PyTorch.md)
111
146
 
112
- 【精度预检】
113
- - 在 PyTorch 场景,支持部分 NPU 融合算子预检。
147
+ ### 12 数码关联
114
148
 
115
- 【精度比对】
116
- - 解决了使用 MindSpore 需要安装 PyTorch 的问题。
149
+ 该功能只支持 MindSpore 静态图场景,用于将IR图与dump数据进行关联,获取dump数据和代码调用栈的关联关系。
117
150
 
118
- 【无标杆比对】
119
- - 补充在 PyTorch 场景的性能基线报告;
120
- - 支持 MindSpore 场景的 change_value 扰动模式。
151
+ [MindSpore 场景的数码关联](./docs/24.code_mapping_Mindspore.md)
121
152
 
122
153
  ## 📑 补充材料
123
154
 
@@ -128,4 +159,4 @@ MindSpore 动态图场景的[离线预检](./docs/09.accuracy_checker_MindSpore.
128
159
 
129
160
  ## ❓ FAQ
130
161
 
131
- [FAQ for PyTorch](./docs/FAQ_PyTorch.md)
162
+ [FAQ for PyTorch](./docs/FAQ.md)
@@ -5,12 +5,11 @@
5
5
  "step": [],
6
6
  "level": "L1",
7
7
  "enable_dataloader": false,
8
- "acl_config": "",
8
+ "async_dump": false,
9
9
  "tensor": {
10
10
  "scope": [],
11
11
  "list":[],
12
- "data_mode": ["all"],
13
- "backward_input": [],
12
+ "data_mode": ["all"],
14
13
  "file_format": "npy"
15
14
  },
16
15
  "statistics": {
@@ -62,7 +62,12 @@ class Advisor:
62
62
  .format(item[CompareConst.NPU_NAME]))
63
63
 
64
64
  def gen_advisor_result(self, pd_data):
65
- first_failing_data = pd_data.iloc[0]
65
+ try:
66
+ first_failing_data = pd_data.iloc[0]
67
+ except IndexError as e:
68
+ err_msg = "index out of bounds error occurs, pd_data is empty, please check!"
69
+ logger.error(err_msg)
70
+ raise CompareException(CompareException.INDEX_OUT_OF_BOUNDS_ERROR) from e
66
71
  node_name = first_failing_data[CompareConst.NPU_NAME]
67
72
  index = first_failing_data['index']
68
73
  message = self.gen_advisor_message(node_name)
@@ -87,7 +92,7 @@ class Advisor:
87
92
  return message
88
93
 
89
94
  def analysis(self):
90
- self._check_path_vaild()
95
+ self._check_path_valid()
91
96
  analyze_data = self._parse_input_data()
92
97
  logger.info("Start analyzing the comparison result: %s" % self.file_type)
93
98
  self.analyze_unmatched(analyze_data)
@@ -119,6 +124,6 @@ class Advisor:
119
124
  df = self.input_data.reset_index()
120
125
  return df
121
126
 
122
- def _check_path_vaild(self):
127
+ def _check_path_valid(self):
123
128
  out_path_checker = FileChecker(self.out_path, FileCheckConst.DIR, FileCheckConst.WRITE_ABLE)
124
129
  out_path_checker.common_check()