diff --git a/src/agentic_pd_hybrid/benchmark.py b/src/agentic_pd_hybrid/benchmark.py index 0811331..c362998 100644 --- a/src/agentic_pd_hybrid/benchmark.py +++ b/src/agentic_pd_hybrid/benchmark.py @@ -49,6 +49,7 @@ class BenchmarkConfig: backpressure_max_pause_s: float = 2.0 kvcache_migration_reject_threshold: int = 3 kvcache_load_floor_bonus: int = 0 + enable_d_to_p_sync: bool = False sample_profile: str = "default" min_initial_input_tokens: int | None = None max_initial_input_tokens: int | None = None @@ -199,6 +200,7 @@ def run_live_benchmark(config: BenchmarkConfig) -> BenchmarkArtifacts: pool_poll_interval_s=config.pool_poll_interval_s, pool_poll_include_sessions=config.pool_poll_include_sessions, enable_backpressure=config.enable_backpressure, + enable_d_to_p_sync=config.enable_d_to_p_sync, backpressure_max_pause_s=config.backpressure_max_pause_s, kvcache_migration_reject_threshold=config.kvcache_migration_reject_threshold, kvcache_load_floor_bonus=config.kvcache_load_floor_bonus, diff --git a/src/agentic_pd_hybrid/cli.py b/src/agentic_pd_hybrid/cli.py index 4788266..712fb26 100644 --- a/src/agentic_pd_hybrid/cli.py +++ b/src/agentic_pd_hybrid/cli.py @@ -819,6 +819,7 @@ def main() -> None: backpressure_max_pause_s=args.backpressure_max_pause_s, kvcache_migration_reject_threshold=args.kvcache_migration_reject_threshold, kvcache_load_floor_bonus=args.kvcache_load_floor_bonus, + enable_d_to_p_sync=args.enable_d_to_p_sync, sample_profile=args.sample_profile, min_initial_input_tokens=args.min_initial_input_tokens, max_initial_input_tokens=args.max_initial_input_tokens,