diff --git a/tests/integration/test_lists/waives.txt b/tests/integration/test_lists/waives.txt
index 35cad832cf4b..f7c24c1f63a4 100644
--- a/tests/integration/test_lists/waives.txt
+++ b/tests/integration/test_lists/waives.txt
@@ -346,10 +346,6 @@ test_e2e.py::test_trtllm_bench_llmapi_launch[pytorch_backend-llama-v3-llama3-8b]
 unittest/_torch/attention/sparse/dsa/test_req_idx_per_token.py::test_on_update_kv_lens_rebuilds_stale_map SKIP (https://nvbugs/6574939)
 unittest/_torch/attention/test_attention_backends.py::test_attention_backend[deepseekv3_mla-gen-bf16-HND-p32-v1] SKIP (https://nvbugs/6507109)
 unittest/_torch/attention/test_attention_backends.py::test_attention_backend[qwen2_0_5b_gqa_hd64-ctx-bf16-HND-p32-v1] SKIP (https://nvbugs/6641268)
-unittest/_torch/executor/test_kv_pool_rebalance.py::TestPpLoopDrainWiring::test_drain_length_follows_the_ring_depth SKIP (https://nvbugs/6627197)
-unittest/_torch/executor/test_kv_pool_rebalance.py::TestPpLoopDrainWiring::test_drain_suppresses_queueing_then_rebalances_and_resumes SKIP (https://nvbugs/6627197)
-unittest/_torch/executor/test_kv_pool_rebalance.py::TestPpLoopDrainWiring::test_loop_queues_normally_when_no_rebalance_is_pending SKIP (https://nvbugs/6627197)
-unittest/_torch/executor/test_kv_pool_rebalance.py::TestPpLoopDrainWiring::test_no_drain_is_started_while_one_is_pending SKIP (https://nvbugs/6627197)
 unittest/_torch/executor/test_overlap_scheduler.py::test_overlap_scheduler_block_reuse_cache_hit[TorchSampler] SKIP (https://nvbugs/6608387)
 unittest/_torch/executor/test_overlap_scheduler.py::test_overlap_scheduler_consistency[no_reuse-cpp_scheduler-TorchSampler] SKIP (https://nvbugs/6561559)
 unittest/_torch/modeling/test_gemma4_e2e_dummy.py::test_e2e_text_31b_dummy SKIP (https://nvbugs/6607482)
diff --git a/tests/unittest/_torch/executor/test_kv_pool_rebalance.py b/tests/unittest/_torch/executor/test_kv_pool_rebalance.py
index f88596dd855b..74f006ac5927 100644
--- a/tests/unittest/_torch/executor/test_kv_pool_rebalance.py
+++ b/tests/unittest/_torch/executor/test_kv_pool_rebalance.py
@@ -907,6 +907,7 @@ def _make_pp_loop_executor(monkeypatch, *, num_micro_batches=2, agreement=(True,
     exe.enable_iter_perf_stats = False
     exe.iter_counter = 0
     exe._mm_encoder_item_scheduling_enabled = False
+    exe._pending_recompute_pause_ids = set()
 
     # Rebalance state.  _uses_kv_manager_v2() reads the explicit flag first.
     exe._is_kv_manager_v2 = True
@@ -988,6 +989,11 @@ def _fetch_new_requests():
     # fact and ends the loop.
     exe._forward_step_inter_pp = MagicMock(side_effect=_ReachedForward)
 
+    # These control-flow tests run in CPU CI; CUDA synchronization is out of scope.
+    monkeypatch.setattr(
+        "tensorrt_llm._torch.pyexecutor.py_executor.torch.cuda.current_stream",
+        MagicMock(),
+    )
     monkeypatch.setattr(
         "tensorrt_llm._torch.pyexecutor.py_executor.torch.cuda.set_device", MagicMock()
     )