mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-12 17:57:38 +00:00
### Description <!-- Describe your changes. --> New logic to share allocators among module, optimizer and eval sessions for Training scenario ### 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. --> Previously on device training using shared allocator by sharing EP, now with new mechanism to share allocator, we need to explicitly register allocator in the environment. --------- Co-authored-by: Lei Cao <leca@microsoft.com> |
||
|---|---|---|
| .. | ||
| orttraining | ||
| pytorch_frontend_examples | ||
| tools | ||