Skip to content

Commit

Permalink
[PyTorch] Add patches for distributed randomness (#29)
Browse files Browse the repository at this point in the history 
In this PR, we update the pytorch patch for including the DTensor
sharding info in Cuda RNG states.
  • Loading branch information
@lichen225
lichen225 authored Apr 17, 2024
1 parent bbf2860 commit 2f2daaa
Showing 1 changed file with 679 additions and 90 deletions.
Loading

0 comments on commit 2f2daaa

Please sign in to comment.