onnxruntime/orttraining/orttraining/python/training/ortmodule
Justin Chu ad312d9677
Enable comprehension simplification in ruff rules (#23414)
Enable comprehension simplification rules (C4) for ruff and apply
autofix.
2025-01-17 08:43:06 -08:00
..
experimental Use ruff as the formatter to replace black-isort (#23397) 2025-01-16 11:14:15 -08:00
graph_optimizers Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
torch_cpp_extensions Bump clang-format from 18.1.8 to 19.1.6 (#23346) 2025-01-14 09:02:04 -08:00
__init__.py Fix typos according to reviewdog report. (#21335) 2024-07-22 13:37:32 -07:00
_custom_autograd_function.py
_custom_autograd_function_exporter.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -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 Enable comprehension simplification in ruff rules (#23414) 2025-01-17 08:43:06 -08:00
_execution_agent.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_fallback.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -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 Enable comprehension simplification in ruff rules (#23414) 2025-01-17 08:43:06 -08:00
_graph_execution_interface.py
_graph_execution_manager.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_graph_execution_manager_factory.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_graph_transition_manager.py Enable comprehension simplification in ruff rules (#23414) 2025-01-17 08:43:06 -08:00
_inference_manager.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_io.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_logger.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_mem_efficient_grad_mgmt.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
_onnx_models.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_pythonop_helper.py ORTModule memory improvement (#18924) 2024-01-16 08:57:37 +08:00
_runtime_inspector.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_torch_module_factory.py
_torch_module_interface.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_torch_module_ort.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_torch_module_pytorch.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_training_manager.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
_utils.py Enable comprehension simplification in ruff rules (#23414) 2025-01-17 08:43:06 -08:00
_zero_stage3_compatibility.py Enable comprehension simplification in ruff rules (#23414) 2025-01-17 08:43:06 -08:00
graph_optimizer_registry.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00
options.py Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
ortmodule.py Target py310 and modernize codebase with ruff (#23401) 2025-01-16 19:10:14 -08:00