From ac05301f9ae88e704f5bbc0e4bf052f94af65531 Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Fri, 15 Aug 2025 16:05:27 +0300
Subject: [PATCH] Enable explicit synchronize for ROCm on Windows

---
 modules/processing_callbacks.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/modules/processing_callbacks.py b/modules/processing_callbacks.py
index 72429cf55..447358226 100644
--- a/modules/processing_callbacks.py
+++ b/modules/processing_callbacks.py
@@ -1,5 +1,6 @@
 import typing
 import os
+import sys
 import time
 import torch
 import numpy as np
@@ -53,7 +54,7 @@ def diffusers_callback_legacy(step: int, timestep: int, latents: typing.Union[to
 
 def diffusers_callback(pipe, step: int = 0, timestep: int = 0, kwargs: dict = {}):
     t0 = time.time()
-    if devices.backend == "zluda":
+    if devices.backend == "zluda" or (devices.backend == "rocm" and sys.platform == "win32"):
         torch.cuda.synchronize(devices.device)
     latents = kwargs.get('latents', None)
     if debug: