Deploy Kimi-K2.5 (1 trillion parameters) on AMD Instinct GPUs.
Kimi-K2.5 requires vLLM nightly build (rocm/vllm-dev:nightly). The stable release does not yet support this model.
These settings are mandatory for Kimi-K2.5 on AMD:
VLLM_ROCM_USE_AITER=0 - AITER disabled (MLA head count incompatibility with TP=4)--tensor-parallel-size 4 - Not 8! Required for MLA attention head distribution (64/4=16 heads per GPU)--block-size 1 - Required for MLA architecture--trust-remote-code - Model uses custom codeVLLM_USE_TRITON_FLASH_ATTN=0 - Required for vision encoderWhy AITER is Disabled?
Unlike other models, Kimi-K2.5 runs with AITER disabled. The AITER MLA backend requires specific head counts (16 or 128 per GPU). With TP=8, Kimi-K2.5's 64 heads would give 8 heads per GPU (unsupported). With TP=4, we get 16 heads per GPU, but AITER still has compatibility issues with this model's MLA implementation.
Enable chat, tool calling, and reasoning mode:
MI325X (256GB) easily fits the model with TP=4, leaving significant room for KV cache.
Multi-run means (n=5).
See Kimi-K2.5 Stress Testing for detailed benchmark results including saturation testing up to 1,000 concurrent requests.
Unlike other models, Kimi-K2.5 requires TP=4:
The MLA architecture has 64 attention heads that must be evenly distributed with 16 heads per GPU.
Like DeepSeek V3.2, Kimi-K2.5 uses MLA which is incompatible with vLLM's KV cache offloading. The 256GB per MI325X provides sufficient capacity without offloading.
Do not use --kv-cache-dtype fp8:
The model uses its own compressed-tensors quantization format.
Ensure AITER_ENABLE_VSKIP=0 is set. If unset, it defaults to true which causes crashes on MI300X/MI325X.
If you see attention head distribution errors, verify you're using --tensor-parallel-size 4, not 8.
Kimi-K2.5 requires the nightly vLLM build. Use rocm/vllm-dev:nightly instead of the stable release.
Ensure VLLM_USE_TRITON_FLASH_ATTN=0 is set for the MoonViT encoder to work correctly.
0 Comments
Be the first to comment and share your perspective with the community.