@@ -74,32 +74,41 @@ export VLLM_REPLICA_COUNT_D="${VLLM_REPLICA_COUNT_D:-2}"
7474# Data Parallel size
7575export VLLM_DATA_PARALLEL_SIZE=" ${VLLM_DATA_PARALLEL_SIZE:- 1} "
7676
77- PRIMARY_PORT=" 0"
78- if [ " ${PD_ENABLED} " != " \" true\" " ] && [ ${VLLM_DATA_PARALLEL_SIZE} -eq 1 ]; then
79- if [ " ${KV_CACHE_ENABLED} " != " true" ]; then
80- DEFAULT_EPP_CONFIG=" deploy/config/sim-epp-config.yaml"
81- else
82- DEFAULT_EPP_CONFIG=" deploy/config/sim-epp-kvcache-config.yaml"
83- fi
84- else
85- if [ " ${KV_CACHE_ENABLED} " != " true" ]; then
86- if [ " ${PD_ENABLED} " == " \" true\" " ]; then
87- DEFAULT_EPP_CONFIG=" deploy/config/sim-pd-epp-config.yaml"
88- if [ ${VLLM_DATA_PARALLEL_SIZE} -ne 1 ]; then
89- PRIMARY_PORT=" 8000"
90- fi
91- else
92- DEFAULT_EPP_CONFIG=" deploy/config/sim-dp-epp-config.yaml"
93- fi
94- else
77+ # Validate configuration constraints
78+ if [ " ${KV_CACHE_ENABLED} " == " true" ]; then
79+ # KV cache requires simple mode: no PD and DP size must be 1
80+ if [ " ${PD_ENABLED} " == " \" true\" " ] || [ ${VLLM_DATA_PARALLEL_SIZE} -ne 1 ]; then
9581 echo " Invalid configuration: PD_ENABLED=true and KV_CACHE_ENABLED=true is not supported"
9682 exit 1
9783 fi
9884fi
9985
100- export EPP_CONFIG=" ${EPP_CONFIG:- ${DEFAULT_EPP_CONFIG} } "
86+ # Set PRIMARY_PORT based on PD mode with data parallelism
87+ if [ " ${PD_ENABLED} " == " \" true\" " ] && [ ${VLLM_DATA_PARALLEL_SIZE} -ne 1 ]; then
88+ PRIMARY_PORT=" 8000"
89+ else
90+ PRIMARY_PORT=" 0"
91+ fi
10192export PRIMARY_PORT
10293
94+ # Determine EPP config file based on feature flags
95+ if [ " ${KV_CACHE_ENABLED} " == " true" ]; then
96+ # KV cache mode (simple mode only)
97+ DEFAULT_EPP_CONFIG=" deploy/config/sim-epp-kvcache-config.yaml"
98+ elif [ " ${PD_ENABLED} " == " \" true\" " ]; then
99+ # Prefill-Decode mode
100+ DEFAULT_EPP_CONFIG=" deploy/config/sim-pd-epp-config.yaml"
101+ elif [ ${VLLM_DATA_PARALLEL_SIZE} -ne 1 ]; then
102+ # Data Parallel mode (only needed for Istio pre-1.28.1)
103+ # Not really called in kind(docker.io/istio/pilot:1.28.1) by "make env-dev-kind"
104+ DEFAULT_EPP_CONFIG=" deploy/config/dp-epp-config.yaml"
105+ else
106+ # Simple mode
107+ DEFAULT_EPP_CONFIG=" deploy/config/sim-epp-config.yaml"
108+ fi
109+
110+ export EPP_CONFIG=" ${EPP_CONFIG:- ${DEFAULT_EPP_CONFIG} } "
111+
103112# ------------------------------------------------------------------------------
104113# Setup & Requirement Checks
105114# ------------------------------------------------------------------------------
0 commit comments