Revert "Reland "Removing nondeterminism from partitioner eliminates o…

…ne graph" (#6703)" (#6774)
pytorch · Mar 19, 2024 · 30a261a · 30a261a
1 parent bd4420c
commit 30a261a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/dynamo/test_dynamo.py b/test/dynamo/test_dynamo.py
@@ -574,9 +574,10 @@ def test_resnet18(self):
     # Graph 2: backward
     # Graph 3: optimizer
     # Graph 4: sync input for backward
-    # Graph 5, 6: PyTorch has updated the number of captured by resnet
+    # Graph 5: sync input for backward (TODO(JackCaoG) understand why there are two graphs)
+    # Graph 6, 7: PyTorch has updated the number of captured by resnet
     # (https://github.com/pytorch/pytorch/pull/117434)
-    self.assertEqual(met.metric_data('CompileTime')[0], 6)
+    self.assertEqual(met.metric_data('CompileTime')[0], 7)
     # We execute 4 graphs per step (+ 1 for SGD) when optimizer is enabled.
     self.assertEqual(met.metric_data('ExecuteTime')[0], sample_count * 4 + 1)
     # one for each forward, backward and optimizer