mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-05 00:14:54 +08:00
Fix code descriptions in the test package. (#148145)
The parameter and function description have something wrong and make them correct. Pull Request resolved: https://github.com/pytorch/pytorch/pull/148145 Approved by: https://github.com/janeyx99
This commit is contained in:
@ -45,7 +45,7 @@ def freeze_rng_state():
|
||||
# In the long run torch.cuda.set_rng_state should probably be
|
||||
# an operator.
|
||||
#
|
||||
# NB: Mode disable is to avoid running cross-ref tests on thes seeding
|
||||
# NB: Mode disable is to avoid running cross-ref tests on this seeding
|
||||
with torch.utils._mode_utils.no_dispatch(), torch._C._DisableFuncTorch():
|
||||
if torch.cuda.is_available():
|
||||
torch.cuda.set_rng_state(cuda_rng_state) # type: ignore[possibly-undefined]
|
||||
|
||||
Reference in New Issue
Block a user