mindstudio-probe 1.1.1__tar.gz → 1.2.2__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.1 → mindstudio_probe-1.2.2}/MANIFEST.in +1 -0
  2. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/PKG-INFO +3 -2
  3. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/PKG-INFO +3 -2
  4. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/SOURCES.txt +79 -5
  5. mindstudio_probe-1.2.2/msprobe/CMakeLists.txt +5 -0
  6. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/README.md +16 -21
  7. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/config.json +1 -0
  8. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/const.py +185 -11
  9. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/exceptions.py +3 -1
  10. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/file_utils.py +33 -7
  11. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/inplace_ops.yaml +4 -0
  12. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/utils.py +42 -14
  13. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common_config.py +6 -0
  14. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/acc_compare.py +139 -128
  15. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/check.py +31 -29
  16. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/compare_cli.py +17 -16
  17. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/highlight.py +186 -99
  18. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/layer_mapping/data_scope_parser.py +19 -8
  19. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/layer_mapping/layer_mapping.py +21 -14
  20. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/layer_mapping/postprocess_pass.py +4 -3
  21. mindstudio_probe-1.2.2/msprobe/core/compare/merge_result/merge_result.py +381 -0
  22. mindstudio_probe-1.2.2/msprobe/core/compare/merge_result/merge_result_cli.py +31 -0
  23. mindstudio_probe-1.2.2/msprobe/core/compare/merge_result/utils.py +81 -0
  24. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/multiprocessing_compute.py +2 -2
  25. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/npy_compare.py +109 -147
  26. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/utils.py +199 -69
  27. mindstudio_probe-1.2.2/msprobe/core/data_dump/data_collector.py +230 -0
  28. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/data_dump/data_processor/base.py +130 -28
  29. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/data_dump/data_processor/factory.py +8 -3
  30. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/data_dump/data_processor/mindspore_processor.py +170 -23
  31. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/data_dump/data_processor/pytorch_processor.py +175 -64
  32. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/data_dump/json_writer.py +54 -8
  33. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/data_dump/scope.py +19 -18
  34. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/abnormal_scene.py +9 -5
  35. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/checker.py +1 -1
  36. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/utils.py +1 -1
  37. mindstudio_probe-1.2.2/msprobe/docs/01.installation.md +213 -0
  38. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/02.config_introduction.md +18 -16
  39. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/03.config_examples.md +24 -0
  40. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/05.data_dump_PyTorch.md +107 -58
  41. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/06.data_dump_MindSpore.md +95 -34
  42. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/07.accuracy_checker_PyTorch.md +18 -18
  43. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/09.accuracy_checker_MindSpore.md +8 -6
  44. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/10.accuracy_compare_PyTorch.md +99 -41
  45. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/11.accuracy_compare_MindSpore.md +249 -48
  46. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/12.overflow_check_PyTorch.md +1 -1
  47. mindstudio_probe-1.2.2/msprobe/docs/19.monitor.md +558 -0
  48. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/21.visualization_PyTorch.md +125 -35
  49. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/22.visualization_MindSpore.md +149 -41
  50. mindstudio_probe-1.2.2/msprobe/docs/23.generate_operator_PyTorch.md +107 -0
  51. mindstudio_probe-1.2.2/msprobe/docs/24.code_mapping_Mindspore.md +28 -0
  52. mindstudio_probe-1.1.1/msprobe/docs/23.tool_function_introduction.md → mindstudio_probe-1.2.2/msprobe/docs/25.tool_function_introduction.md +1 -0
  53. mindstudio_probe-1.2.2/msprobe/docs/26.data_dump_PyTorch_baseline.md +37 -0
  54. mindstudio_probe-1.2.2/msprobe/docs/27.dump_json_instruction.md +525 -0
  55. mindstudio_probe-1.2.2/msprobe/docs/28.debugger_save_instruction.md +94 -0
  56. mindstudio_probe-1.2.2/msprobe/docs/28.kernel_dump_MindSpore.md +69 -0
  57. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/FAQ.md +26 -2
  58. mindstudio_probe-1.2.2/msprobe/docs/accuracy_checker_MindSpore/accuracy_checker_MindSpore_baseline.md +14 -0
  59. mindstudio_probe-1.2.2/msprobe/docs/data_dump_MindSpore/data_dump_MindSpore_baseline.md +22 -0
  60. mindstudio_probe-1.2.2/msprobe/docs/img/merge_result.png +0 -0
  61. mindstudio_probe-1.2.2/msprobe/docs/img/monitor/step_count_per_record.png +0 -0
  62. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/fuzzy_match_ms.png +0 -0
  63. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/fuzzy_match_pt.png +0 -0
  64. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/tensorboard_1.png +0 -0
  65. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/tensorboard_2.png +0 -0
  66. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_browser_1.png +0 -0
  67. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_browser_2.png +0 -0
  68. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_precision_info.png +0 -0
  69. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_search_info.png +0 -0
  70. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_show_info.png +0 -0
  71. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_showcase.png +0 -0
  72. mindstudio_probe-1.2.2/msprobe/docs/img/visualization/vis_unmatch_info.png +0 -0
  73. mindstudio_probe-1.2.2/msprobe/docs/visualization/GPTModel.png +0 -0
  74. mindstudio_probe-1.2.2/msprobe/docs/visualization/ParallelMLP.png +0 -0
  75. mindstudio_probe-1.2.2/msprobe/docs/visualization/layer_mapping_example.md +132 -0
  76. mindstudio_probe-1.2.2/msprobe/docs/visualization/mapping.png +0 -0
  77. mindstudio_probe-1.2.2/msprobe/docs/visualization/mapping1.png +0 -0
  78. mindstudio_probe-1.2.2/msprobe/docs/visualization/module_name.png +0 -0
  79. mindstudio_probe-1.2.2/msprobe/docs/visualization/module_name1.png +0 -0
  80. mindstudio_probe-1.2.2/msprobe/docs/visualization/no_mapping.png +0 -0
  81. mindstudio_probe-1.2.2/msprobe/docs/visualization/no_mapping1.png +0 -0
  82. mindstudio_probe-1.2.2/msprobe/docs/visualization/no_mapping_analyze.png +0 -0
  83. mindstudio_probe-1.2.2/msprobe/docs/visualization/top_layer.png +0 -0
  84. mindstudio_probe-1.2.2/msprobe/mindspore/__init__.py +28 -0
  85. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +80 -28
  86. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/api_runner.py +54 -16
  87. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/cmd_parser.py +2 -1
  88. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/compute_element.py +52 -8
  89. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/data_manager.py +37 -0
  90. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/main.py +1 -0
  91. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/multi_api_accuracy_checker.py +12 -6
  92. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/multi_data_manager.py +3 -1
  93. mindstudio_probe-1.2.2/msprobe/mindspore/api_accuracy_checker/torch_mindtorch_importer.py +129 -0
  94. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/type_mapping.py +24 -1
  95. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/utils.py +6 -1
  96. mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping/bind.py +264 -0
  97. mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping/cmd_parser.py +40 -0
  98. mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping/graph.py +49 -0
  99. mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping/graph_parser.py +226 -0
  100. mindstudio_probe-1.1.1/msprobe/mindspore/__init__.py → mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping/main.py +9 -2
  101. mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping/processor.py +34 -0
  102. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/common/const.py +3 -1
  103. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/common/utils.py +68 -5
  104. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/compare/distributed_compare.py +0 -2
  105. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/compare/ms_compare.py +105 -63
  106. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/compare/ms_graph_compare.py +14 -5
  107. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/debugger/debugger_config.py +28 -2
  108. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/debugger/precision_debugger.py +100 -12
  109. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/hook_cell/api_registry.py +85 -16
  110. mindstudio_probe-1.2.2/msprobe/mindspore/dump/hook_cell/hook_cell.py +77 -0
  111. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/hook_cell/primitive_hooks.py +33 -15
  112. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +11 -1
  113. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/hook_cell/wrap_api.py +92 -1
  114. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/jit_dump.py +7 -6
  115. mindstudio_probe-1.2.2/msprobe/mindspore/dump/kernel_dump/kernel_config.py +33 -0
  116. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/kernel_graph_dump.py +7 -0
  117. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/api_pynative_self_check.py +13 -4
  118. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +2 -2
  119. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/grad_probe/grad_analyzer.py +24 -12
  120. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/grad_probe/hook.py +13 -4
  121. mindstudio_probe-1.2.2/msprobe/mindspore/mindtorch/__init__.py +18 -0
  122. mindstudio_probe-1.2.2/msprobe/mindspore/mindtorch/mindtorch_adaptor.py +255 -0
  123. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/anomaly_detect.py +404 -0
  124. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/distributed/distributed_ops.yaml +15 -0
  125. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/distributed/stack_blacklist.yaml +5 -0
  126. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/distributed/wrap_distributed.py +300 -0
  127. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/features.py +63 -0
  128. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/module_hook.py +821 -0
  129. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/module_spec_verifier.py +94 -0
  130. mindstudio_probe-1.2.2/msprobe/mindspore/monitor/utils.py +267 -0
  131. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/ms_config.py +13 -3
  132. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +7 -0
  133. mindstudio_probe-1.2.2/msprobe/mindspore/service.py +543 -0
  134. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/msprobe.py +24 -3
  135. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/__init__.py +7 -7
  136. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/common/utils.py +31 -16
  137. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +41 -8
  138. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +100 -267
  139. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +4 -1
  140. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/compare.py +69 -68
  141. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +54 -0
  142. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/compare/compare_input.py +51 -0
  143. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +2 -4
  144. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/generate_op_script/op_generator.py +55 -31
  145. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/absolute_threshold.py +106 -0
  146. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/accumulative_error_compare.py +107 -0
  147. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/base_standard.py +151 -0
  148. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/benchmark_compare.py +226 -0
  149. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/binary_consistency.py +68 -0
  150. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/standard_config.py +218 -0
  151. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/standard_register.py +104 -0
  152. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/thousandth_standard.py +63 -0
  153. mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/precision_standard/ulp_compare.py +200 -0
  154. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +57 -1
  155. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +2 -1
  156. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +42 -14
  157. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +64 -19
  158. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +34 -4
  159. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +5 -3
  160. mindstudio_probe-1.2.2/msprobe/pytorch/bench_functions/apply_adam.py +215 -0
  161. mindstudio_probe-1.2.2/msprobe/pytorch/bench_functions/group_norm_silu.py +27 -0
  162. mindstudio_probe-1.2.2/msprobe/pytorch/bench_functions/mish.py +21 -0
  163. mindstudio_probe-1.2.2/msprobe/pytorch/bench_functions/moe_gating_top_k_softmax.py +44 -0
  164. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/npu_fusion_attention.py +42 -10
  165. mindstudio_probe-1.2.2/msprobe/pytorch/bench_functions/sort_v2.py +21 -0
  166. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/common/parse_json.py +2 -1
  167. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/common/utils.py +116 -2
  168. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/compare/distributed_compare.py +17 -29
  169. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/compare/pt_compare.py +40 -20
  170. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/debugger/debugger_config.py +42 -17
  171. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/debugger/precision_debugger.py +56 -12
  172. mindstudio_probe-1.2.2/msprobe/pytorch/dump/module_dump/module_dump.py +86 -0
  173. mindstudio_probe-1.2.2/msprobe/pytorch/dump/module_dump/module_processer.py +204 -0
  174. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/common/params.py +2 -1
  175. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/common/utils.py +3 -0
  176. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/compare/grad_saver.py +0 -2
  177. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +31 -47
  178. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +0 -4
  179. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/function_factory.py +7 -1
  180. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/__init__.py +1 -1
  181. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/hook_module.py +14 -11
  182. mindstudio_probe-1.2.2/msprobe/pytorch/hook_module/register_optimizer_hook.py +59 -0
  183. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/support_wrap_ops.yaml +36 -1
  184. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/wrap_distributed.py +10 -8
  185. mindstudio_probe-1.2.2/msprobe/pytorch/hook_module/wrap_functional.py +66 -0
  186. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/anomaly_analyse.py +1 -1
  187. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/anomaly_detect.py +98 -28
  188. mindstudio_probe-1.2.2/msprobe/pytorch/monitor/csv2tb.py +164 -0
  189. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/distributed/wrap_distributed.py +25 -14
  190. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/features.py +3 -3
  191. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/module_hook.py +543 -318
  192. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/module_metric.py +27 -48
  193. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/module_spec_verifier.py +3 -1
  194. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/optimizer_collect.py +76 -56
  195. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/unittest/test_monitor.py +24 -9
  196. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/utils.py +84 -48
  197. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/dispatch.py +8 -2
  198. mindstudio_probe-1.2.2/msprobe/pytorch/parse_tool/__init__.py +0 -0
  199. mindstudio_probe-1.2.2/msprobe/pytorch/parse_tool/lib/__init__.py +0 -0
  200. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/compare.py +10 -10
  201. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/config.py +5 -7
  202. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/file_desc.py +15 -1
  203. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/interactive_cli.py +10 -10
  204. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/parse_exception.py +7 -7
  205. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/parse_tool.py +11 -10
  206. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/utils.py +18 -19
  207. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/lib/visualization.py +9 -10
  208. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/pt_config.py +19 -22
  209. mindstudio_probe-1.2.2/msprobe/pytorch/service.py +470 -0
  210. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/builder/graph_builder.py +93 -10
  211. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/builder/msprobe_adapter.py +30 -6
  212. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/compare/graph_comparator.py +64 -14
  213. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/compare/mode_adapter.py +1 -15
  214. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/graph/base_node.py +15 -19
  215. mindstudio_probe-1.2.2/msprobe/visualization/graph/distributed_analyzer.py +395 -0
  216. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/graph/graph.py +9 -0
  217. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/graph/node_op.py +4 -2
  218. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/graph_service.py +100 -27
  219. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/utils.py +24 -31
  220. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/setup.py +45 -3
  221. mindstudio_probe-1.1.1/msprobe/core/data_dump/data_collector.py +0 -155
  222. mindstudio_probe-1.1.1/msprobe/docs/01.installation.md +0 -109
  223. mindstudio_probe-1.1.1/msprobe/docs/19.monitor.md +0 -468
  224. mindstudio_probe-1.1.1/msprobe/mindspore/dump/hook_cell/hook_cell.py +0 -55
  225. mindstudio_probe-1.1.1/msprobe/mindspore/service.py +0 -303
  226. mindstudio_probe-1.1.1/msprobe/pytorch/functional/module_dump.py +0 -84
  227. mindstudio_probe-1.1.1/msprobe/pytorch/hook_module/wrap_functional.py +0 -106
  228. mindstudio_probe-1.1.1/msprobe/pytorch/module_processer.py +0 -150
  229. mindstudio_probe-1.1.1/msprobe/pytorch/service.py +0 -321
  230. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/LICENSE +0 -0
  231. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/dependency_links.txt +0 -0
  232. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/entry_points.txt +0 -0
  233. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/not-zip-safe +0 -0
  234. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/requires.txt +0 -0
  235. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/mindstudio_probe.egg-info/top_level.txt +0 -0
  236. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/__init__.py +0 -0
  237. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/__init__.py +0 -0
  238. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/advisor/advisor.py +0 -0
  239. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/advisor/advisor_const.py +0 -0
  240. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/advisor/advisor_result.py +0 -0
  241. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/inplace_op_checker.py +0 -0
  242. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/common/log.py +0 -0
  243. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/compare/layer_mapping/__init__.py +0 -0
  244. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/grad_probe/__init__.py +0 -0
  245. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/grad_probe/constant.py +0 -0
  246. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/grad_probe/grad_compare.py +0 -0
  247. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/grad_probe/utils.py +0 -0
  248. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/api_info.py +0 -0
  249. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/filter.py +0 -0
  250. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/ignore_rules.yaml +0 -0
  251. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/core/overflow_check/level.py +0 -0
  252. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/04.kernel_dump_PyTorch.md +0 -0
  253. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/08.accuracy_checker_online_PyTorch.md +0 -0
  254. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/13.overflow_check_MindSpore.md +0 -0
  255. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/14.data_parse_PyTorch.md +0 -0
  256. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/15.free_benchmarking_PyTorch.md +0 -0
  257. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/16.free_benchmarking_MindSpore.md +0 -0
  258. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/17.grad_probe.md +0 -0
  259. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/18.online_dispatch.md +0 -0
  260. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/20.monitor_performance_baseline.md +0 -0
  261. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +0 -0
  262. {mindstudio_probe-1.1.1/msprobe/docs/data_dump_Mindspore → mindstudio_probe-1.2.2/msprobe/docs/data_dump_MindSpore}/dynamic_graph_quick_start_example.md +0 -0
  263. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/BLOOM-7B_1.png +0 -0
  264. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/BLOOM-7B_2.png +0 -0
  265. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/BLOOM-7B_3.png +0 -0
  266. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/BLOOM-7B_4.png +0 -0
  267. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_1.png +0 -0
  268. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_2.png +0 -0
  269. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_3.png +0 -0
  270. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_4.png +0 -0
  271. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_5.png +0 -0
  272. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_6.png +0 -0
  273. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_7.png +0 -0
  274. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/GPT-3_8.png +0 -0
  275. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/YOLOV5S_1.png +0 -0
  276. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/YOLOV5S_2.png +0 -0
  277. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/accuracy_checking_details.png +0 -0
  278. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/accuracy_checking_result.png +0 -0
  279. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/api_precision_compare_details.png +0 -0
  280. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/api_precision_compare_result.png +0 -0
  281. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/auto_analyze_log.png +0 -0
  282. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/compare_result.png +0 -0
  283. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/compare_result_pkl.png +0 -0
  284. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/compare_result_pkl_md5.png.png +0 -0
  285. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/cpu_info.png +0 -0
  286. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/free_benchmark.png +0 -0
  287. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/free_benchmark_framework.png +0 -0
  288. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/grad_probe_image-1.png +0 -0
  289. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/grad_probe_image-2.png +0 -0
  290. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/grad_probe_image-3.png +0 -0
  291. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/grad_probe_image-4.png +0 -0
  292. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/grad_probe_image.png +0 -0
  293. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/module_compare.png +0 -0
  294. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/monitor/cpu_info.png +0 -0
  295. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/ms_dump.png +0 -0
  296. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/ms_layer.png +0 -0
  297. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/docs/img/pt_dump.png +0 -0
  298. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/__init__.py +0 -0
  299. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/api_info.py +0 -0
  300. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +0 -0
  301. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/api_accuracy_checker/checker_support_api.yaml +0 -0
  302. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/cell_processor.py +0 -0
  303. {mindstudio_probe-1.1.1/msprobe/mindspore/debugger → mindstudio_probe-1.2.2/msprobe/mindspore/code_mapping}/__init__.py +0 -0
  304. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/common/log.py +0 -0
  305. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/compare/ms_to_pt_api.yaml +0 -0
  306. {mindstudio_probe-1.1.1/msprobe/mindspore/dump → mindstudio_probe-1.2.2/msprobe/mindspore/debugger}/__init__.py +0 -0
  307. {mindstudio_probe-1.1.1/msprobe/mindspore/free_benchmark → mindstudio_probe-1.2.2/msprobe/mindspore/dump}/__init__.py +0 -0
  308. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/dump_tool_factory.py +0 -0
  309. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dump/kernel_kbyk_dump.py +0 -0
  310. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dym_loader/hook_dynamic_loader.cc +0 -0
  311. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/dym_loader/hook_dynamic_loader.h +0 -0
  312. {mindstudio_probe-1.1.1/msprobe/mindspore/free_benchmark/common → mindstudio_probe-1.2.2/msprobe/mindspore/free_benchmark}/__init__.py +0 -0
  313. {mindstudio_probe-1.1.1/msprobe/mindspore/free_benchmark/handler → mindstudio_probe-1.2.2/msprobe/mindspore/free_benchmark/common}/__init__.py +0 -0
  314. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/common/config.py +0 -0
  315. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/common/handler_params.py +0 -0
  316. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/common/utils.py +0 -0
  317. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +0 -0
  318. {mindstudio_probe-1.1.1/msprobe/mindspore/grad_probe → mindstudio_probe-1.2.2/msprobe/mindspore/free_benchmark/handler}/__init__.py +0 -0
  319. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/handler/base_handler.py +0 -0
  320. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/handler/check_handler.py +0 -0
  321. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/handler/fix_handler.py +0 -0
  322. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/handler/handler_factory.py +0 -0
  323. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/add_noise.py +0 -0
  324. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +0 -0
  325. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +0 -0
  326. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +0 -0
  327. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/no_change.py +0 -0
  328. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +0 -0
  329. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/free_benchmark/self_check_tool_factory.py +0 -0
  330. {mindstudio_probe-1.1.1/msprobe/mindspore/overflow_check → mindstudio_probe-1.2.2/msprobe/mindspore/grad_probe}/__init__.py +0 -0
  331. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/grad_probe/global_context.py +0 -0
  332. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/grad_probe/grad_monitor.py +0 -0
  333. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/grad_probe/grad_stat_csv.py +0 -0
  334. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/grad_probe/utils.py +0 -0
  335. {mindstudio_probe-1.1.1/msprobe/pytorch/api_accuracy_checker → mindstudio_probe-1.2.2/msprobe/mindspore/monitor/distributed}/__init__.py +0 -0
  336. {mindstudio_probe-1.1.1/msprobe/pytorch/api_accuracy_checker/common → mindstudio_probe-1.2.2/msprobe/mindspore/overflow_check}/__init__.py +0 -0
  337. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +0 -0
  338. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/runtime.py +0 -0
  339. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/mindspore/task_handler_factory.py +0 -0
  340. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/.keep +0 -0
  341. {mindstudio_probe-1.1.1/msprobe/pytorch/api_accuracy_checker/compare → mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker}/__init__.py +0 -0
  342. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/common/.keep +0 -0
  343. {mindstudio_probe-1.1.1/msprobe/pytorch/api_accuracy_checker/run_ut → mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/common}/__init__.py +0 -0
  344. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/common/config.py +0 -0
  345. {mindstudio_probe-1.1.1/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer → mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/compare}/__init__.py +0 -0
  346. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +0 -0
  347. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/config.yaml +0 -0
  348. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/generate_op_script/config_op.json +0 -0
  349. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/generate_op_script/operator_replication.template +0 -0
  350. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/.keep +0 -0
  351. {mindstudio_probe-1.1.1/msprobe/pytorch/debugger → mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/run_ut}/__init__.py +0 -0
  352. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +0 -0
  353. {mindstudio_probe-1.1.1/msprobe/pytorch/free_benchmark/common → mindstudio_probe-1.2.2/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer}/__init__.py +0 -0
  354. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +0 -0
  355. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +0 -0
  356. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py +0 -0
  357. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +0 -0
  358. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml +0 -0
  359. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py +0 -0
  360. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/__init__.py +0 -0
  361. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/apply_adam_w.py +0 -0
  362. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/confusion_transpose.py +0 -0
  363. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/fast_gelu.py +0 -0
  364. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/layer_norm_eval.py +0 -0
  365. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/linear.py +0 -0
  366. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/matmul_backward.py +0 -0
  367. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/rms_norm.py +0 -0
  368. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/rotary_mul.py +0 -0
  369. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/scaled_mask_softmax.py +0 -0
  370. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/bench_functions/swiglu.py +0 -0
  371. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/common/__init__.py +0 -0
  372. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/common/compare_script.template +0 -0
  373. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/common/log.py +0 -0
  374. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/compare/mapping.yaml +0 -0
  375. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/compare/match.py +0 -0
  376. {mindstudio_probe-1.1.1/msprobe/pytorch/free_benchmark/perturbed_layers → mindstudio_probe-1.2.2/msprobe/pytorch/debugger}/__init__.py +0 -0
  377. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/dump/kernel_dump/kernel_config.py +0 -0
  378. {mindstudio_probe-1.1.1/msprobe/pytorch/free_benchmark/perturbed_layers/npu → mindstudio_probe-1.2.2/msprobe/pytorch/dump/module_dump}/__init__.py +0 -0
  379. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/__init__.py +0 -0
  380. {mindstudio_probe-1.1.1/msprobe/pytorch/free_benchmark/result_handlers → mindstudio_probe-1.2.2/msprobe/pytorch/free_benchmark/common}/__init__.py +0 -0
  381. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/common/constant.py +0 -0
  382. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/common/counter.py +0 -0
  383. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/common/enums.py +0 -0
  384. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/compare/single_benchmark.py +0 -0
  385. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/main.py +0 -0
  386. {mindstudio_probe-1.1.1/msprobe/pytorch/functional → mindstudio_probe-1.2.2/msprobe/pytorch/free_benchmark/perturbed_layers}/__init__.py +0 -0
  387. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +0 -0
  388. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +0 -0
  389. {mindstudio_probe-1.1.1/msprobe/pytorch/grad_probe → mindstudio_probe-1.2.2/msprobe/pytorch/free_benchmark/perturbed_layers/npu}/__init__.py +0 -0
  390. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +0 -0
  391. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +0 -0
  392. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +0 -0
  393. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +0 -0
  394. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +0 -0
  395. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +0 -0
  396. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +0 -0
  397. {mindstudio_probe-1.1.1/msprobe/pytorch/monitor → mindstudio_probe-1.2.2/msprobe/pytorch/free_benchmark/result_handlers}/__init__.py +0 -0
  398. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +0 -0
  399. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +0 -0
  400. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +0 -0
  401. {mindstudio_probe-1.1.1/msprobe/pytorch/monitor/distributed → mindstudio_probe-1.2.2/msprobe/pytorch/grad_probe}/__init__.py +0 -0
  402. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/grad_probe/grad_monitor.py +0 -0
  403. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/grad_probe/grad_stat_csv.py +0 -0
  404. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/api_registry.py +0 -0
  405. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/utils.py +0 -0
  406. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/wrap_aten.py +0 -0
  407. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/wrap_npu_custom.py +0 -0
  408. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/wrap_tensor.py +0 -0
  409. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/wrap_torch.py +0 -0
  410. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/hook_module/wrap_vf.py +0 -0
  411. {mindstudio_probe-1.1.1/msprobe/pytorch/monitor/unittest → mindstudio_probe-1.2.2/msprobe/pytorch/monitor}/__init__.py +0 -0
  412. {mindstudio_probe-1.1.1/msprobe/pytorch/parse_tool → mindstudio_probe-1.2.2/msprobe/pytorch/monitor/distributed}/__init__.py +0 -0
  413. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/distributed/distributed_ops.yaml +0 -0
  414. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/distributed/stack_blacklist.yaml +0 -0
  415. {mindstudio_probe-1.1.1/msprobe/pytorch/parse_tool/lib → mindstudio_probe-1.2.2/msprobe/pytorch/monitor/unittest}/__init__.py +0 -0
  416. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/monitor/visualizer.py +0 -0
  417. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/__init__.py +0 -0
  418. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/compare.py +0 -0
  419. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/dump_compare.py +0 -0
  420. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/single_compare.py +0 -0
  421. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/torch_ops_config.yaml +0 -0
  422. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/online_dispatch/utils.py +0 -0
  423. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse.py +0 -0
  424. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/pytorch/parse_tool/cli.py +0 -0
  425. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/__init__.py +0 -0
  426. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/builder/__init__.py +0 -0
  427. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/compare/__init__.py +0 -0
  428. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/graph/__init__.py +0 -0
  429. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/msprobe/visualization/graph/node_colors.py +0 -0
  430. {mindstudio_probe-1.1.1 → mindstudio_probe-1.2.2}/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,7 +1,7 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mindstudio-probe
