Compiled autograd configs in TLS (#137821)
Summary:
Multithreaded doesn't work yet, this adds python side TLS only for the python side state
X-link: https://github.com/pytorch/pytorch/pull/137821
Approved by: https://github.com/jansel, https://github.com/yf225
ghstack dependencies: #137953
Reviewed By: wdvr
Differential Revision: D64796212
Pulled By: xmfan
fbshipit-source-id: aa1d9ef8f6e61207dfb352866e37d5e7cc98df42
