├── .clang-format ├── .flake8 ├── .github ├── actions │ └── checkout-code │ │ └── action.yml └── workflows │ ├── lint.yml │ ├── runs_on_ascend.yml │ ├── runs_on_camb.yml │ └── runs_on_nv.yml ├── .gitignore ├── CODE_OF_CONDUCT.md ├── CODE_OF_CONDUCT_cn.md ├── Contributors.md ├── LICENSE ├── README.md ├── ci ├── lint.sh ├── run_individual_test_cases.sh └── run_pytorch_test_cases.sh ├── csrc └── README.md ├── ditorch.png ├── ditorch ├── __init__.py ├── common_adapter │ ├── __init__.py │ └── common_mock.py ├── test │ ├── ascend_summary_of_pytorch_test_case_testing.csv.tar │ ├── camb_summary_of_pytorch_test_case_testing.csv.tar │ ├── command_runner.py │ ├── copy_and_process_pytorch_test_scripts.py │ ├── discover_pytorch_test_case.py │ ├── generate_test_shell_script_from_testcase_json.py │ ├── individual │ │ ├── test_dist.py │ │ └── test_utils.py │ └── summary_test_results.py ├── torch_biren_adapter │ └── __init__.py ├── torch_dipu_adapter │ └── __init__.py ├── torch_mlu_adapter │ ├── __init__.py │ └── mock_runtime.py ├── torch_npu_adapter │ ├── __init__.py │ ├── mock_dist.py │ └── mock_runtime.py └── utils.py ├── ditorch_config.sh ├── module_tools ├── __init__.py ├── cmp_accuracy.py ├── compare.py └── dir_lock.lock └── op_tools ├── README.md ├── __init__.py ├── apply_hook.py ├── base_hook.py ├── custom_apply_hook.py ├── op_autocompare_hook.py ├── op_capture.png ├── op_capture_hook.py ├── op_dtype_cast_hook.py ├── op_fallback_hook.py ├── op_observe_hook.py ├── op_overflow_check_hook.py ├── op_runner.py ├── op_time_measure_hook.py ├── pretty_print.py ├── process_monitor.py ├── requirements.txt ├── run_op_from_data.py ├── save_op_args.py ├── test ├── test_compare_result.py ├── test_current_location.py ├── test_current_stream.py ├── test_custom_apply_hook.py ├── test_custom_condition_to_enable_hook_example.py ├── test_embedding.py ├── test_event.py ├── test_get_error_tolerance.py ├── test_get_function_from_string.py ├── test_is_cpu_op.py ├── test_is_inf_or_nan.py ├── test_op_autocompare.py ├── test_op_capture.py ├── test_op_dtype_cast.py ├── test_op_fallback.py ├── test_op_observe.py ├── test_op_time_measure.py ├── test_op_tools.py ├── test_opname_match.py ├── test_pretty_print.py ├── test_profile.py ├── test_register_torch_hook.py ├── test_run_op_from_data.py └── test_tool_with_special_op.py └── utils.py /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.clang-format -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/actions/checkout-code/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.github/actions/checkout-code/action.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/runs_on_ascend.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.github/workflows/runs_on_ascend.yml -------------------------------------------------------------------------------- /.github/workflows/runs_on_camb.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.github/workflows/runs_on_camb.yml -------------------------------------------------------------------------------- /.github/workflows/runs_on_nv.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.github/workflows/runs_on_nv.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/CODE_OF_CONDUCT_cn.md -------------------------------------------------------------------------------- /Contributors.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/Contributors.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/README.md -------------------------------------------------------------------------------- /ci/lint.sh: -------------------------------------------------------------------------------- 1 | flake8 ditorch op_tools -v -------------------------------------------------------------------------------- /ci/run_individual_test_cases.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ci/run_individual_test_cases.sh -------------------------------------------------------------------------------- /ci/run_pytorch_test_cases.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ci/run_pytorch_test_cases.sh -------------------------------------------------------------------------------- /csrc/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ditorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch.png -------------------------------------------------------------------------------- /ditorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/__init__.py -------------------------------------------------------------------------------- /ditorch/common_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/common_adapter/__init__.py -------------------------------------------------------------------------------- /ditorch/common_adapter/common_mock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/common_adapter/common_mock.py -------------------------------------------------------------------------------- /ditorch/test/ascend_summary_of_pytorch_test_case_testing.csv.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/ascend_summary_of_pytorch_test_case_testing.csv.tar -------------------------------------------------------------------------------- /ditorch/test/camb_summary_of_pytorch_test_case_testing.csv.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/camb_summary_of_pytorch_test_case_testing.csv.tar -------------------------------------------------------------------------------- /ditorch/test/command_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/command_runner.py -------------------------------------------------------------------------------- /ditorch/test/copy_and_process_pytorch_test_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/copy_and_process_pytorch_test_scripts.py -------------------------------------------------------------------------------- /ditorch/test/discover_pytorch_test_case.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/discover_pytorch_test_case.py -------------------------------------------------------------------------------- /ditorch/test/generate_test_shell_script_from_testcase_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/generate_test_shell_script_from_testcase_json.py -------------------------------------------------------------------------------- /ditorch/test/individual/test_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/individual/test_dist.py -------------------------------------------------------------------------------- /ditorch/test/individual/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/individual/test_utils.py -------------------------------------------------------------------------------- /ditorch/test/summary_test_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/test/summary_test_results.py -------------------------------------------------------------------------------- /ditorch/torch_biren_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_biren_adapter/__init__.py -------------------------------------------------------------------------------- /ditorch/torch_dipu_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_dipu_adapter/__init__.py -------------------------------------------------------------------------------- /ditorch/torch_mlu_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_mlu_adapter/__init__.py -------------------------------------------------------------------------------- /ditorch/torch_mlu_adapter/mock_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_mlu_adapter/mock_runtime.py -------------------------------------------------------------------------------- /ditorch/torch_npu_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_npu_adapter/__init__.py -------------------------------------------------------------------------------- /ditorch/torch_npu_adapter/mock_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_npu_adapter/mock_dist.py -------------------------------------------------------------------------------- /ditorch/torch_npu_adapter/mock_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/torch_npu_adapter/mock_runtime.py -------------------------------------------------------------------------------- /ditorch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch/utils.py -------------------------------------------------------------------------------- /ditorch_config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/ditorch_config.sh -------------------------------------------------------------------------------- /module_tools/__init__.py: -------------------------------------------------------------------------------- 1 | from .cmp_accuracy import CompLayerAcc 2 | -------------------------------------------------------------------------------- /module_tools/cmp_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/module_tools/cmp_accuracy.py -------------------------------------------------------------------------------- /module_tools/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/module_tools/compare.py -------------------------------------------------------------------------------- /module_tools/dir_lock.lock: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /op_tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/README.md -------------------------------------------------------------------------------- /op_tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/__init__.py -------------------------------------------------------------------------------- /op_tools/apply_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/apply_hook.py -------------------------------------------------------------------------------- /op_tools/base_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/base_hook.py -------------------------------------------------------------------------------- /op_tools/custom_apply_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/custom_apply_hook.py -------------------------------------------------------------------------------- /op_tools/op_autocompare_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_autocompare_hook.py -------------------------------------------------------------------------------- /op_tools/op_capture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_capture.png -------------------------------------------------------------------------------- /op_tools/op_capture_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_capture_hook.py -------------------------------------------------------------------------------- /op_tools/op_dtype_cast_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_dtype_cast_hook.py -------------------------------------------------------------------------------- /op_tools/op_fallback_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_fallback_hook.py -------------------------------------------------------------------------------- /op_tools/op_observe_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_observe_hook.py -------------------------------------------------------------------------------- /op_tools/op_overflow_check_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_overflow_check_hook.py -------------------------------------------------------------------------------- /op_tools/op_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_runner.py -------------------------------------------------------------------------------- /op_tools/op_time_measure_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/op_time_measure_hook.py -------------------------------------------------------------------------------- /op_tools/pretty_print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/pretty_print.py -------------------------------------------------------------------------------- /op_tools/process_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/process_monitor.py -------------------------------------------------------------------------------- /op_tools/requirements.txt: -------------------------------------------------------------------------------- 1 | prettytable 2 | psutil -------------------------------------------------------------------------------- /op_tools/run_op_from_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/run_op_from_data.py -------------------------------------------------------------------------------- /op_tools/save_op_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/save_op_args.py -------------------------------------------------------------------------------- /op_tools/test/test_compare_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_compare_result.py -------------------------------------------------------------------------------- /op_tools/test/test_current_location.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_current_location.py -------------------------------------------------------------------------------- /op_tools/test/test_current_stream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_current_stream.py -------------------------------------------------------------------------------- /op_tools/test/test_custom_apply_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_custom_apply_hook.py -------------------------------------------------------------------------------- /op_tools/test/test_custom_condition_to_enable_hook_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_custom_condition_to_enable_hook_example.py -------------------------------------------------------------------------------- /op_tools/test/test_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_embedding.py -------------------------------------------------------------------------------- /op_tools/test/test_event.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_event.py -------------------------------------------------------------------------------- /op_tools/test/test_get_error_tolerance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_get_error_tolerance.py -------------------------------------------------------------------------------- /op_tools/test/test_get_function_from_string.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_get_function_from_string.py -------------------------------------------------------------------------------- /op_tools/test/test_is_cpu_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_is_cpu_op.py -------------------------------------------------------------------------------- /op_tools/test/test_is_inf_or_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_is_inf_or_nan.py -------------------------------------------------------------------------------- /op_tools/test/test_op_autocompare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_autocompare.py -------------------------------------------------------------------------------- /op_tools/test/test_op_capture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_capture.py -------------------------------------------------------------------------------- /op_tools/test/test_op_dtype_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_dtype_cast.py -------------------------------------------------------------------------------- /op_tools/test/test_op_fallback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_fallback.py -------------------------------------------------------------------------------- /op_tools/test/test_op_observe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_observe.py -------------------------------------------------------------------------------- /op_tools/test/test_op_time_measure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_time_measure.py -------------------------------------------------------------------------------- /op_tools/test/test_op_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_op_tools.py -------------------------------------------------------------------------------- /op_tools/test/test_opname_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_opname_match.py -------------------------------------------------------------------------------- /op_tools/test/test_pretty_print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_pretty_print.py -------------------------------------------------------------------------------- /op_tools/test/test_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_profile.py -------------------------------------------------------------------------------- /op_tools/test/test_register_torch_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_register_torch_hook.py -------------------------------------------------------------------------------- /op_tools/test/test_run_op_from_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_run_op_from_data.py -------------------------------------------------------------------------------- /op_tools/test/test_tool_with_special_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/test/test_tool_with_special_op.py -------------------------------------------------------------------------------- /op_tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DeepLink-org/ditorch/HEAD/op_tools/utils.py --------------------------------------------------------------------------------