3
- Version: 1.1.1
4
- Summary: Pytorch Ascend Probe Utils
3
+ Version: 1.2.2
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
@@ -11,6 +11,7 @@ Classifier: Intended Audience :: Developers
11
11
  Classifier: Intended Audience :: Education
12
12
  Classifier: Intended Audience :: Science/Research
13
13
  Classifier: Programming Language :: Python :: 3
14
+ Classifier: Programming Language :: C++
14
15
  Classifier: Topic :: Scientific/Engineering
15
16
  Classifier: Topic :: Scientific/Engineering :: Mathematics
16
17
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -1,7 +1,7 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mindstudio-probe
3
- Version: 1.1.1
4
- Summary: Pytorch Ascend Probe Utils
3
+ Version: 1.2.2
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
@@ -11,6 +11,7 @@ Classifier: Intended Audience :: Developers
11
11
  Classifier: Intended Audience :: Education
12
12
  Classifier: Intended Audience :: Science/Research
13
13
  Classifier: Programming Language :: Python :: 3
14
+ Classifier: Programming Language :: C++
14
15
  Classifier: Topic :: Scientific/Engineering
15
16
  Classifier: Topic :: Scientific/Engineering :: Mathematics
16
17
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -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
@@ -35,6 +36,9 @@ msprobe/core/compare/layer_mapping/__init__.py
35
36
  msprobe/core/compare/layer_mapping/data_scope_parser.py
