pytorch

d4f065d2 - Return mode object from __enter__ (#80998)

Return mode object from __enter__ (#80998) This makes `with Mode() as m:` work. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/80998 Approved by: https://github.com/samdow