[2/N] elastic-ep: Enable EPLB after scale-up (#30553)

2026-07-30 01:06:56 +08:00
parent d19999b755
commit 62d0f81f16
6 changed files with 113 additions and 8 deletions
@@ -407,6 +407,7 @@ class _ElasticScaleUpEndToEndBase(CustomTestCase):
        self._generate_logprob_ok("post-scale")

        self._run_post_scale_gsm8k()
+        self._generate_logprob_ok("after post-scale workload")


@unittest.skipUnless(
@@ -458,6 +459,7 @@ class TestElasticScaleUp4To5To6(_ElasticScaleUpEndToEndBase):
        self._generate_ok("after second scale")
        self._generate_logprob_ok("after second scale")
        self._run_post_scale_gsm8k()
+        self._generate_logprob_ok("after post-scale workload")


@unittest.skipUnless(