36
37
  msprobe/core/compare/layer_mapping/layer_mapping.py
37
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
41
+ msprobe/core/compare/merge_result/utils.py
38
42
  msprobe/core/data_dump/data_collector.py
39
43
  msprobe/core/data_dump/json_writer.py
40
44
  msprobe/core/data_dump/scope.py
@@ -75,10 +79,18 @@ msprobe/docs/19.monitor.md
75
79
  msprobe/docs/20.monitor_performance_baseline.md
76
80
  msprobe/docs/21.visualization_PyTorch.md
77
81
  msprobe/docs/22.visualization_MindSpore.md
78
- msprobe/docs/23.tool_function_introduction.md
82
+ msprobe/docs/23.generate_operator_PyTorch.md
83
+ msprobe/docs/24.code_mapping_Mindspore.md
84
+ msprobe/docs/25.tool_function_introduction.md
85
+ msprobe/docs/26.data_dump_PyTorch_baseline.md
86
+ msprobe/docs/27.dump_json_instruction.md
87
+ msprobe/docs/28.debugger_save_instruction.md
88
+ msprobe/docs/28.kernel_dump_MindSpore.md
79
89
  msprobe/docs/FAQ.md
80
90
  msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md
81
- msprobe/docs/data_dump_Mindspore/dynamic_graph_quick_start_example.md
91
+ msprobe/docs/accuracy_checker_MindSpore/accuracy_checker_MindSpore_baseline.md
92
+ msprobe/docs/data_dump_MindSpore/data_dump_MindSpore_baseline.md
93
+ msprobe/docs/data_dump_MindSpore/dynamic_graph_quick_start_example.md
82
94
  msprobe/docs/img/BLOOM-7B_1.png
