onnxruntime/orttraining/orttraining/python/training/ortmodule
pengwa a49bb85cfe
Manage ORTModule configurations consistently (#16396)
### Manage ORTModule options

Move all env vars that used for feature ON/OFF into runtime options for
consistent managements.


Be noted: the features' switch are assigned in 2 phases: default values,
overwritten by env vars (if specified by users). So env vars take the
highest priority when all 2 phases both given value explicitly for one
feature.



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
2023-06-27 19:19:36 +08:00
..
experimental Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
torch_cpp_extensions Run clang-format in CI (#15524) 2023-04-18 09:26:58 -07:00
__init__.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_custom_autograd_function.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_custom_autograd_function_exporter.py Consolidate ORTModule logging (#16078) 2023-06-01 10:09:12 +08:00
_custom_autograd_function_runner.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_custom_gradient_registry.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_custom_op_symbolic_registry.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_execution_agent.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_fallback.py Consolidate ORTModule logging (#16078) 2023-06-01 10:09:12 +08:00
_fallback_exceptions.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_gradient_accumulation_manager.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_graph_execution_interface.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_graph_execution_manager.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_graph_execution_manager_factory.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_inference_manager.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_io.py Consolidate ORTModule logging (#16078) 2023-06-01 10:09:12 +08:00
_logger.py Consolidate ORTModule logging (#16078) 2023-06-01 10:09:12 +08:00
_onnx_models.py Type hint for ORTModule (#15938) 2023-05-25 09:28:20 +08:00
_runtime_inspector.py Introduce memory observer for ORTModule (#16213) 2023-06-15 15:45:36 +08:00
_torch_module_factory.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_torch_module_interface.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_torch_module_ort.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_torch_module_pytorch.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_training_manager.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
_utils.py Consolidate ORTModule logging (#16078) 2023-06-01 10:09:12 +08:00
options.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
ortmodule.py Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00