mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-27 00:06:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix xpu function not implemented. (#9026)

Browse Source
This commit is contained in:
comfyanonymous 2025-07-23 15:10:59 -07:00 committed by GitHub
parent 5ad33787de
commit 39dda1d40d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -186,10 +186,7 @@ def get_total_memory(dev=None, torch_total_too=False):
elif is_intel_xpu(): elif is_intel_xpu():
stats = torch.xpu.memory_stats(dev) stats = torch.xpu.memory_stats(dev)
mem_reserved = stats['reserved_bytes.all.current'] mem_reserved = stats['reserved_bytes.all.current']
if torch_version_numeric < (2, 6): mem_total_xpu = torch.xpu.get_device_properties(dev).total_memory
mem_total_xpu = torch.xpu.get_device_properties(dev).total_memory
else:
_, mem_total_xpu = torch.xpu.mem_get_info(dev)
mem_total_torch = mem_reserved mem_total_torch = mem_reserved
mem_total = mem_total_xpu mem_total = mem_total_xpu
elif is_ascend_npu(): elif is_ascend_npu():