Revert "[RL] Support Rollout Routing Replay (#5321)" (#5402)

This reverts commit 96d2d4877b.
This commit is contained in:
Jiang-Jia-Jun
2025-12-05 20:19:39 +08:00
committed by GitHub
parent 94c57e4175
commit c45e064f3d
24 changed files with 24 additions and 592 deletions
+1 -3
View File
@@ -90,7 +90,7 @@ class MockAttentionBackend:
class MockQuantMethod:
def apply(self, layer, x, gate, topk_ids_hookfunc=None):
def apply(self, layer, x, gate):
return x
@@ -129,7 +129,6 @@ class TestChunkedMoE(unittest.TestCase):
model_runner.speculative_decoding = False
model_runner._init_share_inputs(mock_fd_config.scheduler_config.max_num_seqs)
model_runner.share_inputs["caches"] = None
model_runner.routing_replay_manager = None
if dist.get_rank() == 0:
model_runner.share_inputs["ids_remove_padding"] = paddle.ones([10])
@@ -149,7 +148,6 @@ class TestChunkedMoE(unittest.TestCase):
fused_moe.fd_config = mock_fd_config
fused_moe.quant_method = MockQuantMethod()
fused_moe.enable_routing_replay = None
return fused_moe
def run_model_runner(self):