transformer_engine/pytorch/distributed.py · 811e090859a94d79148c5d902f62cf8d9539af20 · OpenDAS / TransformerEngine

Robin Zhang authored Dec 12, 2025



set_all_rng_states in set_states
Signed-off-by: Robin Zhang <robinz@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

811e0908

distributed.py 75.7 KB