PyPI - mindstudio-probe - Versions diffs - 1.0.1__py3-none-any.whl → 1.0.4__py3-none-any.whl - Mend

mindstudio-probe 1.0.1py3-none-any.whl → 1.0.4py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (323) hide show

{mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/LICENSE +201 -201
{mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/METADATA +36 -30
mindstudio_probe-1.0.4.dist-info/RECORD +276 -0
{mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/WHEEL +1 -1
{mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/entry_points.txt +1 -0
msprobe/README.md +101 -182
msprobe/__init__.py +1 -0
msprobe/{config/config.json → config.json} +49 -27
msprobe/core/__init__.py +0 -0
msprobe/{pytorch → core}/advisor/advisor.py +124 -124
msprobe/{pytorch → core}/advisor/advisor_const.py +59 -59
msprobe/{pytorch → core}/advisor/advisor_result.py +58 -58
msprobe/core/common/const.py +341 -241
msprobe/core/common/exceptions.py +100 -88
msprobe/core/common/{file_check.py → file_utils.py} +478 -265
msprobe/core/common/log.py +76 -55
msprobe/core/common/utils.py +385 -516
msprobe/core/common_config.py +85 -58
msprobe/core/compare/acc_compare.py +300 -0
msprobe/core/compare/check.py +95 -0
msprobe/core/compare/compare_cli.py +49 -0
msprobe/core/compare/highlight.py +223 -0
msprobe/core/compare/multiprocessing_compute.py +149 -0
msprobe/{pytorch → core}/compare/npy_compare.py +295 -244
msprobe/core/compare/utils.py +430 -0
msprobe/core/data_dump/data_collector.py +154 -140
msprobe/core/data_dump/data_processor/base.py +314 -245
msprobe/core/data_dump/data_processor/factory.py +59 -61
msprobe/core/data_dump/data_processor/mindspore_processor.py +186 -0
msprobe/core/data_dump/data_processor/pytorch_processor.py +366 -346
msprobe/core/data_dump/json_writer.py +96 -116
msprobe/core/data_dump/scope.py +178 -178
msprobe/core/grad_probe/__init__.py +0 -0
msprobe/core/grad_probe/constant.py +71 -0
msprobe/core/grad_probe/grad_compare.py +171 -0
msprobe/core/grad_probe/utils.py +64 -0
msprobe/docs/01.installation.md +89 -0
msprobe/docs/02.config_introduction.md +165 -0
msprobe/docs/03.config_examples.md +247 -0
msprobe/docs/04.acl_config_examples.md +76 -0
msprobe/docs/05.data_dump_PyTorch.md +198 -0
msprobe/docs/06.data_dump_MindSpore.md +243 -0
msprobe/docs/07.accuracy_checker_PyTorch.md +274 -0
msprobe/docs/08.accuracy_checker_online_PyTorch.md +198 -0
msprobe/docs/09.accuracy_checker_MindSpore.md +68 -0
msprobe/docs/10.accuracy_compare_PyTorch.md +245 -0
msprobe/docs/11.accuracy_compare_MindSpore.md +202 -0
msprobe/docs/12.overflow_check_PyTorch.md +79 -0
msprobe/docs/13.overflow_check_MindSpore.md +31 -0
msprobe/{pytorch/doc/parse_tool.md → docs/14.data_parse_PyTorch.md} +283 -286
msprobe/docs/15.free_benchmarking_PyTorch.md +164 -0
msprobe/docs/17.grad_probe.md +207 -0
msprobe/docs/FAQ_PyTorch.md +177 -0
msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +146 -0
msprobe/docs/img/free_benchmark_framework.png +0 -0
msprobe/docs/img/grad_probe_image-1.png +0 -0
msprobe/docs/img/grad_probe_image-2.png +0 -0
msprobe/docs/img/grad_probe_image-3.png +0 -0
msprobe/docs/img/grad_probe_image-4.png +0 -0
msprobe/docs/img/grad_probe_image.png +0 -0
msprobe/mindspore/__init__.py +1 -1
msprobe/mindspore/api_accuracy_checker/__init__.py +0 -0
msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +255 -0
msprobe/mindspore/api_accuracy_checker/api_info.py +69 -0
msprobe/mindspore/api_accuracy_checker/api_runner.py +156 -0
msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +197 -0
msprobe/mindspore/api_accuracy_checker/cmd_parser.py +6 -0
msprobe/mindspore/api_accuracy_checker/compute_element.py +239 -0
msprobe/mindspore/api_accuracy_checker/main.py +9 -0
msprobe/mindspore/api_accuracy_checker/type_mapping.py +114 -0
msprobe/mindspore/api_accuracy_checker/utils.py +80 -0
msprobe/mindspore/cell_processor.py +34 -0
msprobe/mindspore/common/const.py +106 -0
msprobe/mindspore/common/log.py +38 -0
msprobe/mindspore/common/utils.py +81 -0
msprobe/mindspore/compare/distributed_compare.py +75 -0
msprobe/mindspore/compare/ms_compare.py +219 -0
msprobe/mindspore/compare/ms_graph_compare.py +348 -0
msprobe/mindspore/compare/ms_to_pt_api.yaml +399 -0
msprobe/mindspore/debugger/debugger_config.py +66 -51
msprobe/mindspore/debugger/precision_debugger.py +126 -32
msprobe/mindspore/dump/dump_tool_factory.py +35 -38
msprobe/mindspore/dump/hook_cell/api_registry.py +118 -0
msprobe/mindspore/dump/hook_cell/hook_cell.py +55 -0
msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +922 -0
msprobe/mindspore/dump/hook_cell/wrap_api.py +113 -0
msprobe/mindspore/dump/jit_dump.py +72 -0
msprobe/mindspore/dump/kernel_graph_dump.py +59 -60
msprobe/mindspore/dump/kernel_kbyk_dump.py +64 -0
msprobe/mindspore/free_benchmark/__init__.py +0 -0
msprobe/mindspore/free_benchmark/api_pynative_self_check.py +116 -0
msprobe/mindspore/free_benchmark/common/__init__.py +0 -0
msprobe/mindspore/free_benchmark/common/config.py +12 -0
msprobe/mindspore/free_benchmark/common/handler_params.py +17 -0
msprobe/mindspore/free_benchmark/common/utils.py +71 -0
msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +842 -0
msprobe/mindspore/free_benchmark/decorator/__init__.py +0 -0
msprobe/mindspore/free_benchmark/decorator/dec_forward.py +43 -0
msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +107 -0
msprobe/mindspore/free_benchmark/handler/__init__.py +0 -0
msprobe/mindspore/free_benchmark/handler/base_handler.py +90 -0
msprobe/mindspore/free_benchmark/handler/check_handler.py +41 -0
msprobe/mindspore/free_benchmark/handler/fix_handler.py +36 -0
msprobe/mindspore/free_benchmark/handler/handler_factory.py +21 -0
msprobe/mindspore/free_benchmark/perturbation/add_noise.py +67 -0
msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +21 -0
msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +63 -0
msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +51 -0
msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +35 -0
msprobe/mindspore/free_benchmark/perturbation/no_change.py +12 -0
msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +29 -0
msprobe/mindspore/free_benchmark/self_check_tool_factory.py +33 -0
msprobe/mindspore/grad_probe/__init__.py +0 -0
msprobe/mindspore/grad_probe/global_context.py +90 -0
msprobe/mindspore/grad_probe/grad_analyzer.py +231 -0
msprobe/mindspore/grad_probe/grad_monitor.py +27 -0
msprobe/mindspore/grad_probe/grad_stat_csv.py +132 -0
msprobe/mindspore/grad_probe/hook.py +94 -0
msprobe/mindspore/grad_probe/utils.py +30 -0
msprobe/mindspore/ms_config.py +128 -78
msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +44 -45
msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +34 -32
msprobe/mindspore/runtime.py +4 -0
msprobe/mindspore/service.py +378 -0
msprobe/mindspore/task_handler_factory.py +24 -21
msprobe/msprobe.py +105 -67
msprobe/pytorch/__init__.py +4 -4
msprobe/pytorch/api_accuracy_checker/common/config.py +53 -50
msprobe/pytorch/api_accuracy_checker/common/utils.py +214 -224
msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +213 -216
msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +606 -545
msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +132 -132
msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +390 -390
msprobe/pytorch/api_accuracy_checker/compare/compare.py +386 -345
msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +73 -73
msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +245 -248
msprobe/pytorch/api_accuracy_checker/config.yaml +10 -4
msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +335 -328
msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +200 -203
msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +133 -127
msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +592 -493
msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +70 -7
msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +7 -4
msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/__init__.py +0 -0
msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +197 -0
msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +325 -0
msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +204 -0
msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +219 -0
msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/ssl_config.py +10 -0
msprobe/pytorch/bench_functions/__init__.py +15 -0
msprobe/pytorch/bench_functions/apply_adam_w.py +28 -0
msprobe/pytorch/bench_functions/confusion_transpose.py +19 -0
msprobe/pytorch/bench_functions/fast_gelu.py +55 -0
msprobe/pytorch/bench_functions/layer_norm_eval.py +6 -0
msprobe/pytorch/bench_functions/linear.py +12 -0
msprobe/pytorch/bench_functions/matmul_backward.py +48 -0
msprobe/pytorch/bench_functions/npu_fusion_attention.py +509 -0
msprobe/pytorch/bench_functions/rms_norm.py +15 -0
msprobe/pytorch/bench_functions/rotary_mul.py +52 -0
msprobe/pytorch/bench_functions/scaled_mask_softmax.py +26 -0
msprobe/pytorch/bench_functions/swiglu.py +55 -0
msprobe/pytorch/common/__init__.py +2 -2
msprobe/pytorch/common/compare_script.template +14 -14
msprobe/pytorch/common/log.py +20 -31
msprobe/pytorch/common/parse_json.py +39 -37
msprobe/pytorch/common/utils.py +305 -224
msprobe/pytorch/compare/distributed_compare.py +66 -111
msprobe/pytorch/compare/mapping.yaml +607 -607
msprobe/pytorch/compare/match.py +34 -36
msprobe/pytorch/compare/pt_compare.py +50 -0
msprobe/pytorch/debugger/debugger_config.py +95 -86
msprobe/pytorch/debugger/precision_debugger.py +125 -95
msprobe/pytorch/free_benchmark/__init__.py +8 -8
msprobe/pytorch/free_benchmark/common/constant.py +70 -67
msprobe/pytorch/free_benchmark/common/counter.py +71 -71
msprobe/pytorch/free_benchmark/common/enums.py +37 -37
msprobe/pytorch/free_benchmark/common/params.py +129 -129
msprobe/pytorch/free_benchmark/common/utils.py +102 -98
msprobe/pytorch/free_benchmark/compare/grad_saver.py +179 -183
msprobe/pytorch/free_benchmark/compare/single_benchmark.py +104 -104
msprobe/pytorch/free_benchmark/main.py +105 -102
msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +13 -13
msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +41 -41
msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +90 -90
msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +104 -104
msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +63 -63
msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +68 -68
msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +28 -28
msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +45 -45
msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +19 -19
msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +217 -203
msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +39 -39
msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +23 -23
msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +30 -31
msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +170 -170
msprobe/pytorch/function_factory.py +76 -0
msprobe/pytorch/functional/dump_module.py +39 -39
msprobe/pytorch/grad_probe/__init__.py +0 -0
msprobe/pytorch/grad_probe/grad_monitor.py +91 -0
msprobe/pytorch/grad_probe/grad_stat_csv.py +129 -0
msprobe/pytorch/hook_module/api_registry.py +161 -161
msprobe/pytorch/hook_module/hook_module.py +120 -109
msprobe/pytorch/hook_module/support_wrap_ops.yaml +1879 -1876
msprobe/pytorch/hook_module/utils.py +30 -29
msprobe/pytorch/hook_module/wrap_aten.py +110 -100
msprobe/pytorch/hook_module/wrap_distributed.py +78 -75
msprobe/pytorch/hook_module/wrap_functional.py +105 -108
msprobe/pytorch/hook_module/wrap_npu_custom.py +93 -73
msprobe/pytorch/hook_module/wrap_tensor.py +71 -72
msprobe/pytorch/hook_module/wrap_torch.py +86 -88
msprobe/pytorch/hook_module/wrap_vf.py +62 -64
msprobe/pytorch/module_processer.py +138 -98
msprobe/pytorch/online_dispatch/__init__.py +20 -20
msprobe/pytorch/online_dispatch/compare.py +236 -236
msprobe/pytorch/online_dispatch/dispatch.py +271 -273
msprobe/pytorch/online_dispatch/dump_compare.py +155 -186
msprobe/pytorch/online_dispatch/single_compare.py +391 -391
msprobe/pytorch/online_dispatch/torch_ops_config.yaml +49 -49
msprobe/pytorch/online_dispatch/utils.py +130 -187
msprobe/pytorch/parse.py +4 -4
msprobe/pytorch/parse_tool/cli.py +32 -32
msprobe/pytorch/parse_tool/lib/compare.py +260 -259
msprobe/pytorch/parse_tool/lib/config.py +52 -51
msprobe/pytorch/parse_tool/lib/file_desc.py +31 -31
msprobe/pytorch/parse_tool/lib/interactive_cli.py +102 -102
msprobe/pytorch/parse_tool/lib/parse_exception.py +54 -54
msprobe/pytorch/parse_tool/lib/parse_tool.py +158 -158
msprobe/pytorch/parse_tool/lib/utils.py +316 -367
msprobe/pytorch/parse_tool/lib/visualization.py +85 -90
msprobe/pytorch/pt_config.py +188 -93
msprobe/pytorch/service.py +246 -167
mindstudio_probe-1.0.1.dist-info/RECORD +0 -228
msprobe/config/README.md +0 -397
msprobe/mindspore/doc/dump.md +0 -65
msprobe/mindspore/dump/api_kbk_dump.py +0 -55
msprobe/pytorch/compare/acc_compare.py +0 -1024
msprobe/pytorch/compare/highlight.py +0 -100
msprobe/pytorch/doc/FAQ.md +0 -193
msprobe/pytorch/doc/api_accuracy_checker.md +0 -269
msprobe/pytorch/doc/atat/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/205/342/225/226/320/265/321/205/320/225/342/225/226/321/206/320/245/342/226/221/321/206/320/235/320/276dump/321/206/320/260/320/227/321/205/320/227/320/226/321/206/320/220/320/267/321/210/320/223/342/225/234/321/205/320/257/342/225/221/321/207/342/225/221/342/224/220/321/206/320/232/320/265/321/205/320/241/320/232.md +0 -182
msprobe/pytorch/doc/dump.md +0 -207
msprobe/pytorch/doc/ptdbg_ascend_compare.md +0 -176
msprobe/pytorch/doc/ptdbg_ascend_overview.md +0 -68
msprobe/pytorch/doc/ptdbg_ascend_quickstart.md +0 -381
msprobe/pytorch/doc/run_overflow_check.md +0 -25
msprobe/pytorch/doc//321/205/320/254/320/270/321/207/342/225/221/342/224/220/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/206/320/277/320/244/321/205/320/277/342/225/243.md +0 -90
msprobe/test/core_ut/common/test_utils.py +0 -345
msprobe/test/core_ut/data_dump/test_data_collector.py +0 -47
msprobe/test/core_ut/data_dump/test_json_writer.py +0 -183
msprobe/test/core_ut/data_dump/test_scope.py +0 -151
msprobe/test/core_ut/test_common_config.py +0 -152
msprobe/test/core_ut/test_file_check.py +0 -218
msprobe/test/core_ut/test_log.py +0 -109
msprobe/test/mindspore_ut/test_api_kbk_dump.py +0 -51
msprobe/test/mindspore_ut/test_debugger_config.py +0 -42
msprobe/test/mindspore_ut/test_dump_tool_factory.py +0 -51
msprobe/test/mindspore_ut/test_kernel_graph_dump.py +0 -66
msprobe/test/mindspore_ut/test_kernel_graph_overflow_check.py +0 -63
msprobe/test/mindspore_ut/test_ms_config.py +0 -69
msprobe/test/mindspore_ut/test_overflow_check_tool_factory.py +0 -51
msprobe/test/mindspore_ut/test_precision_debugger.py +0 -56
msprobe/test/mindspore_ut/test_task_handler_factory.py +0 -58
msprobe/test/pytorch_ut/advisor/test_advisor.py +0 -83
msprobe/test/pytorch_ut/api_accuracy_checker/common/test_common_utils.py +0 -108
msprobe/test/pytorch_ut/api_accuracy_checker/common/test_config.py +0 -39
msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_algorithm.py +0 -112
msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_api_precision_compare.py +0 -77
msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_compare.py +0 -125
msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_compare_column.py +0 -10
msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_compare_utils.py +0 -43
msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/dump.json +0 -179
msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/forward.json +0 -63
msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/test_data_generate.py +0 -99
msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/test_multi_run_ut.py +0 -115
msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/test_run_ut.py +0 -72
msprobe/test/pytorch_ut/compare/test_acc_compare.py +0 -17
msprobe/test/pytorch_ut/free_benchmark/perturbed_layers/test_perturbed_layser.py +0 -105
msprobe/test/pytorch_ut/free_benchmark/result_handlers/test_result_handler.py +0 -121
msprobe/test/pytorch_ut/free_benchmark/test_main.py +0 -101
msprobe/test/pytorch_ut/functional/test_dump_module.py +0 -15
msprobe/test/pytorch_ut/hook_module/test_api_registry.py +0 -130
msprobe/test/pytorch_ut/hook_module/test_hook_module.py +0 -42
msprobe/test/pytorch_ut/hook_module/test_wrap_aten.py +0 -65
msprobe/test/pytorch_ut/hook_module/test_wrap_distributed.py +0 -35
msprobe/test/pytorch_ut/hook_module/test_wrap_functional.py +0 -20
msprobe/test/pytorch_ut/hook_module/test_wrap_tensor.py +0 -35
msprobe/test/pytorch_ut/hook_module/test_wrap_torch.py +0 -43
msprobe/test/pytorch_ut/hook_module/test_wrap_vf.py +0 -11
msprobe/test/pytorch_ut/test_pt_config.py +0 -69
msprobe/test/pytorch_ut/test_service.py +0 -59
msprobe/test/resources/advisor.txt +0 -3
msprobe/test/resources/compare_result_20230703104808.csv +0 -9
msprobe/test/resources/compare_result_without_accuracy.csv +0 -9
msprobe/test/resources/config.yaml +0 -3
msprobe/test/resources/npu_test.pkl +0 -8
msprobe/test/run_test.sh +0 -30
msprobe/test/run_ut.py +0 -58
msprobe/test/test_module_processer.py +0 -64
{mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/top_level.txt +0 -0
/msprobe/{pytorch/doc → docs}/img/BLOOM-7B_1.png +0 -0
/msprobe/{pytorch/doc → docs}/img/BLOOM-7B_2.png +0 -0
/msprobe/{pytorch/doc → docs}/img/BLOOM-7B_3.png +0 -0
/msprobe/{pytorch/doc → docs}/img/BLOOM-7B_4.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_1.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_2.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_3.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_4.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_5.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_6.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_7.png +0 -0
/msprobe/{pytorch/doc → docs}/img/GPT-3_8.png +0 -0
/msprobe/{pytorch/doc → docs}/img/YOLOV5S_1.png +0 -0
/msprobe/{pytorch/doc → docs}/img/YOLOV5S_2.png +0 -0
/msprobe/{pytorch/doc → docs}/img/accuracy_checking_details.png +0 -0
/msprobe/{pytorch/doc → docs}/img/accuracy_checking_result.png +0 -0
/msprobe/{pytorch/doc → docs}/img/api_precision_compare_details.png +0 -0
/msprobe/{pytorch/doc → docs}/img/api_precision_compare_result.png +0 -0
/msprobe/{pytorch/doc → docs}/img/auto_analyze_log.png +0 -0
/msprobe/{pytorch/doc → docs}/img/compare_result_pkl.png +0 -0
/msprobe/{pytorch/doc → docs}/img/compare_result_pkl_md5.png.png +0 -0
/msprobe/{pytorch/doc → docs}/img/cpu_info.png +0 -0
/msprobe/{config → docs}/img/free_benchmark.png +0 -0
/msprobe/{pytorch/doc → docs}/img/module_compare.png +0 -0

msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml ADDED Viewed

@@ -0,0 +1,842 @@
+# List of apis that support self check
+communication:
+  - all_gather_into_tensor
+  - gather_into_tensor
+  - all_reduce
+  - reduce
+  - reduce_scatter_tensor
+ops:
+  - adaptive_avg_pool1d
+  - adaptive_avg_pool2d
+  - adaptive_avg_pool3d
+  - adaptive_max_pool1d
+  - adaptive_max_pool2d
+  - avg_pool1d
+  - avg_pool2d
+  - avg_pool3d
+  - batch_norm
+  - bias_add
+  - ctc_greedy_decoder
+  - conv1d
+  - conv2d
+  - conv3d
+  - deformable_conv2d
+  - dense
+  - dropout
+  - dropout1d
+  - dropout2d
+  - dropout3d
+  - flatten
+  - fold
+  - fractional_max_pool3d
+  - lp_pool1d
+  - lp_pool2d
+  - lrn
+  - max_pool2d
+  - max_pool3d
+  - max_unpool1d
+  - max_unpool2d
+  - max_unpool3d
+  - unfold
+  - binary_cross_entropy
+  - binary_cross_entropy_with_logits
+  - cosine_embedding_loss
+  - cross_entropy
+  - ctc_loss
+  - gaussian_nll_loss
+  - hinge_embedding_loss
+  - huber_loss
+  - kl_div
+  - l1_loss
+  - margin_ranking_loss
+  - mse_loss
+  - multi_margin_loss
+  - multilabel_margin_loss
+  - multilabel_soft_margin_loss
+  - nll_loss
+  - smooth_l1_loss
+  - triplet_margin_loss
+  - elu
+  - fast_gelu
+  - gelu
+  - glu
+  - gumbel_softmax
+  - hardshrink
+  - hardsigmoid
+  - hardswish
+  - hardtanh
+  - leaky_relu
+  - log_softmax
+  - logsigmoid
+  - mish
+  - prelu
+  - relu
+  - relu6
+  - rrelu
+  - selu
+  - sigmoid
+  - silu
+  - softmax
+  - softmin
+  - softshrink
+  - softsign
+  - tanh
+  - threshold
+  - cdist
+  - dist
+  - pdist
+  - choice_with_mask
+  - random_categorical
+  - log_uniform_candidate_sampler
+  - uniform_candidate_sampler
+  - affine_grid
+  - bounding_box_decode
+  - bounding_box_encode
+  - col2im
+  - check_valid
+  - crop_and_resize
+  - grid_sample
+  - interpolate
+  - iou
+  - pad
+  - padding
+  - pixel_shuffle
+  - pixel_unshuffle
+  - upsample
+  - abs
+  - absolute
+  - accumulate_n
+  - acos
+  - arccos
+  - acosh
+  - add
+  - addcdiv
+  - addcmul
+  - addmv
+  - addn
+  - angle
+  - arccosh
+  - arcsin
+  - arcsinh
+  - arctan
+  - arctanh
+  - arctan2
+  - asin
+  - asinh
+  - atan
+  - atan2
+  - atanh
+  - atleast_1d
+  - atleast_2d
+  - atleast_3d
+  - bessel_i0
+  - bessel_i0e
+  - bessel_i1
+  - bessel_i1e
+  - bessel_j0
+  - bessel_j1
+  - bessel_k0
+  - bessel_k0e
+  - bessel_k1
+  - bessel_k1e
+  - bessel_y0
+  - bessel_y1
+  - bitwise_and
+  - bitwise_left_shift
+  - bitwise_or
+  - bitwise_right_shift
+  - bitwise_xor
+  - ceil
+  - clamp
+  - clip
+  - combinations
+  - copysign
+  - cos
+  - cosh
+  - cosine_similarity
+  - cov
+  - diag_embed
+  - diff
+  - deg2rad
+  - digamma
+  - div
+  - divide
+  - erf
+  - erfc
+  - erfinv
+  - exp
+  - exp2
+  - expm1
+  - floor
+  - floor_div
+  - floor_mod
+  - float_power
+  - fmod
+  - frac
+  - gcd
+  - hypot
+  - igamma
+  - igammac
+  - imag
+  - i0
+  - inv
+  - invert
+  - lcm
+  - ldexp
+  - lerp
+  - log
+  - log2
+  - log10
+  - log1p
+  - logaddexp
+  - logaddexp2
+  - logical_and
+  - logical_not
+  - logical_or
+  - logical_xor
+  - logit
+  - mul
+  - multiply
+  - mvlgamma
+  - neg
+  - negative
+  - nextafter
+  - polar
+  - polygamma
+  - positive
+  - pow
+  - rad2deg
+  - ravel
+  - real
+  - reciprocal
+  - remainder
+  - rot90
+  - round
+  - rsqrt
+  - sgn
+  - sign
+  - signbit
+  - sin
+  - sinc
+  - sinh
+  - sqrt
+  - square
+  - sub
+  - subtract
+  - t
+  - tan
+  - tanhshrink
+  - trapz
+  - tril_indices
+  - triu_indices
+  - true_divide
+  - trunc
+  - truncate_div
+  - truncate_mod
+  - xdivy
+  - xlogy
+  - zeta
+  - all
+  - amax
+  - amin
+  - aminmax
+  - any
+  - argmax
+  - argmin
+  - cummax
+  - cummin
+  - cumprod
+  - cumsum
+  - fmax
+  - histc
+  - logsumexp
+  - max
+  - mean
+  - median
+  - min
+  - norm
+  - prod
+  - std
+  - std_mean
+  - var
+  - var_mean
+  - argsort
+  - approximate_equal
+  - equal
+  - ge
+  - greater
+  - greater_equal
+  - gt
+  - intopk
+  - isclose
+  - isfinite
+  - isinf
+  - isnan
+  - isneginf
+  - isposinf
+  - isreal
+  - le
+  - less
+  - less_equal
+  - lt
+  - maximum
+  - minimum
+  - msort
+  - ne
+  - not_equal
+  - searchsorted
+  - topk
+  - bmm
+  - addbmm
+  - addmm
+  - baddbmm
+  - addr
+  - adjoint
+  - cholesky
+  - cholesky_solve
+  - batch_dot
+  - dot
+  - eig
+  - inner
+  - inverse
+  - geqrf
+  - ger
+  - kron
+  - lu_solve
+  - lu_unpack
+  - matmul
+  - matrix_solve
+  - matrix_band_part
+  - matrix_diag
+  - matrix_diag_part
+  - matrix_set_diag
+  - mm
+  - mv
+  - outer
+  - orgqr
+  - ormqr
+  - pinv
+  - svd
+  - tensor_dot
+  - logdet
+  - slogdet
+  - qr
+  - trace
+  - bartlett_window
+  - blackman_window
+  - hamming_window
+  - hann_window
+  - kaiser_window
+  - eye
+  - fill
+  - full
+  - full_like
+  - linspace
+  - logspace
+  - one_hot
+  - arange
+  - range
+  - heaviside
+  - bernoulli
+  - gamma
+  - laplace
+  - multinomial
+  - multinomial_with_replacement
+  - rand
+  - rand_like
+  - randint
+  - randint_like
+  - randn
+  - randn_like
+  - random_gamma
+  - random_poisson
+  - randperm
+  - standard_laplace
+  - standard_normal
+  - uniform
+  - argwhere
+  - batch_to_space_nd
+  - bincount
+  - block_diag
+  - broadcast_to
+  - cat
+  - channel_shuffle
+  - chunk
+  - column_stack
+  - concat
+  - conj
+  - count_nonzero
+  - deepcopy
+  - diag
+  - diagflat
+  - diagonal
+  - dyn_shape
+  - dsplit
+  - dstack
+  - einsum
+  - expand
+  - expand_dims
+  - flip
+  - fliplr
+  - flipud
+  - gather_d
+  - gather_elements
+  - gather_nd
+  - hsplit
+  - hstack
+  - masked_fill
+  - masked_select
+  - meshgrid
+  - moveaxis
+  - movedim
+  - narrow
+  - nan_to_num
+  - nansum
+  - normal
+  - nonzero
+  - population_count
+  - rank
+  - repeat_elements
+  - repeat_interleave
+  - reshape
+  - reverse
+  - reverse_sequence
+  - roll
+  - select
+  - sequence_mask
+  - shuffle
+  - size
+  - slice
+  - sort
+  - space_to_batch_nd
+  - sparse_segment_mean
+  - split
+  - squeeze
+  - stack
+  - strided_slice
+  - sum
+  - swapaxes
+  - swapdims
+  - tensor_split
+  - tile
+  - tril
+  - triu
+  - transpose
+  - unbind
+  - unique
+  - unique_consecutive
+  - unique_with_pad
+  - unsorted_segment_max
+  - unsorted_segment_min
+  - unsorted_segment_prod
+  - unsorted_segment_sum
+  - unsqueeze
+  - unstack
+  - view_as_real
+  - vsplit
+  - vstack
+  - where
+  - cross
+  - renorm
+  - tuple_to_array
+  - clip_by_global_norm
+  - clip_by_value
+  - derivative
+  - jet
+Tensor:
+  - __abs__
+  - __add__
+  - __and__
+  - __iadd__
+  - __ifloordiv__
+  - __imatmul__
+  - __imod__
+  - __imul__
+  - __isub__
+  - __matmul__
+  - __mod__
+  - __mul__
+  - __neg__
+  - __or__
+  - __pow__
+  - __radd__
+  - __rmatmul__
+  - __rmod__
+  - __rmul__
+  - __rpow__
+  - __rsub__
+  - __sub__
+  - __truediv__
+  - __xor__
+  - abs
+  - absolute
+  - acos
+  - acosh
+  - add
+  - addbmm
+  - addcdiv
+  - addcmul
+  - addmm
+  - addmv
+  - addr
+  - all
+  - amax
+  - amin
+  - any
+  - arccos
+  - arccosh
+  - argmax
+  - angle
+  - arcsin
+  - arcsinh
+  - arctan
+  - arctanh
+  - argmin
+  - argsort
+  - asin
+  - asinh
+  - atan
+  - atan2
+  - atanh
+  - baddbmm
+  - bernoulli
+  - bincount
+  - bitwise_and
+  - bitwise_or
+  - bitwise_xor
+  - bmm
+  - broadcast_to
+  - ceil
+  - cholesky_solve
+  - cholesky
+  - clamp
+  - clip
+  - conj
+  - copysign
+  - cos
+  - cosh
+  - cross
+  - cummax
+  - cummin
+  - cumprod
+  - cumsum
+  - deg2rad
+  - diag
+  - diagflat
+  - diff
+  - digamma
+  - div
+  - divide
+  - equal
+  - erf
+  - erfc
+  - erfinv
+  - exp
+  - expand_as
+  - expm1
+  - flip
+  - fliplr
+  - flipud
+  - float_power
+  - floor
+  - fmod
+  - frac
+  - gather_elements
+  - geqrf
+  - ger
+  - greater
+  - greater_equal
+  - half
+  - hardshrink
+  - heaviside
+  - histc
+  - hypot
+  - i0
+  - igamma
+  - igammac
+  - imag
+  - index_add
+  - index_fill
+  - index_put
+  - index_select
+  - inner
+  - int
+  - inverse
+  - item
+  - lcm
+  - ldexp
+  - lerp
+  - log
+  - log10
+  - log1p
+  - log2
+  - logaddexp
+  - logaddexp2
+  - logdet
+  - logical_and
+  - logical_not
+  - logical_or
+  - logical_xor
+  - logit
+  - logsumexp
+  - long
+  - masked_fill
+  - masked_scatter
+  - masked_select
+  - matmul
+  - max
+  - maximum
+  - mean
+  - median
+  - min
+  - minimum
+  - moveaxis
+  - movedim
+  - msort
+  - multinomial
+  - multiply
+  - mvlgamma
+  - nan_to_num
+  - nansum
+  - narrow
+  - neg
+  - negative
+  - nelement
+  - new_ones
+  - new_zeros
+  - nextafter
+  - norm
+  - nonzero
+  - not_equal
+  - ormqr
+  - permute
+  - pow
+  - prod
+  - qr
+  - ravel
+  - real
+  - reciprocal
+  - remainder
+  - renorm
+  - rad2deg
+  - tile
+  - repeat_interleave
+  - reshape
+  - reshape
+  - round
+  - rot90
+  - rsqrt
+  - sum_to_size
+  - scatter
+  - sgn
+  - short
+  - sigmoid
+  - sign
+  - signbit
+  - sin
+  - sinc
+  - sinh
+  - slogdet
+  - sort
+  - split
+  - sqrt
+  - square
+  - squeeze
+  - std
+  - subtract
+  - subtract
+  - svd
+  - swapaxes
+  - swapdims
+  - t
+  - take
+  - tan
+  - tanh
+  - trace
+  - swapaxes
+  - tile
+  - topk
+  - tril
+  - tensor_split
+  - transpose
+  - true_divide
+  - trunc
+  - unbind
+  - unique_consecutive
+  - unsqueeze
+  - var
+  - view
+  - where
+  - xlogy
+  - from_numpy
+  - std
+  - take
+  - var
+  - all
+  - any
+  - copy
+  - diagonal
+  - flatten
+  - resize
+  - sum
+mint:
+  - abs
+  - absolute_import
+  - add
+  - add_ex
+  - all
+  - any
+  - any_ex
+  - arange
+  - argmax
+  - avg_pool2d
+  - baddbmm
+  - baddbmm_ex
+  - batch_norm
+  - binary_cross_entropy_with_logits
+  - bitwise_and
+  - bitwise_or
+  - bitwise_xor
+  - bmm
+  - broadcast_to
+  - cat
+  - cat_ex
+  - ceil
+  - chunk
+  - clamp
+  - conv2d
+  - conv_transpose2d
+  - cos
+  - cross
+  - cummax
+  - cummin
+  - cumsum
+  - div
+  - divide
+  - dropout
+  - embedding
+  - eq
+  - erf
+  - erfinv
+  - exp
+  - flatten
+  - flip
+  - flip_ex
+  - fold
+  - full
+  - gather
+  - gelu
+  - greater
+  - grid_sample
+  - group_norm
+  - gt
+  - index_select
+  - interpolate
+  - isclose
+  - isfinite
+  - layer_norm
+  - le
+  - leaky_relu
+  - less
+  - less_equal
+  - linear
+  - linspace
+  - log
+  - logical_and
+  - logical_not
+  - logical_or
+  - lt
+  - masked_select
+  - matmul
+  - max
+  - max_pool2d
+  - maximum
+  - mean
+  - mean_ex
+  - min
+  - minimum
+  - mul
+  - ne
+  - neg
+  - negative
+  - nonzero
+  - normal
+  - one_hot
+  - ones
+  - ones_ex
+  - ones_like
+  - pad
+  - permute
+  - permute_ex
+  - pow
+  - prod
+  - reciprocal
+  - relu
+  - remainder
+  - repeat_interleave
+  - rsqrt
+  - searchsorted
+  - sigmoid
+  - silu
+  - sin
+  - softmax
+  - softplus
+  - sort
+  - split
+  - sqrt
+  - sqrt_ex
+  - square
+  - stack
+  - sub
+  - sub_ex
+  - sum
+  - tanh
+  - tile
+  - topk
+  - tril
+  - triu
+  - unfold
+  - unique
+  - where
+  - xlogy
+  - zeros
+  - zeros_ex
+  - zeros_like
+mint.nn.functional:
+  - absolute_import
+  - avg_pool2d
+  - batch_norm
+  - batch_norm_ex
+  - bce_with_logits
+  - binary_cross_entropy_with_logits
+  - conv_transpose2d
+  - dense
+  - dropout
+  - embedding
+  - fold
+  - gelu
+  - grid_sample
+  - group_norm
+  - interpolate
+  - layer_norm
+  - leaky_relu
+  - linear
+  - max_pool2d
+  - max_pool2d_ex
+  - normal
+  - one_hot
+  - one_hot_ext
+  - pad
+  - relu
+  - sigmoid
+  - silu
+  - softmax
+  - softmax_ex
+  - softplus
+  - tanh
+  - unfold

mindstudio-probe 1.0.1__py3-none-any.whl → 1.0.4__py3-none-any.whl

mindstudio-probe 1.0.1py3-none-any.whl → 1.0.4py3-none-any.whl