onnxruntime/orttraining/orttraining/python
Ashwini Khade cc7799835e
Enable a single build with optimized inference and on device training (#14241)
### Description
Right now prepacking code is not compiled when training is enabled. Our
partners want a single build of ort which can do both optimized
inference + training on device. This PR enables prepacking code in a
training build and controls whether it is enabled or not using already
existing session option - kOrtSessionOptionsConfigDisablePrepacking

For Inference scenarios - prepacking will be turned on by default and
this behavior remains the same after this PR too.
For training scenarios - prepacking will be disabled by default and if
user explicitly enables it then an error will be thrown.



### Motivation and Context
Enable both optimized inference as well as on device training in a
single build. For on device training use flag --enable_training_apis.
2023-01-12 21:36:43 -08:00
..
deprecated Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
training Enable a single build with optimized inference and on device training (#14241) 2023-01-12 21:36:43 -08:00
checkpointing_utils.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
ort_trainer.py Replace distutils by setuptools to import build_ext (#14108) 2023-01-09 11:48:01 +01:00
orttraining_pybind_common.h expose lr scheduler python bindings for on device training. (#13882) 2022-12-22 18:44:04 -08:00
orttraining_pybind_state.cc Improve custom op library handle cleanup (#14099) 2023-01-04 17:56:29 -08:00
orttraining_python_module.cc Support loading widechar paths on windows (#14066) 2022-12-30 16:30:11 -08:00
orttraining_python_module_eager.h Abjindal/clean eager backend (#10055) 2022-01-19 14:20:09 -08:00
pt_patch.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00