minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Fix mlx unit test batch mock (#34181)

Browse Source
This commit is contained in:
Ke Bao
2026-08-10 00:55:55 +08:00
committed by GitHub
parent fb72a37fde
commit 2ab96d0fe0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/unit/hardware_backend/mlx/test_attention_patching.py
+5 -1
View File
@@ -1222,7 +1222,11 @@ class TestMlxOverlapScheduler(unittest.TestCase):
set_completion_time=lambda: events.append(("completion", "r0")) set_completion_time=lambda: events.append(("completion", "r0"))
), ),
) )
batch = SimpleNamespace() batch = SimpleNamespace(
mamba_track_mask_cpu=None,
mamba_track_mask_next_cpu=None,
mamba_decode_batch_idx_cpu=None,
)
result = SimpleNamespace() result = SimpleNamespace()
i = 0 i = 0
logits_output = SimpleNamespace(customized_info=None) logits_output = SimpleNamespace(customized_info=None)