Don't re-wrap exceptions in exc.TorchOpTracingError (#84)

Author: jansel

helion/_compiler/type_propagation.py

@@ -424,6 +424,8 @@ def propagate_unary(self, op: ast.unaryop, origin: Origin) -> TypeInfo:
             return SymBoolType.new_unbacked(origin)
         try:
             return TypeInfo.from_example(_eval_unary(op, self.fake_value), origin)
+        except exc.Base:
+            raise
         except Exception as e:
             raise exc.TorchOpTracingError(e) from e
 
@@ -1564,6 +1566,8 @@ def _compare(self, op: ast.cmpop, left: TypeInfo, right: TypeInfo) -> TypeInfo:
                         _eval_compare(op, left_example, right_example),
                         self.origin(),
                     )
+                except exc.Base:
+                    raise
                 except Exception as e:
                     raise exc.TorchOpTracingError(e) from e
         if isinstance(left, UnknownType):
@@ -1733,6 +1737,8 @@ def visit_BinOp(self, node: ast.BinOp) -> TypeInfo:
                     _eval_binary(node.op, left_example, right_example),
                     self.origin(),
                 )
+            except exc.Base:
+                raise
             except Exception as e:
                 raise exc.TorchOpTracingError(e) from e