onnxruntime/orttraining
M. Zeeshan Siddiqui 1a2f1bd23a
Enable external CUDA allocator in ORTModule. (#6745)
* Enable external CUDA allocator in ORTModule.

* Fix assert after unification of allocators.

* Update no grad memory test.

* update comments.

* fix provider options array when not sharing allocator.
2021-02-18 20:01:13 -08:00
..
orttraining Enable external CUDA allocator in ORTModule. (#6745) 2021-02-18 20:01:13 -08:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Add nightly pipeline for MI100 to run convergence and batch size test similar to V100. (#6611) 2021-02-12 13:22:06 -08:00