Skip compile tests

Xilinx · Nov 25, 2024 · 55e3666 · 55e3666
1 parent 81b6f47
commit 55e3666
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)