onnxruntime/orttraining
pengwa d209fe29b9
custom autograd func memory refinement (#8993)
* Release torch tensor referenced by torch gradient graph (created in PythonOp)

* Update orttraining/orttraining/python/training/ortmodule/torch_cpp_extensions/torch_interop_utils/torch_interop_utils.cc

* refine with comments

Co-authored-by: Wei-Sheng Chin <wschin@outlook.com>
2021-09-09 18:37:24 +08:00
..
orttraining custom autograd func memory refinement (#8993) 2021-09-09 18:37:24 +08:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Add hugging-face models loss curve and performance guards to ROCm CI pipeline. (#8915) 2021-09-01 09:03:10 -07:00