OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
1afbc985fe
pytorch/torch/distributed/tensor
History
alanhe151220037 1afbc985fe Make RNGStateTracker support cuda-like device (#106771) 
replace  `CudaRNGStateTracker` with `RNGStateTracker` by rewriting some Cuda-binding code with `device_handle`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106771
Approved by: https://github.com/wanchaol
2023-08-10 19:14:33 +00:00
..
parallel Make RNGStateTracker support cuda-like device (#106771) 2023-08-10 19:14:33 +00:00
__init__.py