From: Zachary DeVito <zdevito@fb.com>
Date: Wed, 28 Nov 2018 03:11:47 +0000 (-0800)
Subject: fix enable_cpu_fuser
X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~2665
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=23f901a737c184be1432a37e7d2b37850bbed050;p=platform%2Fupstream%2Fpytorch.git

fix enable_cpu_fuser

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/14440

Differential Revision: D13226354

Pulled By: zdevito

fbshipit-source-id: e4ed023eece8b5b670a4a27d24a8688907b36b90
---

diff --git a/test/test_jit.py b/test/test_jit.py
index 6c81284..c66700b 100644
--- a/test/test_jit.py
+++ b/test/test_jit.py
@@ -207,9 +207,8 @@ def enable_cpu_fuser(fn):
         torch._C._jit_override_can_fuse_on_cpu(True)
         try:
             fn(*args, **kwargs)
-        except Exception:
+        finally:
             torch._C._jit_override_can_fuse_on_cpu(False)
-            raise
     return wrapper