mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-28 22:56:32 +00:00
Motivation: PythonOp is saving input for backward, it's risky since ONNX Runtime backend is not aware of this, the tensor buffer may be "released" by ORT, then potentially modified by other operators before backward function executes. Fix: This pr just clone all input of PythonOp before forward is invoked. This may be high overhead, it's just a workaround before a better fix. |
||
|---|---|---|
| .. | ||
| 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 | ||