Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

IPEX disable Linalg Solve CPU offload with IPEX 2.1

Browse Source
This commit is contained in:
Disty0
2023-12-29 20:12:44 +03:00
parent f4d4f8dad5
commit a166a81e4d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/intel/ipex/hijacks.py
+2 -1
View File
@@ -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