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)