IPEX decrease Torch GC Threshold to 80

2026-09-19 09:14:35 +02:00 · 2023-11-21 16:47:35 +03:00
parent 1793c2e116
commit bd141bbfeb
3 changed files with 3 additions and 2 deletions
@@ -134,7 +134,7 @@ def torch_gc(force=False):
    if oom > previous_oom:
        previous_oom = oom
        log.warning(f'GPU out-of-memory error: {mem}')
-    if used > shared.opts.torch_gc_threshold:
+    if used >= shared.opts.torch_gc_threshold:
        log.info(f'GPU high memory utilization: {used}% {mem}')
        force = True
    if not force: