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
Cherry-pick fused_adam_cuda related patches and add its unit tests to the ROCm extension test script #93
Open
hubertlu-tw opened 1 year ago
jenkins: retest this please