hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Initial IPEX support

Browse Source
This commit is contained in:
Nuullll
2023-11-10 11:06:26 +08:00
parent f0f100e67b
commit 8b40f475a3
2 changed files with 51 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
modules/devices.py
View File

@@ -3,7 +3,7 @@ import contextlib
from functools import lru_cache
import torch
from modules import errors, shared
from modules import errors, shared, xpu_specific
if sys.platform == "darwin":
from modules import mac_specific
@@ -30,6 +30,9 @@ def get_optimal_device_name():
if has_mps():
return "mps"
if xpu_specific.has_ipex:
return xpu_specific.get_xpu_device_string()
return "cpu"
@@ -100,11 +103,15 @@ def autocast(disable=False):
if dtype == torch.float32 or shared.cmd_opts.precision == "full":
return contextlib.nullcontext()
if xpu_specific.has_xpu:
return torch.autocast("xpu")
return torch.autocast("cuda")
def without_autocast(disable=False):
return torch.autocast("cuda", enabled=False) if torch.is_autocast_enabled() and not disable else contextlib.nullcontext()
device_type = "xpu" if xpu_specific.has_xpu else "cuda"
return torch.autocast(device_type, enabled=False) if torch.is_autocast_enabled() and not disable else contextlib.nullcontext()
class NansException(Exception):