Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Autodetect IPEX

Browse Source
This commit is contained in:
Disty0
2023-07-04 23:37:36 +03:00
parent 875d0db103
commit 966eed8dd9
15 changed files with 55 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/api/api.py
+1 -1
View File
@@ -605,7 +605,7 @@ class Api:
ram = { 'error': f'{err}' }
try:
import torch
if shared.cmd_opts.use_ipex:
if devices.backend == 'ipex':
system = { 'free': (torch.xpu.get_device_properties(shared.device).total_memory - torch.xpu.memory_allocated()), 'used': torch.xpu.memory_allocated(), 'total': torch.xpu.get_device_properties(shared.device).total_memory }
s = dict(torch.xpu.memory_stats())
allocated = { 'current': s['allocated_bytes.all.current'], 'peak': s['allocated_bytes.all.peak'] }