83
95
  msprobe/docs/img/BLOOM-7B_2.png
84
96
  msprobe/docs/img/BLOOM-7B_3.png
@@ -109,11 +121,35 @@ msprobe/docs/img/grad_probe_image-2.png
109
121
  msprobe/docs/img/grad_probe_image-3.png
110
122
  msprobe/docs/img/grad_probe_image-4.png
111
123
  msprobe/docs/img/grad_probe_image.png
124
+ msprobe/docs/img/merge_result.png
112
125
  msprobe/docs/img/module_compare.png
113
126
  msprobe/docs/img/ms_dump.png
114
127
  msprobe/docs/img/ms_layer.png
115
128
  msprobe/docs/img/pt_dump.png
116
129
  msprobe/docs/img/monitor/cpu_info.png
130
+ msprobe/docs/img/monitor/step_count_per_record.png
131
+ msprobe/docs/img/visualization/fuzzy_match_ms.png
132
+ msprobe/docs/img/visualization/fuzzy_match_pt.png
133
+ msprobe/docs/img/visualization/tensorboard_1.png
134
+ msprobe/docs/img/visualization/tensorboard_2.png
135
+ msprobe/docs/img/visualization/vis_browser_1.png
136
+ msprobe/docs/img/visualization/vis_browser_2.png
137
+ msprobe/docs/img/visualization/vis_precision_info.png
138
+ msprobe/docs/img/visualization/vis_search_info.png
139
+ msprobe/docs/img/visualization/vis_show_info.png
140
+ msprobe/docs/img/visualization/vis_showcase.png
141
+ msprobe/docs/img/visualization/vis_unmatch_info.png
142
+ msprobe/docs/visualization/GPTModel.png
143
+ msprobe/docs/visualization/ParallelMLP.png
144
+ msprobe/docs/visualization/layer_mapping_example.md
145
+ msprobe/docs/visualization/mapping.png
146
+ msprobe/docs/visualization/mapping1.png
147
+ msprobe/docs/visualization/module_name.png
148
+ msprobe/docs/visualization/module_name1.png
149
+ msprobe/docs/visualization/no_mapping.png
150
+ msprobe/docs/visualization/no_mapping1.png
151
+ msprobe/docs/visualization/no_mapping_analyze.png
152
+ msprobe/docs/visualization/top_layer.png
117
153
  msprobe/mindspore/__init__.py
