From a166a81e4d1b4ebf2951b2385171ab125ad8be16 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 29 Dec 2023 20:12:44 +0300 Subject: [PATCH] IPEX disable Linalg Solve CPU offload with IPEX 2.1 --- modules/intel/ipex/hijacks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/intel/ipex/hijacks.py b/modules/intel/ipex/hijacks.py index 4573b7f7f..27d367e4f 100644 --- a/modules/intel/ipex/hijacks.py +++ b/modules/intel/ipex/hijacks.py @@ -221,7 +221,8 @@ def ipex_hijacks(): torch.UntypedStorage.is_cuda = is_cuda torch.nn.functional.interpolate = interpolate - torch.linalg.solve = linalg_solve + if hasattr(torch.xpu, "Generator"): + torch.linalg.solve = linalg_solve torch.bmm = torch_bmm torch.cat = torch_cat