IPEX return devices.dtype instead of bf16

2026-09-19 17:24:32 +02:00 · 2025-05-18 04:44:51 +03:00
parent d0e6f01286
commit 3d8390de9b
1 changed files with 1 additions and 1 deletions
@@ -82,7 +82,7 @@ original_get_autocast_dtype = torch.get_autocast_dtype
@wraps(torch.get_autocast_dtype)
 def torch_get_autocast_dtype(device_type=None):
    if device_type is None or check_cuda(device_type) or check_device_type(device_type, "xpu"):
-        return torch.bfloat16
+        return devices.dtype
    else:
        return original_get_autocast_dtype(device_type)