A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
19
stars
19
forks
source link
Skip some failed unit tests from the FusedRMSNorm PR #84
Closed
hubertlu-tw closed 2 years ago
FusedRMSNorm PR: https://github.com/ROCmSoftwarePlatform/apex/pull/78 The failed unit tests which were skipped in this PR: https://github.com/ROCmSoftwarePlatform/apex/pull/78#issuecomment-1201839061