fix traceback in with / try: finally: for Python 3.10

thunder/core/interpreter.py

@@ -7342,7 +7342,7 @@ def _run_frame(
                             runtimectx.exception_stack[-1] = current_exception
                             with frame.interpreter_stack.set_cur_instruction(PseudoInst.EXCEPTION_HANDLER):
                                 frame.interpreter_stack.append(
-                                    current_exception.__traceback__ if exc is not None else None
+                                    current_exception.__traceback__ if current_exception is not None else None
                                 )
                                 frame.interpreter_stack.append(current_exception)
                                 # Python distinguishes explicit exc_type present or NULL/None

thunder/tests/test_interpreter.py

@@ -3559,3 +3559,22 @@ def fn(x):
     res = jfn(x)
     expected = fn(x)
     assert_close(res, expected)
+
+
+def test_reraise_traceback():
+    import traceback
+
+    def bar():
+        print("A" * "A")
+
+    def fn():
+        try:
+            bar()
+        finally:
+            pass
+
+    jfn = thunder.jit(fn)
+    try:
+        jfn()
+    except Exception as e:
+        assert 'print("A" * "A")' in "\n".join(traceback.format_tb(e.__traceback__))