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

log torch.accelerator

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-11 11:26:27 +02:00
parent ff83baf2ee
commit 0608bdfbb1
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+1
View File
@@ -232,6 +232,7 @@ def torch_gc(force: bool = False, fast: bool = False, reason: str | None = None)
if hasattr(torch, "accelerator") and torch.accelerator.is_available(): # torch >= 2.6
torch.accelerator.synchronize()
torch.accelerator.empty_cache()
torch.accelerator.empty_host_cache()
if torch.cuda.is_available() and hasattr(torch.cuda, "ipc_collect"):
torch.cuda.ipc_collect()
elif hasattr(torch, "xpu") and hasattr(torch.xpu, "ipc_collect"):