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

cached cast fix #90

Closed hubertlu-tw closed 2 years ago

hubertlu-tw commented 2 years ago

We can close the issue once this PR has been merged.

hubertlu-tw commented 2 years ago

jenkins: retest this please

zstreet87 commented 2 years ago

I'm good with this. There is also a hot fix that's less change if that is desired? https://github.com/NVIDIA/apex/issues/694#issuecomment-918833904

hubertlu-tw commented 2 years ago

jenkins: retest this please