mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-30 23:18:20 +00:00
### Enable conditional optimization on inputs Label sparsity based optimization can be enabled depending on the input inspection result. So this PR introduce a conditional optimization path for ORTModule, where we automatically detect data sparsity from label or embedding, and enable the graph optimization accordingly without any user interaction. This feature had a new requirement of delaying passing pre_grad graph transformation config to OrtModuleGraphBuilder, from `Initialize` phase to its `Build` phase. Because once after `_initialize_graph_builder` we can detect the input sparsity, and make a decision to enable the label/embed sparisty based graph optimizations. Add UT cases for label/embed input runtime inspector. |
||
|---|---|---|
| .. | ||
| deprecated | ||
| training | ||
| checkpointing_utils.py | ||
| ort_trainer.py | ||
| orttraining_pybind_common.h | ||
| orttraining_pybind_state.cc | ||
| orttraining_python_module.cc | ||
| orttraining_python_module_eager.h | ||
| pt_patch.py | ||