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

refactor(runner): rename runner replay/load/can_run for the shared surface (#28384)

Browse Source
This commit is contained in:
Cheng Wan
2026-06-19 01:45:04 -07:00
committed by GitHub
parent 31c0a98066
commit 1c6331cbd6
20 changed files with 73 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/test/kits/attention_unittest/runner_modes/speculative_draft_extend_runner.py
+2 -2
View File
@@ -675,8 +675,8 @@ def run_eagle_draft_extend_cuda_graph_runner_case(
)
adapter.prepare_replay_state(graph_fixture, case, draft_inputs, settings)
testcase.assertTrue(graph_runner.can_run(graph_batch))
actual = graph_runner.replay(graph_batch)
testcase.assertTrue(graph_runner.can_run_graph(graph_batch))
actual = graph_runner.execute(graph_batch)
adapter.assert_outputs_close(actual, expected, settings)
finally:
_reset_cuda_graph_test_buffers()
python/sglang/test/kits/attention_unittest/runner_modes/speculative_draft_runner.py
+4 -4
View File
@@ -544,8 +544,8 @@ def run_eagle_draft_cuda_graph_runner_case(
)
adapter.prepare_replay_state(graph_fixture, case, draft_inputs, settings)
testcase.assertTrue(graph_runner.can_run(graph_batch))
actual = graph_runner.replay(graph_batch)
testcase.assertTrue(graph_runner.can_run_graph(graph_batch))
actual = graph_runner.execute(graph_batch)
adapter.assert_outputs_close(actual, expected, settings)
finally:
_reset_cuda_graph_test_buffers()
@@ -590,8 +590,8 @@ def run_frozen_kv_mtp_cuda_graph_runner_case(
graph_runner = _capture_frozen_kv_mtp_graph_runner(graph_worker)
adapter.prepare_replay_state(graph_fixture, case, draft_inputs, settings)
testcase.assertTrue(graph_runner.can_run(graph_batch))
actual = graph_runner.replay(graph_batch)
testcase.assertTrue(graph_runner.can_run_graph(graph_batch))
actual = graph_runner.execute(graph_batch)
adapter.assert_outputs_close(actual, expected, settings)
finally:
_reset_cuda_graph_test_buffers()