mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-19 19:00:47 +00:00
cc7799835e
### 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. |
||
|---|---|---|
| .. | ||
| experimental | ||
| torch_cpp_extensions | ||
| __init__.py | ||
| _custom_autograd_function.py | ||
| _custom_autograd_function_exporter.py | ||
| _custom_autograd_function_runner.py | ||
| _custom_gradient_registry.py | ||
| _custom_op_symbolic_registry.py | ||
| _execution_agent.py | ||
| _fallback.py | ||
| _fallback_exceptions.py | ||
| _gradient_accumulation_manager.py | ||
| _graph_execution_interface.py | ||
| _graph_execution_manager.py | ||
| _graph_execution_manager_factory.py | ||
| _inference_manager.py | ||
| _io.py | ||
| _logger.py | ||
| _onnx_models.py | ||
| _torch_module_factory.py | ||
| _torch_module_interface.py | ||
| _torch_module_ort.py | ||
| _torch_module_pytorch.py | ||
| _training_manager.py | ||
| _utils.py | ||
| debug_options.py | ||
| ortmodule.py | ||