onnxruntime/orttraining
pengwa 89ef987ab1
Improve NonZero on CUDA/ROCM (#10307)
* improve NonZero

* fix megatron_fp16 optimzier, fix the doc

* multi_tensor_applier

* resolve comment

* fix building warning

* fix build error when enabling training and use tensorrt
2022-03-25 07:35:45 +08:00
..
orttraining Improve NonZero on CUDA/ROCM (#10307) 2022-03-25 07:35:45 +08:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools update the amd ci pipeline (#10723) 2022-03-07 18:39:42 +08:00