..
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
[ORTModule] ATen Efficient Attention and Triton Flash Attention ( #17959 )
2023-10-27 10:29:27 +08:00
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