118
154
  msprobe/mindspore/cell_processor.py
119
155
  msprobe/mindspore/ms_config.py
@@ -132,8 +168,16 @@ msprobe/mindspore/api_accuracy_checker/data_manager.py
132
168
  msprobe/mindspore/api_accuracy_checker/main.py
133
169
  msprobe/mindspore/api_accuracy_checker/multi_api_accuracy_checker.py
134
170
  msprobe/mindspore/api_accuracy_checker/multi_data_manager.py
171
+ msprobe/mindspore/api_accuracy_checker/torch_mindtorch_importer.py
135
172
  msprobe/mindspore/api_accuracy_checker/type_mapping.py
136
173
  msprobe/mindspore/api_accuracy_checker/utils.py
174
+ msprobe/mindspore/code_mapping/__init__.py
175
+ msprobe/mindspore/code_mapping/bind.py
176
+ msprobe/mindspore/code_mapping/cmd_parser.py
177
+ msprobe/mindspore/code_mapping/graph.py
178
+ msprobe/mindspore/code_mapping/graph_parser.py
179
+ msprobe/mindspore/code_mapping/main.py
180
+ msprobe/mindspore/code_mapping/processor.py
137
181
  msprobe/mindspore/common/const.py
138
182
  msprobe/mindspore/common/log.py
