Skip to content

Commit

Permalink
[compiled autograd] clear compiled_autograd_verbose once test is done
Browse files Browse the repository at this point in the history
ghstack-source-id: 03f335799def42f89f37eda70dcdb1cbd4656c62
Pull Request resolved: #126148
  • Loading branch information
xmfan committed May 16, 2024
1 parent 0a5fd75 commit 0272611
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions torch/_dynamo/compiled_autograd.py
Original file line number Diff line number Diff line change
Expand Up @@ -326,3 +326,4 @@ def reset() -> None:
compiled_autograd_enable = False
assert compiled_autograd_enabled_count == 0
torch._C._dynamo.compiled_autograd.set_autograd_compiler(None)
torch._C._dynamo.compiled_autograd.set_verbose_logging(False)

0 comments on commit 0272611

Please sign in to comment.