Merge branch 'main' into fix/position_embeddings

intel · Jul 25, 2024 · 2c696ca · 2c696ca
2 parents 1e9c07c + f3b112a
commit 2c696ca
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test/python/test_compile.py b/test/python/test_compile.py
@@ -89,9 +89,7 @@ def test_torch_compile():
     model = NN()
     y_ref = model(x.to(torch.float32)).detach()
 
-    if (
-        sys.platform == "win32" and Version(torch.__version__) < Version("2.2.2")
-    ) or sys.version_info >= (3, 12):
+    if sys.platform == "win32" and Version(torch.__version__) < Version("2.2.2"):
         with pytest.raises(RuntimeError) as e:
             compiled_model = torch.compile(model, backend="npu")
         assert str(e.value) == "Windows not yet supported for torch.compile"