139
183
  msprobe/mindspore/common/utils.py
@@ -154,6 +198,7 @@ msprobe/mindspore/dump/hook_cell/hook_cell.py
154
198
  msprobe/mindspore/dump/hook_cell/primitive_hooks.py
155
199
  msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml
156
200
  msprobe/mindspore/dump/hook_cell/wrap_api.py
201
+ msprobe/mindspore/dump/kernel_dump/kernel_config.py
157
202
  msprobe/mindspore/dym_loader/hook_dynamic_loader.cc
158
203
  msprobe/mindspore/dym_loader/hook_dynamic_loader.h
159
204
  msprobe/mindspore/free_benchmark/__init__.py
@@ -183,12 +228,22 @@ msprobe/mindspore/grad_probe/grad_monitor.py
183
228
  msprobe/mindspore/grad_probe/grad_stat_csv.py
184
229
  msprobe/mindspore/grad_probe/hook.py
185
230
  msprobe/mindspore/grad_probe/utils.py
231
+ msprobe/mindspore/mindtorch/__init__.py
232
+ msprobe/mindspore/mindtorch/mindtorch_adaptor.py
233
+ msprobe/mindspore/monitor/anomaly_detect.py
234
+ msprobe/mindspore/monitor/features.py
235
+ msprobe/mindspore/monitor/module_hook.py
236
+ msprobe/mindspore/monitor/module_spec_verifier.py
237
+ msprobe/mindspore/monitor/utils.py
238
+ msprobe/mindspore/monitor/distributed/__init__.py
239
+ msprobe/mindspore/monitor/distributed/distributed_ops.yaml
240
+ msprobe/mindspore/monitor/distributed/stack_blacklist.yaml
241
+ msprobe/mindspore/monitor/distributed/wrap_distributed.py
186
242
  msprobe/mindspore/overflow_check/__init__.py
