Make compile count in Dynamo test more general (#6802)

pytorch · Mar 22, 2024 · 2e5c3b6 · 2e5c3b6
1 parent f43ac41
commit 2e5c3b6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/test/dynamo/test_dynamo.py b/test/dynamo/test_dynamo.py
@@ -577,9 +577,10 @@ def test_resnet18(self):
     # 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], 7)
+    self.assertLessEqual(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)
+    self.assertLessEqual(
+        met.metric_data('ExecuteTime')[0], sample_count * 4 + 1)
     # one for each forward, backward and optimizer
     self.assertEqual(
         met.metric_data('RunCachedGraphInputData')[0], sample_count * 3)

diff --git a/torch_patches/.torch_pin b/torch_patches/.torch_pin
@@ -0,0 +1 @@
+#122349