issues
search
ROCm
/
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
18
stars
14
forks
source link
use __launch_bounds__ for multi_tensor_apply
#44
Closed
jeffdaily
closed
3 years ago