From bbfcda48cebfbde2ed4be4e8273eb593c764e638 Mon Sep 17 00:00:00 2001
From: Cheng Wan <54331508+ch-wan@users.noreply.github.com>
Date: Fri, 18 Sep 2026 01:59:07 -0700
Subject: [PATCH] [CI] Add a unified-memory rerun test group (#40147)

---
 scripts/ci/rerun_test_groups.json | 46 +++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/scripts/ci/rerun_test_groups.json b/scripts/ci/rerun_test_groups.json
index 57069f798..f5b13d1f2 100644
--- a/scripts/ci/rerun_test_groups.json
+++ b/scripts/ci/rerun_test_groups.json
@@ -4,5 +4,51 @@
     "registered/core/test_srt_endpoint.py",
     "registered/vlm/test_rust_native_mm_e2e.py",
     "registered/vlm/test_rust_native_mm_mmmu.py"
+  ],
+  "unified-memory": [
+    "registered/attention/test_gemma4_unified_swa_virtual_ids.py",
+    "registered/attention/test_unified_memory_deterministic.py",
+    "registered/disaggregation/test_disaggregation_unified_memory.py",
+    "registered/e2e/disaggregation/test_disaggregation_unified_memory_swa.py",
+    "registered/e2e/disaggregation/test_disaggregation_unified_memory_tri.py",
+    "registered/e2e/models/test_inkling_unified.py",
+    "registered/e2e/models/test_kimi_linear_models.py",
+    "registered/e2e/models/test_kimi_linear_unified_memory.py",
+    "registered/e2e/models/test_kimi_linear_unified_memory_dcp_blackwell.py",
+    "registered/page_major/test_page_major_gpt_oss.py",
+    "registered/page_major/test_page_major_qwen_hybrid.py",
+    "registered/unit/disaggregation/test_unified_memory_move_gate.py",
+    "registered/unit/layers/attention/test_kv_translate_ownership.py",
+    "registered/unit/managers/test_prefill_adder.py",
+    "registered/unit/managers/test_scheduler_init_req_max_new_tokens.py",
+    "registered/unit/mem_cache/test_dsv4_unified_fp8_pool.py",
+    "registered/unit/mem_cache/test_flashkda_strided_state_access.py",
+    "registered/unit/mem_cache/test_full_loc_fast_path.py",
+    "registered/unit/mem_cache/test_hisparse_allocator.py",
+    "registered/unit/mem_cache/test_hisparse_max_token_pool_size.py",
+    "registered/unit/mem_cache/test_kda_fused_decode_strided_state.py",
+    "registered/unit/mem_cache/test_kv_index_translator.py",
+    "registered/unit/mem_cache/test_multi_ended_allocator.py",
+    "registered/unit/mem_cache/test_pd_envelope_transfer_layout.py",
+    "registered/unit/mem_cache/test_prefill_memory_budget.py",
+    "registered/unit/mem_cache/test_swa_locked_full_recover_unified.py",
+    "registered/unit/mem_cache/test_unified_byte_accounting.py",
+    "registered/unit/mem_cache/test_unified_byte_budget_sizing.py",
+    "registered/unit/mem_cache/test_unified_capacity_memo.py",
+    "registered/unit/mem_cache/test_unified_free_no_host_sync.py",
+    "registered/unit/mem_cache/test_unified_handout_zeroing.py",
+    "registered/unit/mem_cache/test_unified_mamba_views.py",
+    "registered/unit/mem_cache/test_unified_mha_views.py",
+    "registered/unit/mem_cache/test_unified_mla_gpu_parity.py",
+    "registered/unit/mem_cache/test_unified_mla_views.py",
+    "registered/unit/mem_cache/test_unified_npool_sweep.py",
+    "registered/unit/mem_cache/test_unified_radix_allocation_eviction.py",
+    "registered/unit/mem_cache/test_unified_swa_shared_virtual_ids.py",
+    "registered/unit/mem_cache/test_unified_tri_pool.py",
+    "registered/unit/model_executor/test_pool_configurator.py",
+    "registered/unit/server_args/test_page_major_backend_allowlist.py",
+    "registered/unit/server_args/test_unified_prefill_cuda_graph_gate.py",
+    "registered/unit/server_args/test_unified_tbo_gate.py",
+    "registered/unit/test_model_overrides.py"
   ]
 }