Skip to content

Instantly share code, notes, and snippets.

@ezyang
Created December 12, 2022 13:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ezyang/2a4894db201d35b645cb700db92b54bb to your computer and use it in GitHub Desktop.
Save ezyang/2a4894db201d35b645cb700db92b54bb to your computer and use it in GitHub Desktop.
diff --git a/torch/_functorch/aot_autograd.py b/torch/_functorch/aot_autograd.py
index c44538e411..cb228e53d0 100644
--- a/torch/_functorch/aot_autograd.py
+++ b/torch/_functorch/aot_autograd.py
@@ -1868,9 +1868,6 @@ def aot_dispatch_autograd(flat_fn, flat_args: List[Tensor], aot_config: AOTConfi
)
fw_outs_including_aliases.append(aliased_out)
- for inner_out, user_out in zip(fw_outs, fw_outs_including_aliases):
- # Sanity check assert
- assert type(inner_out) == type(user_out)
return fw_outs_including_aliases
else:
return fw_outs
diff --git a/torch/fx/experimental/symbolic_shapes.py b/torch/fx/experimental/symbolic_shapes.py
index 021c8382e0..f854731f2b 100644
--- a/torch/fx/experimental/symbolic_shapes.py
+++ b/torch/fx/experimental/symbolic_shapes.py
@@ -878,7 +878,8 @@ class ShapeEnv(object):
pass
return
except RecursionError:
- raise RuntimeError(f"RecursionError in sympy.solve({lhs} - {rhs}, {free[0]})")
+ # raise RuntimeError(f"RecursionError in sympy.solve({lhs} - {rhs}, {free[0]})")
+ return
@lru_cache(256)
def evaluate_expr(self, expr: "sympy.Expr"):
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment