Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

IPEX fix dtype mismatch on tensor.to when input is a tensor

Browse Source
This commit is contained in:
Disty0
2025-03-05 23:02:49 +03:00
parent 6f68f91efe
commit 955acaa149
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/intel/ipex/hijacks.py
+2 -5
View File
@@ -33,13 +33,10 @@ def is_cuda(self):
return self.device.type == "xpu" or self.device.type == "cuda"
def check_device_type(device, device_type: str) -> bool:
if device is None or isinstance(device, torch.dtype): # torch.Tensor.to can get a dtype as input
if device is None or type(device) not in {str, int, torch.device}:
return False
else:
if hasattr(device, "device"): # torch.Tensor.to can get a tensor as input
return bool(torch.device(device.device).type == device_type)
else:
return bool(torch.device(device).type == device_type)
return bool(torch.device(device).type == device_type)
def check_cuda(device) -> bool:
return bool(isinstance(device, int) or check_device_type(device, "cuda"))