mindstudio-probe 1.0.1__py3-none-any.whl → 1.0.4__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {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
|
@@ -1,259 +1,260 @@
|
|
|
1
|
-
#!/usr/bin/env python3
|
|
2
|
-
# -*- coding: utf-8 -*-
|
|
3
|
-
"""
|
|
4
|
-
# Copyright (C) 2022-2024. Huawei Technologies Co., Ltd. All rights reserved.
|
|
5
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
6
|
-
# you may not use this file except in compliance with the License.
|
|
7
|
-
# You may obtain a copy of the License at
|
|
8
|
-
#
|
|
9
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
-
#
|
|
11
|
-
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
-
# See the License for the specific language governing permissions and
|
|
15
|
-
# limitations under the License.
|
|
16
|
-
"""
|
|
17
|
-
|
|
18
|
-
import os
|
|
19
|
-
import time
|
|
20
|
-
import numpy as np
|
|
21
|
-
from collections import namedtuple
|
|
22
|
-
from msprobe.pytorch.parse_tool.lib.utils import Util
|
|
23
|
-
from msprobe.pytorch.parse_tool.lib.config import Const
|
|
24
|
-
from msprobe.pytorch.parse_tool.lib.parse_exception import ParseException
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
self.
|
|
31
|
-
self.
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
self.
|
|
36
|
-
self.
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
self.
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
content.append('
|
|
107
|
-
content.append('
|
|
108
|
-
content.append('
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
self.
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
"
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
1
|
+
#!/usr/bin/env python3
|
|
2
|
+
# -*- coding: utf-8 -*-
|
|
3
|
+
"""
|
|
4
|
+
# Copyright (C) 2022-2024. Huawei Technologies Co., Ltd. All rights reserved.
|
|
5
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
6
|
+
# you may not use this file except in compliance with the License.
|
|
7
|
+
# You may obtain a copy of the License at
|
|
8
|
+
#
|
|
9
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
10
|
+
#
|
|
11
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
12
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
13
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
14
|
+
# See the License for the specific language governing permissions and
|
|
15
|
+
# limitations under the License.
|
|
16
|
+
"""
|
|
17
|
+
|
|
18
|
+
import os
|
|
19
|
+
import time
|
|
20
|
+
import numpy as np
|
|
21
|
+
from collections import namedtuple
|
|
22
|
+
from msprobe.pytorch.parse_tool.lib.utils import Util
|
|
23
|
+
from msprobe.pytorch.parse_tool.lib.config import Const
|
|
24
|
+
from msprobe.pytorch.parse_tool.lib.parse_exception import ParseException
|
|
25
|
+
from msprobe.core.common.file_utils import FileChecker, create_directory, load_npy, save_npy_to_txt, write_csv
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
class Compare:
|
|
29
|
+
def __init__(self):
|
|
30
|
+
self.util = Util()
|
|
31
|
+
self.log = self.util.log
|
|
32
|
+
self.vector_compare_result = {}
|
|
33
|
+
|
|
34
|
+
def npu_vs_npu_compare(self, my_dump_path, golden_dump_path, result_dir, msaccucmp_path):
|
|
35
|
+
self.log.info("Start Compare ...............")
|
|
36
|
+
self.compare_vector(my_dump_path, golden_dump_path, result_dir, msaccucmp_path)
|
|
37
|
+
self.log.info("Compare finished!!")
|
|
38
|
+
|
|
39
|
+
def compare_vector(self, my_dump_path, golden_dump_path, result_dir, msaccucmp_path):
|
|
40
|
+
create_directory(result_dir)
|
|
41
|
+
self.util.check_path_valid(result_dir)
|
|
42
|
+
call_msaccucmp = self.util.check_msaccucmp(msaccucmp_path)
|
|
43
|
+
cmd = '%s %s compare -m %s -g %s -out %s' % (
|
|
44
|
+
self.util.python, call_msaccucmp, my_dump_path, golden_dump_path, result_dir
|
|
45
|
+
)
|
|
46
|
+
return self.util.execute_command(cmd)
|
|
47
|
+
|
|
48
|
+
def convert_dump_to_npy(self, dump_file, data_format, output, msaccucmp_path):
|
|
49
|
+
dump_file = self.util.path_strip(dump_file)
|
|
50
|
+
file_name = ""
|
|
51
|
+
if os.path.isfile(dump_file):
|
|
52
|
+
self.log.info("Covert file is: %s", dump_file)
|
|
53
|
+
file_name = os.path.basename(dump_file)
|
|
54
|
+
elif os.path.isdir(dump_file):
|
|
55
|
+
self.log.info("Convert all files in path: %s", dump_file)
|
|
56
|
+
file_name = ""
|
|
57
|
+
output = output if output else Const.DUMP_CONVERT_DIR
|
|
58
|
+
convert = self.convert(dump_file, data_format, output, msaccucmp_path)
|
|
59
|
+
if convert == 0:
|
|
60
|
+
convert_files = self.util.list_convert_files(output, file_name)
|
|
61
|
+
|
|
62
|
+
summary_txt = ["SrcFile: %s" % dump_file]
|
|
63
|
+
for convert_file in convert_files.values():
|
|
64
|
+
summary_txt.append(" - %s" % convert_file.file_name)
|
|
65
|
+
self.log.info("Transfer result is saved in : %s", os.path.realpath(output))
|
|
66
|
+
self.util.print_panel("\n".join(summary_txt))
|
|
67
|
+
|
|
68
|
+
def convert(self, dump_file, data_format, output, msaccucmp_path):
|
|
69
|
+
create_directory(output)
|
|
70
|
+
self.util.check_path_valid(output)
|
|
71
|
+
call_msaccucmp = self.util.check_msaccucmp(msaccucmp_path)
|
|
72
|
+
if data_format:
|
|
73
|
+
cmd = '%s %s convert -d %s -out %s -f %s' % (
|
|
74
|
+
self.util.python, call_msaccucmp, dump_file, output, data_format
|
|
75
|
+
)
|
|
76
|
+
else:
|
|
77
|
+
cmd = '%s %s convert -d %s -out %s' % (
|
|
78
|
+
self.util.python, call_msaccucmp, dump_file, output
|
|
79
|
+
)
|
|
80
|
+
return self.util.execute_command(cmd)
|
|
81
|
+
|
|
82
|
+
def compare_data(self, args):
|
|
83
|
+
"""Compare data"""
|
|
84
|
+
(left, right, save_txt, rl, al, diff_count) = args
|
|
85
|
+
if left is None or right is None:
|
|
86
|
+
raise ParseException("invalid input or output")
|
|
87
|
+
if self.util.check_path_valid(left) and self.util.check_path_valid(right):
|
|
88
|
+
left_data = load_npy(left)
|
|
89
|
+
right_data = load_npy(right)
|
|
90
|
+
# save to txt
|
|
91
|
+
if save_txt:
|
|
92
|
+
save_npy_to_txt(left_data, left + ".txt")
|
|
93
|
+
save_npy_to_txt(right_data, right + ".txt")
|
|
94
|
+
# compare data
|
|
95
|
+
(total_cnt, all_close, cos_sim, err_percent) = self.do_compare_data(left_data, right_data, rl, al, diff_count)
|
|
96
|
+
content = ['Left:', ' ├─ NpyFile: %s' % left]
|
|
97
|
+
if save_txt:
|
|
98
|
+
content.append(' ├─ TxtFile: [green]%s.txt[/green]' % left)
|
|
99
|
+
content.append(' └─ NpySpec: [yellow]%s[/yellow]' % self.util.gen_npy_info_txt(left_data))
|
|
100
|
+
content.append('Right:')
|
|
101
|
+
content.append(' ├─ NpyFile: %s' % right)
|
|
102
|
+
if save_txt:
|
|
103
|
+
content.append(' ├─ TxtFile: [green]%s.txt[/green]' % right)
|
|
104
|
+
content.append(' └─ NpySpec: [yellow]%s[/yellow]' % self.util.gen_npy_info_txt(right_data))
|
|
105
|
+
content.append('NumCnt: %s' % total_cnt)
|
|
106
|
+
content.append('AllClose: %s' % all_close)
|
|
107
|
+
content.append('CosSim: %s' % cos_sim)
|
|
108
|
+
content.append('ErrorPer: %s (rl= %s, al= %s)' % (err_percent, rl, al))
|
|
109
|
+
self.util.print_panel("\n".join(content))
|
|
110
|
+
|
|
111
|
+
def do_compare_data(self, left, right, rl=0.001, al=0.001, diff_count=20):
|
|
112
|
+
data_left = left.astype(np.float32)
|
|
113
|
+
data_right = right.astype(np.float32)
|
|
114
|
+
shape_left = data_left.shape
|
|
115
|
+
shape_right = data_right.shape
|
|
116
|
+
if shape_left != shape_right:
|
|
117
|
+
self.log.warning("Data shape not equal: %s vs %s", data_left.shape, data_right.shape)
|
|
118
|
+
data_left = data_left.reshape(-1)
|
|
119
|
+
data_right = data_right.reshape(-1)
|
|
120
|
+
if data_left.shape[0] != data_right.shape[0]:
|
|
121
|
+
self.log.warning("Data size not equal: %s vs %s", data_left.shape, data_right.shape)
|
|
122
|
+
if data_left.shape[0] < data_right.shape[0]:
|
|
123
|
+
data_left = np.pad(data_left, (0, data_right.shape[0] - data_left.shape[0]), 'constant')
|
|
124
|
+
else:
|
|
125
|
+
data_right = np.pad(data_right, (0, data_left.shape[0] - data_right.shape[0]), 'constant')
|
|
126
|
+
all_close = np.allclose(data_left, data_right, atol=al, rtol=rl)
|
|
127
|
+
np.seterr(divide='raise')
|
|
128
|
+
cos_sim = np.dot(data_left, data_right) / (
|
|
129
|
+
np.sqrt(np.dot(data_left, data_left)) * np.sqrt(np.dot(data_right, data_right)))
|
|
130
|
+
err_cnt = 0
|
|
131
|
+
total_cnt = data_left.shape[0]
|
|
132
|
+
diff_table_columns = ['Index', 'Left', 'Right', 'Diff']
|
|
133
|
+
err_table = self.util.create_table("Error Item Table", diff_table_columns)
|
|
134
|
+
top_table = self.util.create_table("Top Item Table", diff_table_columns)
|
|
135
|
+
for i in range(total_cnt):
|
|
136
|
+
abs_diff = abs(data_left[i] - data_right[i])
|
|
137
|
+
if i < diff_count:
|
|
138
|
+
top_table.add_row(str(i), str(data_left[i]), str(data_right[i]), str(abs_diff))
|
|
139
|
+
if abs_diff > (al + rl * abs(data_right[i])):
|
|
140
|
+
if err_cnt < diff_count:
|
|
141
|
+
err_table.add_row(str(i), str(data_left[i]), str(data_right[i]), str(abs_diff))
|
|
142
|
+
err_cnt += 1
|
|
143
|
+
if total_cnt == 0:
|
|
144
|
+
err_percent = float(0)
|
|
145
|
+
else:
|
|
146
|
+
err_percent = float(err_cnt / total_cnt)
|
|
147
|
+
self.util.print(self.util.create_columns([err_table, top_table]))
|
|
148
|
+
do_compare_data_result = namedtuple('do_compare_data_result', ['cnt', 'close', 'cos', 'err'])
|
|
149
|
+
res = do_compare_data_result(total_cnt, all_close, cos_sim, err_percent)
|
|
150
|
+
return res
|
|
151
|
+
|
|
152
|
+
def compare_npy(self, file, bench_file, output_path):
|
|
153
|
+
if self.util.check_path_valid(file) and self.util.check_path_valid(bench_file):
|
|
154
|
+
data = load_npy(file)
|
|
155
|
+
bench_data = load_npy(bench_file)
|
|
156
|
+
shape, dtype = data.shape, data.dtype
|
|
157
|
+
bench_shape, bench_dtype = bench_data.shape, bench_data.dtype
|
|
158
|
+
filename = os.path.basename(file)
|
|
159
|
+
bench_filename = os.path.basename(bench_file)
|
|
160
|
+
if shape != bench_shape or dtype != bench_dtype:
|
|
161
|
+
self.log.error(
|
|
162
|
+
"Shape or dtype between two npy files is inconsistent. Please check the two files."
|
|
163
|
+
"File 1: %s, file 2: %s", file, bench_file)
|
|
164
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
165
|
+
return
|
|
166
|
+
md5_consistency = False
|
|
167
|
+
if self.util.get_md5_for_numpy(data) == self.util.get_md5_for_numpy(bench_data):
|
|
168
|
+
md5_consistency = True
|
|
169
|
+
data_mean = np.mean(data)
|
|
170
|
+
bench_data_mean = np.mean(bench_data)
|
|
171
|
+
abs_error = np.abs(data - bench_data)
|
|
172
|
+
bench_data = self.util.deal_with_value_if_has_zero(bench_data)
|
|
173
|
+
rel_error = np.abs(abs_error / bench_data)
|
|
174
|
+
abs_diff_max = abs_error.max()
|
|
175
|
+
rel_diff_max = np.max(rel_error)
|
|
176
|
+
compare_result = [[filename, bench_filename, data_mean, bench_data_mean, md5_consistency, abs_diff_max,
|
|
177
|
+
rel_diff_max]]
|
|
178
|
+
write_csv(compare_result, output_path)
|
|
179
|
+
|
|
180
|
+
def compare_all_file_in_directory(self, my_dump_dir, golden_dump_dir, output_path):
|
|
181
|
+
if not (self.util.is_subdir_count_equal(my_dump_dir, golden_dump_dir)
|
|
182
|
+
and self.util.check_npy_files_valid_in_dir(my_dump_dir)
|
|
183
|
+
and self.util.check_npy_files_valid_in_dir(golden_dump_dir)):
|
|
184
|
+
self.log.error(
|
|
185
|
+
"Top level(Npy files level) directory structure is inconsistent. Please check the two directory.")
|
|
186
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
187
|
+
return
|
|
188
|
+
my_npy_files = self.util.get_sorted_files_names(my_dump_dir)
|
|
189
|
+
golden_npy_files = self.util.get_sorted_files_names(golden_dump_dir)
|
|
190
|
+
for my_npy_file_name, golden_npy_file_name in zip(my_npy_files, golden_npy_files):
|
|
191
|
+
my_npy_path = os.path.join(my_dump_dir, my_npy_file_name)
|
|
192
|
+
golden_npy_path = os.path.join(golden_dump_dir, golden_npy_file_name)
|
|
193
|
+
self.compare_npy(my_npy_path, golden_npy_path, output_path)
|
|
194
|
+
|
|
195
|
+
def compare_timestamp_directory(self, my_dump_dir, golden_dump_dir, output_path):
|
|
196
|
+
if not self.util.is_subdir_count_equal(my_dump_dir, golden_dump_dir):
|
|
197
|
+
self.log.error(
|
|
198
|
+
"Second level(Timestamp level) directory structure is inconsistent. Please check the two directory.")
|
|
199
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
200
|
+
return
|
|
201
|
+
my_ordered_subdirs = self.util.get_sorted_subdirectories_names(my_dump_dir)
|
|
202
|
+
golden_ordered_subdirs = self.util.get_sorted_subdirectories_names(golden_dump_dir)
|
|
203
|
+
for my_subdir_name, golden_subdir_name in zip(my_ordered_subdirs, golden_ordered_subdirs):
|
|
204
|
+
my_subdir_path = os.path.join(my_dump_dir, my_subdir_name)
|
|
205
|
+
golden_subdir_path = os.path.join(golden_dump_dir, golden_subdir_name)
|
|
206
|
+
self.compare_all_file_in_directory(my_subdir_path, golden_subdir_path, output_path)
|
|
207
|
+
|
|
208
|
+
def compare_converted_dir(self, my_dump_dir, golden_dump_dir, output_dir):
|
|
209
|
+
if not self.util.is_subdir_count_equal(my_dump_dir, golden_dump_dir):
|
|
210
|
+
self.log.error(
|
|
211
|
+
"Top level(Opname level) directory structure is inconsistent. Please check the two directory.")
|
|
212
|
+
return
|
|
213
|
+
timestamp = int(time.time())
|
|
214
|
+
output_file_name = f"batch_compare_{timestamp}.csv"
|
|
215
|
+
output_path = os.path.join(output_dir, output_file_name)
|
|
216
|
+
title_rows = [[
|
|
217
|
+
"NPU File Name",
|
|
218
|
+
"Bench File Name",
|
|
219
|
+
"Mean",
|
|
220
|
+
"Bench Mean",
|
|
221
|
+
"Md5 Consistency",
|
|
222
|
+
"Max Abs Error",
|
|
223
|
+
"Max Relative Error"
|
|
224
|
+
]]
|
|
225
|
+
write_csv(title_rows, output_path)
|
|
226
|
+
|
|
227
|
+
my_ordered_subdirs = self.util.get_sorted_subdirectories_names(my_dump_dir)
|
|
228
|
+
golden_ordered_subdirs = self.util.get_sorted_subdirectories_names(golden_dump_dir)
|
|
229
|
+
for my_subdir_name, golden_subdir_name in zip(my_ordered_subdirs, golden_ordered_subdirs):
|
|
230
|
+
if not my_subdir_name == golden_subdir_name:
|
|
231
|
+
self.log.error(
|
|
232
|
+
"Top level(Opname level) directory structure is inconsistent. Please check the two directory.")
|
|
233
|
+
self.util.deal_with_dir_or_file_inconsistency(output_path)
|
|
234
|
+
return
|
|
235
|
+
my_subdir_path = os.path.join(my_dump_dir, my_subdir_name)
|
|
236
|
+
golden_subdir_path = os.path.join(golden_dump_dir, golden_subdir_name)
|
|
237
|
+
self.compare_timestamp_directory(my_subdir_path, golden_subdir_path, output_path)
|
|
238
|
+
self.util.change_filemode_safe(output_path)
|
|
239
|
+
self.log.info("Compare result is saved in : %s", output_path)
|
|
240
|
+
|
|
241
|
+
def convert_api_dir_to_npy(self, dump_dir, param, output_dir, msaccucmp_path):
|
|
242
|
+
dump_dir = self.util.path_strip(dump_dir)
|
|
243
|
+
for root, _, files in os.walk(dump_dir, topdown=True):
|
|
244
|
+
path_checker = FileChecker(root)
|
|
245
|
+
path_checker.common_check()
|
|
246
|
+
for file in files:
|
|
247
|
+
file_path = os.path.join(root, file)
|
|
248
|
+
file_name = os.path.basename(file_path)
|
|
249
|
+
parts = file_name.split(".")
|
|
250
|
+
if len(parts) < 5:
|
|
251
|
+
continue
|
|
252
|
+
op_name = parts[1]
|
|
253
|
+
timestamp = parts[-1]
|
|
254
|
+
output_path = os.path.join(output_dir, op_name, timestamp)
|
|
255
|
+
self.convert_dump_to_npy(file_path, param, output_path, msaccucmp_path)
|
|
256
|
+
path_depth = root.count(os.sep)
|
|
257
|
+
if path_depth <= Const.MAX_TRAVERSAL_DEPTH:
|
|
258
|
+
yield root, _, files
|
|
259
|
+
else:
|
|
260
|
+
_[:] = []
|