Merge branch 'main' into windows-cuda

pytorch · Jun 5, 2023 · f685927 · f685927
2 parents c9144c3 + a5035df
commit f685927
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_models.py b/test/test_models.py
@@ -29,8 +29,10 @@
 def disable_tf32():
     previous = torch.backends.cudnn.allow_tf32
     torch.backends.cudnn.allow_tf32 = False
-    yield
-    torch.backends.cudnn.allow_tf32 = previous
+    try:
+        yield
+    finally:
+        torch.backends.cudnn.allow_tf32 = previous
 
 
 def list_model_fns(module):