From 3d8390de9b8280c03679d87fcf9766e10f964198 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sun, 18 May 2025 04:44:51 +0300 Subject: [PATCH] IPEX return devices.dtype instead of bf16 --- modules/intel/ipex/hijacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/intel/ipex/hijacks.py b/modules/intel/ipex/hijacks.py index db4d0e6f3..8c308f921 100644 --- a/modules/intel/ipex/hijacks.py +++ b/modules/intel/ipex/hijacks.py @@ -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)