mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
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 |
||
|---|---|---|
| .. | ||
| _internal | ||
| __init__.py | ||
| _comparison.py | ||
| _creation.py | ||
| _deprecated.py | ||
| _legacy.py | ||