Cherry-pick fused_adam_cuda related patches and add its unit tests to the ROCm extension test script

ROCm / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

BSD 3-Clause "New" or "Revised" License

17 stars 14 forks source link

Open hubertlu-tw opened 1 year ago

hubertlu-tw commented 1 year ago

jenkins: retest this please

hubertlu-tw commented 1 year ago

jenkins: retest this please

hubertlu-tw commented 1 year ago

jenkins: retest this please