From 1579a82d17e0650b61327e226f489fc4dcdbb21f Mon Sep 17 00:00:00 2001
From: Lijuan Tang <tang.lij@northeastern.edu>
Date: Mon, 6 Jul 2026 13:43:40 -0700
Subject: [PATCH] [MLX] Fix FakeOverlapScheduler test stub broken by forward_ct
 accounting (#30125)

---
 .../unit/hardware_backend/mlx/test_attention_patching.py   | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py
index f57888141..6cfd25d37 100644
--- a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py
+++ b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py
@@ -1538,6 +1538,13 @@ if _HAS_MLX:
             self.last_batch = None
             self.processed_batch = None
             self.processed_result = None
+            # _finalize_mlx_pending_job now advances forward_ct and runs the
+            # profiler batch predicate (mirroring run_batch); stub both so the
+            # overlap accounting added in #29217 has something to call.
+            self.forward_ct = 0
+            self.profiler_manager = SimpleNamespace(
+                _profile_batch_predicate=lambda batch: None
+            )
 
         def process_batch_result(self, batch, result):
             self.processed_batch = batch