From ec10ac6197fe63d8154a6d753752cf6bcbf1b7c5 Mon Sep 17 00:00:00 2001 From: Rahul Kayaith Date: Mon, 16 Mar 2026 10:43:06 -0700 Subject: [PATCH] [BOO] Sync after profiling to fix flaky event loss Fixes https://github.com/iree-org/iree-turbine/issues/1327 --- iree/turbine/kernel/boo/driver/driver.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/iree/turbine/kernel/boo/driver/driver.py b/iree/turbine/kernel/boo/driver/driver.py index a45f1e843..a963bc8e9 100644 --- a/iree/turbine/kernel/boo/driver/driver.py +++ b/iree/turbine/kernel/boo/driver/driver.py @@ -547,8 +547,14 @@ def pause_and_collect_mem(): ) pause_and_collect_mem() if prof is not None: + for device in devices: + torch.cuda.synchronize(device) prof.step() + # Ensure all GPU work and profiler cleanup is complete before returning. + # Without this, subsequent profiling sessions may lose events. + pause_and_collect_mem() + if results is None: results = () if isinstance(results, torch.Tensor):