Created
December 12, 2022 13:25
-
-
Save ezyang/2a4894db201d35b645cb700db92b54bb to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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