pytorch/torch/testing
Richard Zou c9a0204ef4 Disable functorch modes in testing's freeze_rng_state(), part 2 (#81109)
I forgot to update one line in
https://github.com/pytorch/pytorch/pull/81006. torch.get_rng_state()
returns a Tensor that can also be affected by modes so it also needs a
no_functorch() context manager.

Test Plan:
- tested with functorch tests on CUDA (that's how I discovered this
problem)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81109
Approved by: https://github.com/samdow
2022-07-08 20:18:56 +00:00
..
_internal Disable functorch modes in testing's freeze_rng_state(), part 2 (#81109) 2022-07-08 20:18:56 +00:00
__init__.py
_comparison.py move MPS compat into common comparison machinery (#77836) 2022-06-08 08:09:18 +00:00
_creation.py [chalf] div, eq, masked_fill, index_put (#77479) 2022-05-18 17:01:08 +00:00
_deprecated.py
_legacy.py [complex32] real and imag (also remove unused real and imag kernels) 2022-05-05 04:36:58 +00:00