search

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

Cherry-picks some commits to replace torch.Tensor and remove dependency on six #107

Closed hubertlu-tw closed 1 year ago

hubertlu-tw commented 1 year ago

This PR will fix this issue: https://github.com/ROCmSoftwarePlatform/apex/issues/106. I have tested it on my dev node with rocm-5.4.0 and PyTorch 1.13.0a0+git941769a which means it should be backward compatible with not only PT 2.0 but also the older PT versions (PT 1.13+ at least).

zstreet87 commented 1 year ago

LGTM. Can we merge once the CI has finished? @hubertlu-tw