IPEX fix missing dtype from randn

Disty0 · Disty0 · commit e83df5071719 · 2025-09-17T19:40:27.000+03:00
diff --git a/modules/intel/ipex/hijacks.py b/modules/intel/ipex/hijacks.py
@@ -322,9 +322,9 @@ def torch_randn(*args, device=None, dtype=None, **kwargs):
     if dtype is bytes:
         dtype = None
     if check_cuda(device):
-        return original_torch_randn(*args, device=return_xpu(device), **kwargs)
+        return original_torch_randn(*args, device=return_xpu(device), dtype=dtype, **kwargs)
     else:
-        return original_torch_randn(*args, device=device, **kwargs)
+        return original_torch_randn(*args, device=device, dtype=dtype, **kwargs)
 
 original_torch_ones = torch.ones
 @wraps(torch.ones)
