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

Changes to support hipblas migration #113

Closed pruthvistony closed 1 year ago

pruthvistony commented 1 year ago

Changes are migrated from original pytorch PR - https://github.com/pytorch/pytorch/pull/105881