IPEX disable interpolate cpu fallback with IPEX 2.5

2026-09-20 01:31:13 +02:00 · 2024-12-22 03:04:40 +03:00
parent bf67ffd34a
commit 042f5f9fed
2 changed files with 3 additions and 1 deletions
@@ -313,7 +313,7 @@ def torch_load(f, map_location=None, *args, **kwargs):

 # Hijack Functions:
 def ipex_hijacks(legacy=True):
-    if legacy:
+    if legacy and float(torch.__version__[:3]) < 2.5:
        torch.nn.functional.interpolate = interpolate
    torch.tensor = torch_tensor
    torch.Tensor.to = Tensor_to