fix(disagg): allow fake transfer with decode DCP (#35409)

Signed-off-by: Alexandre Milesi <milesial@users.noreply.github.com>
2026-08-19 13:54:37 -07:00
parent defb2a3100
commit ed12d6827d
2 changed files with 20 additions and 4 deletions
@@ -524,12 +524,22 @@ class TestLoadBalanceMethod(unittest.TestCase):
    def test_pd_decode_dcp_rejects_unsupported_transfer_backend(self):
        server_args = ServerArgs(
            model_path="dummy",
+            disaggregation_mode="decode",
+            disaggregation_transfer_backend="mori",
+            dcp_size=4,
+        )
+        with self.assertRaisesRegex(
+            ValueError, "mooncake, nixl, or fake for synthetic benchmarking"
+        ):
+            server_args._handle_pd_disaggregation()
+
+    def test_pd_decode_dcp_allows_fake_transfer_backend(self):
+        server_args = self._load_balance_args(
            disaggregation_mode="decode",
            disaggregation_transfer_backend="fake",
            dcp_size=4,
        )
-        with self.assertRaisesRegex(ValueError, "mooncake or nixl"):
-            server_args._handle_pd_disaggregation()
+        self.assertTrue(server_args.disable_radix_cache)

    def test_pd_decode_dcp_rejects_radix_cache(self):
        server_args = ServerArgs(