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.
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/MANIFEST.in +1 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/PKG-INFO +16 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/PKG-INFO +16 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/SOURCES.txt +119 -10
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/entry_points.txt +0 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/requires.txt +2 -0
- mindstudio_probe-1.2.1/msprobe/CMakeLists.txt +5 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/README.md +51 -20
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/config.json +2 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/advisor/advisor.py +8 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/const.py +264 -15
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/exceptions.py +27 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/file_utils.py +176 -26
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/inplace_op_checker.py +15 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/inplace_ops.yaml +3 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/log.py +27 -9
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common/utils.py +204 -77
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/common_config.py +49 -14
- mindstudio_probe-1.2.1/msprobe/core/compare/acc_compare.py +549 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/compare/check.py +32 -33
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/compare/compare_cli.py +32 -14
- mindstudio_probe-1.2.1/msprobe/core/compare/highlight.py +415 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/__init__.py +19 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/data_scope_parser.py +246 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/layer_mapping.py +249 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/layer_mapping/postprocess_pass.py +95 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/merge_result/merge_result.py +380 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/merge_result/merge_result_cli.py +31 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/compare/multiprocessing_compute.py +2 -2
- mindstudio_probe-1.2.1/msprobe/core/compare/npy_compare.py +301 -0
- mindstudio_probe-1.2.1/msprobe/core/compare/utils.py +609 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_collector.py +60 -28
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_processor/base.py +84 -36
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_processor/factory.py +5 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/data_processor/mindspore_processor.py +152 -18
- mindstudio_probe-1.2.1/msprobe/core/data_dump/data_processor/pytorch_processor.py +540 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/data_dump/json_writer.py +29 -1
- mindstudio_probe-1.2.1/msprobe/core/data_dump/scope.py +274 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/constant.py +27 -13
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/grad_compare.py +18 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/utils.py +30 -2
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/abnormal_scene.py +189 -0
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/api_info.py +55 -0
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/checker.py +138 -0
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/filter.py +157 -0
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/ignore_rules.yaml +55 -0
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/level.py +22 -0
- mindstudio_probe-1.2.1/msprobe/core/overflow_check/utils.py +28 -0
- mindstudio_probe-1.2.1/msprobe/docs/01.installation.md +188 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/02.config_introduction.md +50 -23
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/03.config_examples.md +2 -9
- mindstudio_probe-1.2.1/msprobe/docs/04.kernel_dump_PyTorch.md +73 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/05.data_dump_PyTorch.md +93 -61
- mindstudio_probe-1.2.1/msprobe/docs/06.data_dump_MindSpore.md +390 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/07.accuracy_checker_PyTorch.md +28 -28
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/08.accuracy_checker_online_PyTorch.md +1 -6
- mindstudio_probe-1.2.1/msprobe/docs/09.accuracy_checker_MindSpore.md +104 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/10.accuracy_compare_PyTorch.md +114 -50
- mindstudio_probe-1.2.1/msprobe/docs/11.accuracy_compare_MindSpore.md +625 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/12.overflow_check_PyTorch.md +2 -2
- mindstudio_probe-1.2.1/msprobe/docs/13.overflow_check_MindSpore.md +31 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/15.free_benchmarking_PyTorch.md +4 -5
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/16.free_benchmarking_MindSpore.md +56 -37
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/17.grad_probe.md +5 -6
- mindstudio_probe-1.2.1/msprobe/docs/19.monitor.md +561 -0
- mindstudio_probe-1.2.1/msprobe/docs/20.monitor_performance_baseline.md +52 -0
- mindstudio_probe-1.2.1/msprobe/docs/21.visualization_PyTorch.md +466 -0
- mindstudio_probe-1.2.1/msprobe/docs/22.visualization_MindSpore.md +481 -0
- mindstudio_probe-1.2.1/msprobe/docs/23.generate_operator_PyTorch.md +107 -0
- mindstudio_probe-1.2.1/msprobe/docs/24.code_mapping_Mindspore.md +28 -0
- mindstudio_probe-1.2.1/msprobe/docs/25.tool_function_introduction.md +29 -0
- mindstudio_probe-1.2.1/msprobe/docs/26.data_dump_PyTorch_baseline.md +37 -0
- mindstudio_probe-1.2.1/msprobe/docs/27.dump_json_instruction.md +521 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/FAQ.md +29 -2
- mindstudio_probe-1.2.1/msprobe/docs/accuracy_checker_MindSpore/accuracy_checker_MindSpore_baseline.md +14 -0
- mindstudio_probe-1.2.1/msprobe/docs/data_dump_MindSpore/data_dump_MindSpore_baseline.md +22 -0
- mindstudio_probe-1.2.1/msprobe/docs/data_dump_MindSpore/dynamic_graph_quick_start_example.md +211 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/compare_result.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/merge_result.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/monitor/cpu_info.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/fuzzy_match_ms.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/fuzzy_match_pt.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/tensorboard_1.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/tensorboard_2.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_browser_1.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_browser_2.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_precision_info.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_search_info.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_show_info.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_showcase.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/img/visualization/vis_unmatch_info.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/GPTModel.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/ParallelMLP.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/layer_mapping_example.md +132 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/mapping.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/mapping1.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/module_name.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/module_name1.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/no_mapping.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/no_mapping1.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/no_mapping_analyze.png +0 -0
- mindstudio_probe-1.2.1/msprobe/docs/visualization/top_layer.png +0 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/__init__.py +27 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +279 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/api_info.py +21 -6
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/api_runner.py +43 -18
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +21 -7
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/checker_support_api.yaml +77 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/cmd_parser.py +69 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/compute_element.py +64 -31
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/data_manager.py +301 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/main.py +34 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/multi_api_accuracy_checker.py +212 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/api_accuracy_checker/multi_data_manager.py +60 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/type_mapping.py +22 -5
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/utils.py +34 -17
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/cell_processor.py +33 -12
- mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/bind.py +264 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/cmd_parser.py +40 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/graph.py +49 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/graph_parser.py +226 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/main.py +24 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping/processor.py +34 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/common/const.py +35 -13
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/common/log.py +5 -9
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/common/utils.py +88 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/compare/distributed_compare.py +22 -24
- mindstudio_probe-1.2.1/msprobe/mindspore/compare/ms_compare.py +422 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/compare/ms_graph_compare.py +95 -52
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/debugger/debugger_config.py +7 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/debugger/precision_debugger.py +87 -12
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/dump_tool_factory.py +3 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/api_registry.py +95 -18
- mindstudio_probe-1.2.1/msprobe/mindspore/dump/hook_cell/hook_cell.py +77 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/primitive_hooks.py +45 -30
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +36 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/hook_cell/wrap_api.py +92 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/jit_dump.py +17 -5
- mindstudio_probe-1.2.1/msprobe/mindspore/dump/kernel_dump/kernel_config.py +33 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/kernel_graph_dump.py +9 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/dump/kernel_kbyk_dump.py +2 -4
- mindstudio_probe-1.2.1/msprobe/mindspore/dym_loader/hook_dynamic_loader.cc +140 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/dym_loader/hook_dynamic_loader.h +53 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/free_benchmark/api_pynative_self_check.py +246 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/common/handler_params.py +1 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/common/utils.py +19 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +0 -204
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/base_handler.py +3 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/check_handler.py +4 -5
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/fix_handler.py +4 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/handler_factory.py +4 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/add_noise.py +2 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +15 -6
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +2 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +2 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +13 -6
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +2 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/self_check_tool_factory.py +2 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/global_context.py +28 -8
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/grad_analyzer.py +50 -24
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/grad_monitor.py +16 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/grad_stat_csv.py +33 -5
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/hook.py +35 -12
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/utils.py +18 -5
- mindstudio_probe-1.2.1/msprobe/mindspore/mindtorch/__init__.py +18 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/mindtorch/mindtorch_adaptor.py +255 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/ms_config.py +27 -16
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +9 -4
- mindstudio_probe-1.2.1/msprobe/mindspore/runtime.py +19 -0
- mindstudio_probe-1.2.1/msprobe/mindspore/service.py +469 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/task_handler_factory.py +15 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/msprobe.py +48 -10
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/__init__.py +8 -6
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/common/config.py +132 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/common/utils.py +31 -16
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +41 -8
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +103 -271
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +4 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/compare.py +69 -68
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +54 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/compare/compare_input.py +51 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +2 -4
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/generate_op_script/config_op.json +9 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/generate_op_script/op_generator.py +478 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/generate_op_script/operator_replication.template +365 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/absolute_threshold.py +106 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/accumulative_error_compare.py +107 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/base_standard.py +151 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/benchmark_compare.py +226 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/binary_consistency.py +68 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/standard_config.py +218 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/standard_register.py +104 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/thousandth_standard.py +63 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/api_accuracy_checker/precision_standard/ulp_compare.py +200 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +63 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +21 -15
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +54 -22
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +140 -71
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +49 -8
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +9 -24
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +4 -12
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +5 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/dump_dispatch.py +9 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +3 -11
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/utils.py +2 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/confusion_transpose.py +5 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/matmul_backward.py +12 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/npu_fusion_attention.py +142 -16
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/rotary_mul.py +4 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/swiglu.py +10 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/parse_json.py +7 -6
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/utils.py +101 -7
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/distributed_compare.py +17 -30
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/pt_compare.py +44 -22
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/debugger/debugger_config.py +46 -27
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/debugger/precision_debugger.py +42 -12
- mindstudio_probe-1.2.1/msprobe/pytorch/dump/kernel_dump/kernel_config.py +33 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/dump/module_dump/module_dump.py +86 -0
- {mindstudio-probe-1.1.0/msprobe/pytorch → mindstudio_probe-1.2.1/msprobe/pytorch/dump/module_dump}/module_processer.py +81 -10
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/constant.py +15 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/counter.py +15 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/enums.py +15 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/params.py +10 -2
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/common/utils.py +29 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/compare/grad_saver.py +20 -5
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/compare/single_benchmark.py +2 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +3 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +6 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +2 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +4 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +41 -47
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +6 -5
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +0 -4
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/grad_probe/grad_monitor.py +23 -6
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/grad_probe/grad_stat_csv.py +40 -10
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/__init__.py +1 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/hook_module.py +14 -11
- mindstudio_probe-1.2.1/msprobe/pytorch/hook_module/register_optimizer_hook.py +59 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/support_wrap_ops.yaml +35 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_distributed.py +6 -8
- mindstudio_probe-1.2.1/msprobe/pytorch/hook_module/wrap_functional.py +66 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/anomaly_analyse.py +201 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/anomaly_detect.py +425 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/csv2tb.py +166 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed/distributed_ops.yaml +19 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed/stack_blacklist.yaml +5 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed/wrap_distributed.py +283 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/features.py +108 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/module_hook.py +1076 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/module_metric.py +172 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/module_spec_verifier.py +95 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/optimizer_collect.py +333 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/unittest/__init__.py +0 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/unittest/test_monitor.py +160 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/utils.py +321 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/monitor/visualizer.py +59 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/__init__.py +2 -3
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/compare.py +29 -38
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/dispatch.py +58 -27
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/dump_compare.py +21 -9
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/single_compare.py +53 -32
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/torch_ops_config.yaml +1 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/online_dispatch/utils.py +49 -21
- mindstudio_probe-1.2.1/msprobe/pytorch/parse_tool/__init__.py +0 -0
- mindstudio_probe-1.2.1/msprobe/pytorch/parse_tool/lib/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/compare.py +21 -27
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/config.py +6 -8
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/file_desc.py +15 -1
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/interactive_cli.py +10 -10
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/parse_exception.py +7 -7
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/parse_tool.py +12 -12
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/utils.py +33 -53
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/lib/visualization.py +11 -10
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/pt_config.py +31 -8
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/service.py +188 -108
- mindstudio_probe-1.2.1/msprobe/visualization/__init__.py +14 -0
- mindstudio_probe-1.2.1/msprobe/visualization/builder/__init__.py +14 -0
- mindstudio_probe-1.2.1/msprobe/visualization/builder/graph_builder.py +222 -0
- mindstudio_probe-1.2.1/msprobe/visualization/builder/msprobe_adapter.py +227 -0
- mindstudio_probe-1.2.1/msprobe/visualization/compare/__init__.py +14 -0
- mindstudio_probe-1.2.1/msprobe/visualization/compare/graph_comparator.py +180 -0
- mindstudio_probe-1.2.1/msprobe/visualization/compare/mode_adapter.py +197 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph/__init__.py +14 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph/base_node.py +119 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph/distributed_analyzer.py +318 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph/graph.py +209 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph/node_colors.py +95 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph/node_op.py +39 -0
- mindstudio_probe-1.2.1/msprobe/visualization/graph_service.py +288 -0
- mindstudio_probe-1.2.1/msprobe/visualization/utils.py +217 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/setup.py +48 -7
- mindstudio-probe-1.1.0/msprobe/core/compare/acc_compare.py +0 -473
- mindstudio-probe-1.1.0/msprobe/core/compare/highlight.py +0 -259
- mindstudio-probe-1.1.0/msprobe/core/compare/npy_compare.py +0 -310
- mindstudio-probe-1.1.0/msprobe/core/compare/utils.py +0 -464
- mindstudio-probe-1.1.0/msprobe/core/data_dump/data_processor/pytorch_processor.py +0 -383
- mindstudio-probe-1.1.0/msprobe/core/data_dump/scope.py +0 -194
- mindstudio-probe-1.1.0/msprobe/docs/01.installation.md +0 -99
- mindstudio-probe-1.1.0/msprobe/docs/04.acl_config_examples.md +0 -78
- mindstudio-probe-1.1.0/msprobe/docs/06.data_dump_MindSpore.md +0 -285
- mindstudio-probe-1.1.0/msprobe/docs/09.accuracy_checker_MindSpore.md +0 -68
- mindstudio-probe-1.1.0/msprobe/docs/11.accuracy_compare_MindSpore.md +0 -333
- mindstudio-probe-1.1.0/msprobe/docs/13.overflow_check_MindSpore.md +0 -31
- mindstudio-probe-1.1.0/msprobe/mindspore/__init__.py +0 -2
- mindstudio-probe-1.1.0/msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +0 -279
- mindstudio-probe-1.1.0/msprobe/mindspore/api_accuracy_checker/cmd_parser.py +0 -6
- mindstudio-probe-1.1.0/msprobe/mindspore/api_accuracy_checker/main.py +0 -9
- mindstudio-probe-1.1.0/msprobe/mindspore/compare/layer_mapping.py +0 -146
- mindstudio-probe-1.1.0/msprobe/mindspore/compare/modify_mapping.py +0 -107
- mindstudio-probe-1.1.0/msprobe/mindspore/compare/ms_compare.py +0 -357
- mindstudio-probe-1.1.0/msprobe/mindspore/dump/hook_cell/hook_cell.py +0 -55
- mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/api_pynative_self_check.py +0 -131
- mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/decorator/dec_forward.py +0 -57
- mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +0 -122
- mindstudio-probe-1.1.0/msprobe/mindspore/runtime.py +0 -4
- mindstudio-probe-1.1.0/msprobe/mindspore/service.py +0 -297
- mindstudio-probe-1.1.0/msprobe/pytorch/api_accuracy_checker/common/config.py +0 -70
- mindstudio-probe-1.1.0/msprobe/pytorch/functional/module_dump.py +0 -84
- mindstudio-probe-1.1.0/msprobe/pytorch/hook_module/wrap_functional.py +0 -104
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/LICENSE +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/dependency_links.txt +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/not-zip-safe +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/mindstudio_probe.egg-info/top_level.txt +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/advisor/advisor_const.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/advisor/advisor_result.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/core/grad_probe/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/14.data_parse_PyTorch.md +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/18.online_dispatch.md +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_1.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_2.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_3.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/BLOOM-7B_4.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_1.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_2.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_3.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_4.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_5.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_6.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_7.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/GPT-3_8.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/YOLOV5S_1.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/YOLOV5S_2.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/accuracy_checking_details.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/accuracy_checking_result.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/api_precision_compare_details.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/api_precision_compare_result.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/auto_analyze_log.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/compare_result_pkl.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/compare_result_pkl_md5.png.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/cpu_info.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/free_benchmark.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/free_benchmark_framework.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-1.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-2.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-3.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image-4.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/grad_probe_image.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/module_compare.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/ms_dump.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/ms_layer.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/docs/img/pt_dump.png +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/api_accuracy_checker/__init__.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/mindspore/debugger → mindstudio_probe-1.2.1/msprobe/mindspore/code_mapping}/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/compare/ms_to_pt_api.yaml +0 -0
- {mindstudio-probe-1.1.0/msprobe/mindspore/dump → mindstudio_probe-1.2.1/msprobe/mindspore/debugger}/__init__.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark → mindstudio_probe-1.2.1/msprobe/mindspore/dump}/__init__.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/common → mindstudio_probe-1.2.1/msprobe/mindspore/free_benchmark}/__init__.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/mindspore/free_benchmark/decorator → mindstudio_probe-1.2.1/msprobe/mindspore/free_benchmark/common}/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/common/config.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/handler/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/free_benchmark/perturbation/no_change.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/grad_probe/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/overflow_check/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/.keep +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/common/.keep +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/common/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/config.yaml +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/.keep +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/torch_ops_config.yaml +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/apply_adam_w.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/fast_gelu.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/layer_norm_eval.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/linear.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/rms_norm.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/bench_functions/scaled_mask_softmax.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/compare_script.template +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/common/log.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/mapping.yaml +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/compare/match.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/debugger/__init__.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/pytorch/free_benchmark/common → mindstudio_probe-1.2.1/msprobe/pytorch/dump/module_dump}/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/__init__.py +0 -0
- {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
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/main.py +0 -0
- {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
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +0 -0
- {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
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/pytorch/functional → mindstudio_probe-1.2.1/msprobe/pytorch/free_benchmark/result_handlers}/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/function_factory.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/grad_probe/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/api_registry.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/utils.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_aten.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_npu_custom.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_tensor.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_torch.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/hook_module/wrap_vf.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/pytorch/parse_tool → mindstudio_probe-1.2.1/msprobe/pytorch/monitor}/__init__.py +0 -0
- {mindstudio-probe-1.1.0/msprobe/pytorch/parse_tool/lib → mindstudio_probe-1.2.1/msprobe/pytorch/monitor/distributed}/__init__.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/msprobe/pytorch/parse_tool/cli.py +0 -0
- {mindstudio-probe-1.1.0 → mindstudio_probe-1.2.1}/setup.cfg +0 -0
|
@@ -1,17 +1,17 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: mindstudio-probe
|
|
3
|
-
Version: 1.1
|
|
4
|
-
Summary:
|
|
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
|
|
4
|
-
Summary:
|
|
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.
|
|
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
|
|
@@ -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
|
-
|
|
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/
|
|
162
|
+
[FAQ for PyTorch](./docs/FAQ.md)
|
|
@@ -5,12 +5,11 @@
|
|
|
5
5
|
"step": [],
|
|
6
6
|
"level": "L1",
|
|
7
7
|
"enable_dataloader": false,
|
|
8
|
-
"
|
|
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
|
-
|
|
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.
|
|
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
|
|
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()
|