onnxruntime/orttraining/orttraining/python/training/ortmodule
mindest 009209e016
Fix Orttraining Linux Lazy Tensor CI Pipeline (#21652)
### Description
Fix `Orttraining Linux Lazy Tensor CI Pipeline`
- Remove unused import of `torch.onnx._internal.exporter`, whose path is
changed in newer torch (pytorch/pytorch#132429).
- Move import of `register_custom_op_symbolic` from `torch.onnx` into
local function, which causes circular import when running `import
torch.onnx` (at least in the CI environment).
2024-08-21 18:10:08 +08:00
..
experimental Add LayerSpec Support to ORTPipelineModule (#20410) 2024-04-23 17:57:08 -07:00
graph_optimizers Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
torch_cpp_extensions Update ruff and clang-format versions (#21479) 2024-07-24 11:50:11 -07:00
__init__.py Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
_custom_autograd_function.py Improve perf for stage3 training (#18099) 2023-12-15 13:32:19 +08:00
_custom_autograd_function_exporter.py Flash attention recompute (#20603) 2024-05-21 13:38:19 +08:00
_custom_gradient_registry.py Adds ATen fallback for scaled_dot_product_attention (#21107) 2024-07-22 16:37:04 -07:00
_custom_op_symbolic_registry.py Fix Orttraining Linux Lazy Tensor CI Pipeline (#21652) 2024-08-21 18:10:08 +08:00
_execution_agent.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
_fallback.py Bump linter versions (#18341) 2023-11-08 13:04:40 -08:00
_fallback_exceptions.py Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00
_gradient_accumulation_manager.py
_graph_execution_interface.py
_graph_execution_manager.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
_graph_execution_manager_factory.py
_graph_transition_manager.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
_inference_manager.py ORTModule GraphTransitionManager (#19007) 2024-07-03 10:53:31 +08:00
_io.py ORTModule GraphTransitionManager (#19007) 2024-07-03 10:53:31 +08:00
_logger.py ORTModule GraphTransitionManager (#19007) 2024-07-03 10:53:31 +08:00
_mem_efficient_grad_mgmt.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
_onnx_models.py ORTModule GraphTransitionManager (#19007) 2024-07-03 10:53:31 +08:00
_pythonop_helper.py ORTModule memory improvement (#18924) 2024-01-16 08:57:37 +08:00
_runtime_inspector.py Flash attention recompute (#20603) 2024-05-21 13:38:19 +08:00
_torch_module_factory.py
_torch_module_interface.py
_torch_module_ort.py
_torch_module_pytorch.py
_training_manager.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
_utils.py Update ruff and clang-format versions (#21479) 2024-07-24 11:50:11 -07:00
_zero_stage3_compatibility.py ORTModule GraphTransitionManager (#19007) 2024-07-03 10:53:31 +08:00
graph_optimizer_registry.py
options.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
ortmodule.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00