Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "a224f4f235674e5295f0697488894e2b2c75356c"
Unverified Commit 5eb98f00 authored by Simon Lui's avatar Simon Lui Committed by GitHub
Browse files

Exempt IPEX from non_blocking previews fixing segmentation faults. (#3708)

parent ac151ac1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
comfy/model_management.py
View file @ 5eb98f00
...@@ -659,6 +659,8 @@ def supports_cast(device, dtype): #TODO ...@@ -659,6 +659,8 @@ def supports_cast(device, dtype): #TODO
def device_supports_non_blocking(device): def device_supports_non_blocking(device):
if is_device_mps(device): if is_device_mps(device):
return False #pytorch bug? mps doesn't support non blocking return False #pytorch bug? mps doesn't support non blocking
if is_intel_xpu():
return False
if args.deterministic: #TODO: figure out why deterministic breaks non blocking from gpu to cpu (previews) if args.deterministic: #TODO: figure out why deterministic breaks non blocking from gpu to cpu (previews)
return False return False
if directml_enabled: if directml_enabled:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment