onnxruntime/orttraining
satyajandhyala 0da085ed48
Propagate Cast operations to maximize lower precision (float16) computation (#7191)
* Added propagate_cast_ops option and PropagateCastOps transformation.

* Added test cases to propagate Cast operations.

* Expose GraphTransformerConfiguration to python interface and added propagate_cast_ops options.

* Added functionality to propagate Cast operations.

* Added logging.

* Apply cast propagation to the subgraphs.
2021-04-14 20:54:24 -07:00
..
orttraining Propagate Cast operations to maximize lower precision (float16) computation (#7191) 2021-04-14 20:54:24 -07:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Add BERT-L perf regression test on MI100 and re-enable batch size test (#7240) 2021-04-05 15:51:52 -07:00