Skip to content

Commit 575877c

Browse files
committed
Add rl_perf_repro functional test to CI and make script consistent with verl e2e
Signed-off-by: Shuyi Xiong <219646547+shuyixiong@users.noreply.github.com>
1 parent 3a0721d commit 575877c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/integration/defs/ray_orchestrator/RL/run_rl_perf_reproduce.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def __init__(self, async_llm_kwargs: dict, sampling_kwargs: dict):
4141

4242
async def init_llm(self):
4343
"""Initialize the AsyncLLM instance with configured parameters."""
44-
self.llm = AsyncLLM(
44+
self.llm = await AsyncLLM(
4545
model=self.async_llm_kwargs["model"],
4646
backend="pytorch",
4747
orchestrator_type=self.async_llm_kwargs["orchestrator_type"],
@@ -61,7 +61,6 @@ async def init_llm(self):
6161
batch_wait_timeout_iters=32,
6262
batch_wait_max_tokens_ratio=0.5,
6363
)
64-
await self.llm.setup_async()
6564
self.sampling_params = SamplingParams(
6665
temperature=self.sampling_kwargs["temperature"],
6766
top_p=self.sampling_kwargs["top_p"],

tests/integration/test_lists/test-db/l0_dgx_h100.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,3 +301,4 @@ l0_dgx_h100:
301301
tests:
302302
- unittest/_torch/ray_orchestrator/multi_gpu -m "gpu4"
303303
- unittest/llmapi/test_async_llm.py -m "gpu4"
304+
- ray_orchestrator/RL/test_rl_perf_reproduce.py

0 commit comments

Comments
 (0)