[dynamo][moco] Save global torch state to restore on graph break (#101201)
This is relevant to https://github.com/pytorch/pytorch/pull/100570 as well.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/101201
Approved by: https://github.com/voznesenskym
