diff --git a/.github/configs/amd-master.yaml b/.github/configs/amd-master.yaml index fa48ee1be..b9560803e 100644 --- a/.github/configs/amd-master.yaml +++ b/.github/configs/amd-master.yaml @@ -316,7 +316,7 @@ kimik2.5-int4-mi355x-vllm: - { tp: 8, conc-start: 4, conc-end: 64 } kimik2.5-int4-mi325x-vllm: - image: vllm/vllm-openai-rocm:v0.16.0 + image: vllm/vllm-openai-rocm:v0.18.0 model: moonshotai/Kimi-K2.5 model-prefix: kimik2.5 runner: mi325x diff --git a/benchmarks/single_node/kimik2.5_int4_mi325x.sh b/benchmarks/single_node/kimik2.5_int4_mi325x.sh index e6b7629ea..8ca56dc29 100755 --- a/benchmarks/single_node/kimik2.5_int4_mi325x.sh +++ b/benchmarks/single_node/kimik2.5_int4_mi325x.sh @@ -33,13 +33,14 @@ PORT=${PORT:-8888} start_gpu_monitor set -x +export VLLM_ROCM_USE_AITER=1 vllm serve $MODEL --port $PORT \ --tensor-parallel-size=$TP \ --gpu-memory-utilization 0.95 \ --max-model-len $MAX_MODEL_LEN \ --block-size=64 \ ---disable-log-requests \ --trust-remote-code \ +--max-num-seqs 256 \ --mm-encoder-tp-mode data > $SERVER_LOG 2>&1 & SERVER_PID=$! diff --git a/perf-changelog.yaml b/perf-changelog.yaml index d14463135..6a8a6e666 100644 --- a/perf-changelog.yaml +++ b/perf-changelog.yaml @@ -1093,6 +1093,15 @@ - "Add --max-num-seqs 256, remove --disable-log-requests" pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/950 +- config-keys: + - kimik2.5-int4-mi325x-vllm + description: + - "Upgrade vLLM ROCm image from v0.16.0 to v0.18.0" + - "Enable AITER MLA, export VLLM_ROCM_USE_AITER=1, https://github.com/vllm-project/vllm/issues/35641" + - "Triton Fused Moe Tuning https://github.com/vllm-project/vllm/pull/35093" + - "Add --max-num-seqs 256, remove --disable-log-requests" + pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/957 + - config-keys: - gptoss-fp4-h100-vllm - gptoss-fp4-h200-vllm