diff --git a/tests/brevitas_end_to_end/test_torchvision_models.py b/tests/brevitas_end_to_end/test_torchvision_models.py index 09f0b9253..7ff7a14f9 100644 --- a/tests/brevitas_end_to_end/test_torchvision_models.py +++ b/tests/brevitas_end_to_end/test_torchvision_models.py @@ -132,6 +132,8 @@ def test_torchvision_compile(torchvision_model_compile): torch._dynamo.config.capture_scalar_outputs = True if torchvision_model_compile is None: pytest.skip('Model not instantiated') + if version.parse('2.2.0') <= torch_version <= version.parse('2.3.1'): + pytest.skip('torch.compile not compatible with torch versions between 2.2 and 2.3.1') inp = torch.randn(BATCH, IN_CH, HEIGHT, WIDTH)