commit ef2bb9ca04e77ee2d1f2f11c72ef76ab756701e2
author PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue Dec 20 04:43:26 2022 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue Dec 20 04:43:26 2022 +0000
tree 6e2070d20199efa0b0fbd75e148b1e2ebbd6f124
parent 0f57e7f2d9c9835fd504602dfb2fc12e6c08a74b

Revert "When nopython=True, Dynamo can't allow graph breaks. (#90970)"

This reverts commit 7e9bf2ed860b8b60d252eead4cc457c3fe5f1667.

Reverted https://github.com/pytorch/pytorch/pull/90970 on behalf of https://github.com/kit1980 due to The inductor test fails on master every time after this PR

torch/_dynamo/symbolic_convert.py

diff --git a/torch/_dynamo/symbolic_convert.py b/torch/_dynamo/symbolic_convert.py
index 68bcf5e..436aed9 100644
--- a/torch/_dynamo/symbolic_convert.py
+++ b/torch/_dynamo/symbolic_convert.py
@@ -306,7 +306,7 @@
             try:
                 return inner_fn(self, inst)
             except Unsupported as excp:
-                if self.has_backedge() and self.should_compile_partial_graph():
+                if self.has_backedge():
                     msg = "Skipping frame because there is a graph break in a for/while loop"
                     log.debug(msg)
                     raise exc.SkipFrame(msg) from excp