diff --git a/test/dynamo/test_dynamo.py b/test/dynamo/test_dynamo.py index 76bddba7b2d..78d192996ee 100644 --- 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 new file mode 100644 index 00000000000..020003647c8 --- /dev/null +++ b/torch_patches/.torch_pin @@ -0,0 +1 @@ +#122349