187
243
  msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py
188
244
  msprobe/mindspore/overflow_check/overflow_check_tool_factory.py
189
245
  msprobe/pytorch/__init__.py
190
246
  msprobe/pytorch/function_factory.py
191
- msprobe/pytorch/module_processer.py
192
247
  msprobe/pytorch/parse.py
193
248
  msprobe/pytorch/pt_config.py
194
249
  msprobe/pytorch/service.py
@@ -206,10 +261,20 @@ msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml
206
261
  msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml
207
262
  msprobe/pytorch/api_accuracy_checker/compare/compare.py
208
263
  msprobe/pytorch/api_accuracy_checker/compare/compare_column.py
264
+ msprobe/pytorch/api_accuracy_checker/compare/compare_input.py
209
265
  msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py
210
266
  msprobe/pytorch/api_accuracy_checker/generate_op_script/config_op.json
211
267
  msprobe/pytorch/api_accuracy_checker/generate_op_script/op_generator.py
212
268
  msprobe/pytorch/api_accuracy_checker/generate_op_script/operator_replication.template
269
+ msprobe/pytorch/api_accuracy_checker/precision_standard/absolute_threshold.py
270
+ msprobe/pytorch/api_accuracy_checker/precision_standard/accumulative_error_compare.py
271
+ msprobe/pytorch/api_accuracy_checker/precision_standard/base_standard.py
272
+ msprobe/pytorch/api_accuracy_checker/precision_standard/benchmark_compare.py
273
+ msprobe/pytorch/api_accuracy_checker/precision_standard/binary_consistency.py
274
+ msprobe/pytorch/api_accuracy_checker/precision_standard/standard_config.py
275
+ msprobe/pytorch/api_accuracy_checker/precision_standard/standard_register.py
276
+ msprobe/pytorch/api_accuracy_checker/precision_standard/thousandth_standard.py
277
+ msprobe/pytorch/api_accuracy_checker/precision_standard/ulp_compare.py
213
278
  msprobe/pytorch/api_accuracy_checker/run_ut/.keep
214
279
  msprobe/pytorch/api_accuracy_checker/run_ut/__init__.py
215
280
  msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py
@@ -227,16 +292,21 @@ msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py
227
292
  msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml
228
293
  msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py
229
294
  msprobe/pytorch/bench_functions/__init__.py
295
+ msprobe/pytorch/bench_functions/apply_adam.py
230
296
  msprobe/pytorch/bench_functions/apply_adam_w.py
231
297
  msprobe/pytorch/bench_functions/confusion_transpose.py
232
298
  msprobe/pytorch/bench_functions/fast_gelu.py
299
+ msprobe/pytorch/bench_functions/group_norm_silu.py
233
300
  msprobe/pytorch/bench_functions/layer_norm_eval.py
234
301
  msprobe/pytorch/bench_functions/linear.py
235
302
  msprobe/pytorch/bench_functions/matmul_backward.py
303
+ msprobe/pytorch/bench_functions/mish.py
304
+ msprobe/pytorch/bench_functions/moe_gating_top_k_softmax.py
236
305
  msprobe/pytorch/bench_functions/npu_fusion_attention.py
237
306
  msprobe/pytorch/bench_functions/rms_norm.py
238
307
  msprobe/pytorch/bench_functions/rotary_mul.py
239
308
  msprobe/pytorch/bench_functions/scaled_mask_softmax.py
309
+ msprobe/pytorch/bench_functions/sort_v2.py
240
310
  msprobe/pytorch/bench_functions/swiglu.py
241
311
  msprobe/pytorch/common/__init__.py
242
312
  msprobe/pytorch/common/compare_script.template
@@ -251,6 +321,9 @@ msprobe/pytorch/debugger/__init__.py
251
321
  msprobe/pytorch/debugger/debugger_config.py
252
322
  msprobe/pytorch/debugger/precision_debugger.py
253
323
  msprobe/pytorch/dump/kernel_dump/kernel_config.py
324
+ msprobe/pytorch/dump/module_dump/__init__.py
325
+ msprobe/pytorch/dump/module_dump/module_dump.py
326
+ msprobe/pytorch/dump/module_dump/module_processer.py
254
327
  msprobe/pytorch/free_benchmark/__init__.py
255
328
  msprobe/pytorch/free_benchmark/main.py
256
329
  msprobe/pytorch/free_benchmark/common/__init__.py
@@ -278,14 +351,13 @@ msprobe/pytorch/free_benchmark/result_handlers/check_handler.py
278
351
  msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py
279
352
  msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py
280
353
  msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py
281
- msprobe/pytorch/functional/__init__.py
282
- msprobe/pytorch/functional/module_dump.py
283
354
  msprobe/pytorch/grad_probe/__init__.py
284
355
  msprobe/pytorch/grad_probe/grad_monitor.py
285
356
  msprobe/pytorch/grad_probe/grad_stat_csv.py
286
357
  msprobe/pytorch/hook_module/__init__.py
287
358
  msprobe/pytorch/hook_module/api_registry.py
288
359
  msprobe/pytorch/hook_module/hook_module.py
360
+ msprobe/pytorch/hook_module/register_optimizer_hook.py
289
361
  msprobe/pytorch/hook_module/support_wrap_ops.yaml
