94057c3d3e
PR Test (NPU) / check-changes (push) Has been cancelled
PR Test (NPU) / pr-gate (push) Has been cancelled
PR Test (NPU) / set-image-config (push) Has been cancelled
PR Test (NPU) / stage-b-test-1-npu-a2 (0) (push) Has been cancelled
PR Test (NPU) / stage-b-test-1-npu-a2 (1) (push) Has been cancelled
PR Test (NPU) / stage-b-test-2-npu-a2 (0) (push) Has been cancelled
PR Test (NPU) / stage-b-test-2-npu-a2 (1) (push) Has been cancelled
PR Test (NPU) / stage-b-test-4-npu-a3 (push) Has been cancelled
PR Test (NPU) / stage-b-test-16-npu-a3 (push) Has been cancelled
PR Test (NPU) / multimodal-gen-test-1-npu-a3 (push) Has been cancelled
PR Test (NPU) / multimodal-gen-test-2-npu-a3 (push) Has been cancelled
PR Test (Arm64) / pr-gate (push) Has been cancelled
PR Test (Arm64) / check-changes (push) Has been cancelled
PR Test (Arm64) / build-test (push) Has been cancelled
PR Test (sgl-router) / gate (push) Has been cancelled
PR Test (sgl-router) / tier-1 — lint (push) Has been cancelled
PR Test (sgl-router) / tier-2 — build + test (push) Has been cancelled
PR Test (sgl-router) / tier-3 — docker (placeholder) (push) Has been cancelled
PR Test (sgl-router) / tier-3 — k8s integration (push) Has been cancelled
PR Test (sgl-router) / tier-3 — e2e (push) Has been cancelled
PR Test (sgl-router) / finish (push) Has been cancelled
PR Test (NPU) / single-node-poc (map[name:qwen3_6_27b_w8a8_1p_in64k_out1k_50ms runner:linux-aarch64-a3-2 test_case:test/registered/ascend/performance/qwen3_6_27b/test_npu_qwen3_6_27b_w8a8_1p_in64k_out1k_50ms.py test_type:perf]) (push) Has been cancelled
PR Test (NPU) / pr-test-npu-finish (push) Has been cancelled
PR Test (Xeon) / pr-gate (push) Has been cancelled
PR Test (Xeon) / check-changes (push) Has been cancelled
PR Test (Xeon) / build-test (, xeon-gnr, base-b-test-cpu) (push) Has been cancelled
PR Test (XPU) / check-changes (push) Has been cancelled
PR Test (XPU) / pr-gate (push) Has been cancelled
PR Test (XPU) / stage-a-test-1-gpu-xpu (push) Has been cancelled
PR Test (XPU) / wait-for-stage-a (push) Has been cancelled
PR Test (XPU) / stage-b-test-1-gpu-xpu (push) Has been cancelled
PR Test (XPU) / finish (push) Has been cancelled
CI Model Inventory / build-inventory (push) Has been cancelled
Lint / lint (push) Has been cancelled
PR Benchmark (SMG Components) / Benchmark Compilation Check (push) Has been cancelled
PR Benchmark (SMG Components) / Benchmark - Manual Policy (push) Has been cancelled
PR Benchmark (SMG Components) / Benchmark - Request Processing (push) Has been cancelled
PR Benchmark (SMG Components) / Benchmark Summary (push) Has been cancelled
PR Test (SMG) / build-wheel (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on windows (x86_64 - auto) (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on macos (x86_64 - auto) (push) Has been cancelled
PR Test (SMG) / python-unit-tests (push) Has been cancelled
PR Test (SMG) / unit-tests (push) Has been cancelled
PR Test (SMG) / benchmarks (push) Has been cancelled
PR Test (SMG) / chat-completions (push) Has been cancelled
PR Test (SMG) / chat-completions-4gpu (push) Has been cancelled
PR Test (SMG) / e2e (push) Has been cancelled
PR Test (SMG) / docker-build-test (push) Has been cancelled
PR Test (SMG) / k8s-integration (push) Has been cancelled
PR Test (SMG) / finish (push) Has been cancelled
PR Test (SMG) / summarize-benchmarks (push) Has been cancelled
Release SGLang Model Gateway Docker Image / publish (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on macos (aarch64 - auto) (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on linux (aarch64 - auto) (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on linux (x86_64 - auto) (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on linux (aarch64 - musllinux_1_1) (push) Has been cancelled
Release SGLang Model Gateway to PyPI / build on linux (x86_64 - musllinux_1_1) (push) Has been cancelled
Release SGLang Model Gateway to PyPI / Build SDist (push) Has been cancelled
Release SGLang Model Gateway to PyPI / Upload to PyPI (push) Has been cancelled
Release SGLang Kernels / build-cu129-matrix (aarch64, 12.9, 3.10, arm-kernel-build-node) (push) Has been cancelled
Release SGLang Kernels / build-cu129-matrix (x86_64, 12.9, 3.10, x64-kernel-build-node) (push) Has been cancelled
Release SGLang Kernels / release-cu129 (push) Has been cancelled
Release SGLang Kernels / build-cu130-matrix (aarch64, 13.0, 3.10, arm-kernel-build-node) (push) Has been cancelled
Release SGLang Kernels / build-cu130-matrix (x86_64, 13.0, 3.10, x64-kernel-build-node) (push) Has been cancelled
Release SGLang Kernels / release-cu130 (push) Has been cancelled
Release SGLang Kernels / build-rocm-matrix (3.10, 700) (push) Has been cancelled
Release SGLang Kernels / build-rocm-matrix (3.10, 720) (push) Has been cancelled
Release SGLang Kernels / release-rocm700 (push) Has been cancelled
Release SGLang Kernels / release-rocm720 (push) Has been cancelled
Release SGLang Kernels / build-musa43 (43, 3.10) (push) Has been cancelled
Release SGLang Kernels / release-musa43 (push) Has been cancelled
101 lines
2.9 KiB
Markdown
101 lines
2.9 KiB
Markdown
# Multi-Node Deployment
|
|
|
|
## Llama 3.1 405B
|
|
|
|
**Run 405B (fp16) on Two Nodes**
|
|
|
|
```bash
|
|
# replace 172.16.4.52:20000 with your own node ip address and port of the first node
|
|
|
|
python3 -m sglang.launch_server \
|
|
--model-path meta-llama/Meta-Llama-3.1-405B-Instruct \
|
|
--tp 16 \
|
|
--dist-init-addr 172.16.4.52:20000 \
|
|
--nnodes 2 \
|
|
--node-rank 0
|
|
|
|
python3 -m sglang.launch_server \
|
|
--model-path meta-llama/Meta-Llama-3.1-405B-Instruct \
|
|
--tp 16 \
|
|
--dist-init-addr 172.16.4.52:20000 \
|
|
--nnodes 2 \
|
|
--node-rank 1
|
|
```
|
|
|
|
Note that LLama 405B (fp8) can also be launched on a single node.
|
|
|
|
```bash
|
|
python -m sglang.launch_server --model-path meta-llama/Meta-Llama-3.1-405B-Instruct-FP8 --tp 8
|
|
```
|
|
|
|
## DeepSeek V3/R1
|
|
|
|
Please refer to [DeepSeek documents for reference](https://docs.sglang.io/basic_usage/deepseek_v3.html#running-examples-on-multi-node).
|
|
|
|
## Multi-Node Inference on SLURM
|
|
|
|
This example showcases how to serve SGLang server across multiple nodes by SLURM. Submit the following job to the SLURM cluster.
|
|
|
|
```
|
|
#!/bin/bash -l
|
|
|
|
#SBATCH -o SLURM_Logs/%x_%j_master.out
|
|
#SBATCH -e SLURM_Logs/%x_%j_master.err
|
|
#SBATCH -D ./
|
|
#SBATCH -J Llama-405B-Online-Inference-TP16-SGL
|
|
|
|
#SBATCH --nodes=2
|
|
#SBATCH --ntasks=2
|
|
#SBATCH --ntasks-per-node=1 # Ensure 1 task per node
|
|
#SBATCH --cpus-per-task=18
|
|
#SBATCH --mem=224GB
|
|
#SBATCH --partition="lmsys.org"
|
|
#SBATCH --gres=gpu:8
|
|
#SBATCH --time=12:00:00
|
|
|
|
echo "[INFO] Activating environment on node $SLURM_PROCID"
|
|
if ! source ENV_FOLDER/bin/activate; then
|
|
echo "[ERROR] Failed to activate environment" >&2
|
|
exit 1
|
|
fi
|
|
|
|
# Define parameters
|
|
model=MODEL_PATH
|
|
tp_size=16
|
|
|
|
echo "[INFO] Running inference"
|
|
echo "[INFO] Model: $model"
|
|
echo "[INFO] TP Size: $tp_size"
|
|
|
|
# Set NCCL initialization address using the hostname of the head node
|
|
HEAD_NODE=$(scontrol show hostname "$SLURM_NODELIST" | head -n 1)
|
|
NCCL_INIT_ADDR="${HEAD_NODE}:8000"
|
|
echo "[INFO] NCCL_INIT_ADDR: $NCCL_INIT_ADDR"
|
|
|
|
# Launch the model server on each node using SLURM
|
|
srun --ntasks=2 --nodes=2 --output="SLURM_Logs/%x_%j_node$SLURM_NODEID.out" \
|
|
--error="SLURM_Logs/%x_%j_node$SLURM_NODEID.err" \
|
|
python3 -m sglang.launch_server \
|
|
--model-path "$model" \
|
|
--grammar-backend "xgrammar" \
|
|
--tp "$tp_size" \
|
|
--dist-init-addr "$NCCL_INIT_ADDR" \
|
|
--nnodes 2 \
|
|
--node-rank "$SLURM_NODEID" &
|
|
|
|
# Wait for the NCCL server to be ready on port 30000
|
|
while ! nc -z "$HEAD_NODE" 30000; do
|
|
sleep 1
|
|
echo "[INFO] Waiting for $HEAD_NODE:30000 to accept connections"
|
|
done
|
|
|
|
echo "[INFO] $HEAD_NODE:30000 is ready to accept connections"
|
|
|
|
# Keep the script running until the SLURM job times out
|
|
wait
|
|
```
|
|
|
|
Then, you can test the server by sending requests following other [documents](https://docs.sglang.io/basic_usage/openai_api_completions.html).
|
|
|
|
Thanks for [aflah02](https://github.com/aflah02) for providing the example, based on his [blog post](https://aflah02.substack.com/p/multi-node-llm-inference-with-sglang).
|