OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

TorchDynamo Remove context manager (#85124)

Browse Source 
As title
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85124
Approved by: https://github.com/ezyang
This commit is contained in:
Animesh Jain 2022-09-16 01:20:54 +00:00 committed by PyTorch MergeBot
parent 95a2c3df31
commit 50733c8bba
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
torch/testing/_internal/common_utils.py
View File

@ -1972,8 +1972,9 @@ class TestCase(expecttest.TestCase):
if TEST_WITH_TORCHDYNAMO:
with torchdynamo.optimize("eager"):
super().run(result=result)
# TorchDynamo optimize annotation
super_run = torchdynamo.optimize("eager")(super().run)
super_run(result=result)
# TODO - Reset for each test slows down testing significantly.
# torchdynamo.reset()