-
Notifications
You must be signed in to change notification settings - Fork 368
Add new recipes for Intel XPU(B60, B70) #753
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Changes from 4 commits
0d4e505
5148f3e
5f72314
8d39ffe
f064e20
59f30b0
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -16,6 +16,8 @@ meta: | |
| - google/gemma-4-31B-it | ||
| hardware: | ||
| h100: verified | ||
| arc_pro_b60: verified | ||
| arc_pro_b70: verified | ||
|
|
||
| model: | ||
| model_id: "google/gemma-4-12B-it" | ||
|
|
@@ -74,7 +76,27 @@ compatible_strategies: | |
| - single_node_tp | ||
| - multi_node_tp | ||
|
|
||
| hardware_overrides: {} | ||
| hardware_overrides: | ||
| xpu: | ||
| extra_env: | ||
| VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1" | ||
| VLLM_WORKER_MULTIPROC_METHOD: "spawn" | ||
| extra_args: | ||
| - "--enforce-eager" | ||
| - "--attention-backend" | ||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. it's default option, we can remove. There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I kept
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. got it thanks. |
||
| - "flash_attn" | ||
| - "--trust-remote-code" | ||
| - "--gpu-memory-utilization" | ||
| - "0.92" | ||
| - "--no-enable-prefix-caching" | ||
| - "--max-num-batched-tokens" | ||
| - "8192" | ||
| - "--max-model-len" | ||
| - "5120" | ||
| - "--block-size" | ||
| - "64" | ||
|
jikunshang marked this conversation as resolved.
Outdated
|
||
| - "--tensor-parallel-size" | ||
| - "2" | ||
|
|
||
| strategy_overrides: | ||
| single_node_tp: | ||
|
|
@@ -114,6 +136,7 @@ guide: | | |
| ### Docker (recommended) | ||
| ```bash | ||
| docker pull vllm/vllm-openai:gemma4-unified # NVIDIA (CUDA 13; append -cu129 for CUDA 12.9 hosts) | ||
| docker pull vllm/vllm-openai-xpu:latest # Intel XPU (B60 / B70) | ||
| ``` | ||
| TPU images are published separately by [vllm-project/tpu-inference](https://github.com/vllm-project/tpu-inference); see the Trillium / Ironwood tpu-recipes below for the pinned tag. | ||
|
|
||
|
|
@@ -166,6 +189,38 @@ guide: | | |
| ``` | ||
| On CUDA 12.9 hosts, use the `vllm/vllm-openai:gemma4-unified-cu129` tag instead. | ||
|
|
||
| ### Docker (Intel XPU B60 / B70, TP=2, Full-Featured) | ||
|
|
||
| Validated on 4× Intel Arc Pro B60 / B70 with the official vLLM XPU image. | ||
|
|
||
| ```bash | ||
| docker run --device /dev/dri \ | ||
| -v /dev/dri/by-path:/dev/dri/by-path --shm-size=16g \ | ||
| --privileged --ipc=host -p 8000:8000 \ | ||
| -v ~/.cache/huggingface:/root/.cache/huggingface \ | ||
| -e VLLM_ALLOW_LONG_MAX_MODEL_LEN=1 \ | ||
| -e VLLM_WORKER_MULTIPROC_METHOD=spawn \ | ||
| --entrypoint bash vllm/vllm-openai-xpu:latest \ | ||
| -c "source /opt/intel/oneapi/setvars.sh && exec vllm serve google/gemma-4-12B-it \ | ||
| --enforce-eager \ | ||
| --attention-backend flash_attn \ | ||
| --host 0.0.0.0 \ | ||
| --port 8000 \ | ||
| --trust-remote-code \ | ||
| --enable-auto-tool-choice \ | ||
| --reasoning-parser gemma4 \ | ||
| --tool-call-parser gemma4 \ | ||
| --chat-template examples/tool_chat_template_gemma4.jinja \ | ||
| --limit-mm-per-prompt '{"image": 4, "audio": 1}' \ | ||
| --async-scheduling \ | ||
| --gpu-memory-utilization 0.92 \ | ||
| --no-enable-prefix-caching \ | ||
| --max-num-batched-tokens 8192 \ | ||
| --max-model-len 5120 \ | ||
| --block-size 64 \ | ||
| --tensor-parallel-size 2" | ||
| ``` | ||
|
|
||
| ### Docker (Cloud TPU — Trillium / Ironwood) | ||
| TPU uses the separate `vllm/vllm-tpu` image (no pip wheel). Pull the tag specified by the upstream [Trillium](https://github.com/AI-Hypercomputer/tpu-recipes/tree/main/inference/trillium/vLLM/Gemma4) or [Ironwood](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/inference/ironwood/vLLM/Gemma4/) recipe, then run: | ||
| ```bash | ||
|
|
||
Uh oh!
There was an error while loading. Please reload this page.