Weixing Zhang a3f17c8b0d update lamb and GatherGrad kernel for ROCm EP (#7184) ... With ROCm4.1, the CUDA implementation of Lamb and GatherGrad can be utilized for ROCm EP.		2021-04-02 09:02:49 -07:00
..
orttraining	update lamb and GatherGrad kernel for ROCm EP (#7184)	2021-04-02 09:02:49 -07:00
pytorch_frontend_examples	Sync ORTModule branch with master and fix tests (#6526)	2021-02-02 08:59:56 -08:00
tools	add Dockerfile.rocm4.1.pytorch (#7152)	2021-03-26 21:40:10 -07:00