Searched refs:expected_recompiles (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/dynamo/ |
| D | test_recompile_ux.py | 84 expected_recompiles = 2
86 with torch._dynamo.config.patch("cache_size_limit", expected_recompiles):
94 self.assertEqual(compile_counter.frame_count, expected_recompiles)
|
| D | test_subclasses.py | 40 def _check_recompiles(self, fn, inputs1, inputs2, expected_recompiles): argument
42 self.assertEqual(actual_recompiles, expected_recompiles)
440 def _check_recompiles(self, fn, inputs1, inputs2, expected_recompiles): argument
441 _check_recompiles(self, fn, inputs1, inputs2, expected_recompiles)
1823 def _check_recompiles(self, fn, inputs1, inputs2, expected_recompiles): argument
1824 _check_recompiles(self, fn, inputs1, inputs2, expected_recompiles)
|