290
362
  msprobe/pytorch/hook_module/utils.py
291
363
  msprobe/pytorch/hook_module/wrap_aten.py
@@ -298,6 +370,7 @@ msprobe/pytorch/hook_module/wrap_vf.py
298
370
  msprobe/pytorch/monitor/__init__.py
299
371
  msprobe/pytorch/monitor/anomaly_analyse.py
300
372
  msprobe/pytorch/monitor/anomaly_detect.py
373
+ msprobe/pytorch/monitor/csv2tb.py
301
374
  msprobe/pytorch/monitor/features.py
302
375
  msprobe/pytorch/monitor/module_hook.py
303
376
  msprobe/pytorch/monitor/module_metric.py
@@ -340,6 +413,7 @@ msprobe/visualization/compare/graph_comparator.py
340
413
  msprobe/visualization/compare/mode_adapter.py
341
414
  msprobe/visualization/graph/__init__.py
342
415
  msprobe/visualization/graph/base_node.py
416
+ msprobe/visualization/graph/distributed_analyzer.py
343
417
  msprobe/visualization/graph/graph.py
344
418
  msprobe/visualization/graph/node_colors.py
345
419
  msprobe/visualization/graph/node_op.py
@@ -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()
@@ -15,7 +15,7 @@ debugger = PrecisionDebugger(config_path='./config.json')
15
15
  ...
16
16
  debugger.start() # 一般在训练循环开头启动工具
17
17
  ... # 循环体
18
- debugger.stop() # 一般在训练循环末尾结束工具
18
+ debugger.stop() # 一般在训练循环末尾结束工具。必须调用,否则可能导致精度数据落盘不全
19
19
  debugger.step() # 在训练循环的最后需要重置工具,非循环场景不需要
20
20
  ```
21
21
 
@@ -51,15 +51,21 @@ export MSPROBE_LOG_LEVEL={x}
51
51
 
52
52
  **1. Pytorch 框架下,工具暂不支持 Fully Sharded Data Parallel(FSDP)。**
53
53
 
54
+ **2. 工具读写的所有路径,如config_path、dump_path等,只允许包含大小写字母、数字、下划线、斜杠、点和短横线。**
55
+
54
56
  ## ⚙️ [安装](./docs/01.installation.md)
55
57
 
58
+ ## 🌟 新版本特性
59
+
60
+ 请参见[特性变更说明](./docs/01.installation.md#特性变更说明)。
61
+
56
62
  ## 🛠️ config.json [介绍](./docs/02.config_introduction.md) 和 [示例](./docs/03.config_examples.md)
57
63
 
58
64
  ## 🧰 主要功能
59
65
 
60
66
  ### 0 用前必看
61
67
 
62
- 使用工具前,建议先浏览[**工具功能模块简介、适用场景和当前版本局限性**](./docs/23.tool_function_introduction.md),了解功能特性。
68
+ 使用工具前,建议先浏览[**工具功能模块简介、适用场景和当前版本局限性**](./docs/25.tool_function_introduction.md),了解功能特性。
63
69
 
64
70
  ### 1 数据采集
65
71
 
@@ -121,7 +127,7 @@ MindSpore 动态图场景的[离线预检](./docs/09.accuracy_checker_MindSpore.
121
127
 
122
128
  该功能收集和聚合模型训练过程中的网络层,优化器, 通信算子的中间值,帮助诊断模型训练过程中计算, 通信,优化器各部分出现的异常情况。
123
129
 
124
- [PyTorch 场景的训练状态监控](./docs/19.monitor.md)
130
+ [兼容 PyTorch 和 MindSpore 框架的训练状态监控](./docs/19.monitor.md)
125
131
 
126
132
  ### 10 分级可视化构图比对
127
133
 
@@ -131,29 +137,18 @@ MindSpore 动态图场景的[离线预检](./docs/09.accuracy_checker_MindSpore.
131
137
 
132
138
  [MindSpore 场景的分级可视化构图比对](./docs/22.visualization_MindSpore.md)
133
139
 
134
- ## 🌟 新版本特性
135
140
 
136
- 若查看历史版本特性,请参见[安装](./docs/01.installation.md)。
141
+ ### 11 单算子API自动生成脚本
142
+
143
+ 该功能将msprobe工具dump的精度数据进行解析,自动生成单API脚本,用于复现整网中出现的算子问题,降低用户复现问题的成本,供开发分析算子问题。
137
144
 
138
- 【数据采集】
139
- - 支持 config.json 中的 step 传入范围;
140
- - 优化了指定 step 的机制,指定 step 结束后工具不再采集数据,但训练会继续运行。工具结束运行后,日志提示信息如下:
141
- ```bash
142
- ****************************************
143
- * msprobe ends successfully. *
144
- ****************************************
145
- ```
146
- 注:在多卡场景,每张卡进程训练到指定 step 之后都会打印一次上述信息。
145
+ [PyTorch 单算子API自动生成脚本](./docs/23.generate_operator_PyTorch.md)
147
146
 
148
- 【精度预检】
149
- - 在 PyTorch 场景,支持部分 NPU 融合算子预检。
147
+ ### 12 数码关联
150
148
 
151
- 【精度比对】
152
- - 解决了使用 MindSpore 需要安装 PyTorch 的问题。
149
+ 该功能只支持 MindSpore 静态图场景,用于将IR图与dump数据进行关联,获取dump数据和代码调用栈的关联关系。
153
150
 
154
- 【无标杆比对】
155
- - 补充在 PyTorch 场景的性能基线报告;
156
- - 支持 MindSpore 场景的 change_value 扰动模式。
151
+ [MindSpore 场景的数码关联](./docs/24.code_mapping_Mindspore.md)
157
152
 
158
153
  ## 📑 补充材料
159
154
 
@@ -5,6 +5,7 @@
5
5
  "step": [],
6
6
  "level": "L1",
7
7
  "enable_dataloader": false,
8
+ "async_dump": false,
8
9
  "tensor": {
9
10
  "scope": [],
10
11
  "list":[],