From 9b641becd575b0bc3750004283538b8cc74eb056 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 9 May 2025 11:07:51 +0300 Subject: [PATCH] IPEX fix torch.compile with tensor.to --- modules/intel/ipex/hijacks.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/intel/ipex/hijacks.py b/modules/intel/ipex/hijacks.py index b9630afd3..e439365fb 100644 --- a/modules/intel/ipex/hijacks.py +++ b/modules/intel/ipex/hijacks.py @@ -217,13 +217,13 @@ def torch_tensor(data, *args, dtype=None, device=None, **kwargs): dtype = torch.float32 return original_torch_tensor(data, *args, dtype=dtype, device=device, **kwargs) -original_Tensor_to = torch.Tensor.to +torch.Tensor.original_Tensor_to = torch.Tensor.to @wraps(torch.Tensor.to) def Tensor_to(self, device=None, *args, **kwargs): if check_cuda(device): - return original_Tensor_to(self, return_xpu(device), *args, **kwargs) + return self.original_Tensor_to(return_xpu(device), *args, **kwargs) else: - return original_Tensor_to(self, device, *args, **kwargs) + return self.original_Tensor_to(device, *args, **kwargs) original_Tensor_cuda = torch.Tensor.cuda @wraps(torch.Tensor.cuda)