Skip to content
xLLM xLLM PR preview
EN

Cambricon MLU

Set up an xLLM environment for Cambricon MLU and start a service with the unified xllm serve command from Launch xllm.

  • Install the Cambricon driver and Docker on the Linux host. Obtain an MLU development image with a compatible Neuware runtime and build dependencies from your vendor; a public MLU image is not listed in Quick Start.
  • Check the Model Support List for your backend and prepare the model weights in a directory accessible inside the container.
  • The commands below mount your home directory at the same path. For weights stored elsewhere, add a mount such as -v /path/to/models:/models:ro and use the container path in MODEL_PATH.

Run the following commands on the host. Replace the image placeholder before running them.

Terminal window
XLLM_IMAGE="your-mlu-development-image:tag"
docker pull "$XLLM_IMAGE"
Terminal window
docker run -it \
--name xllm-mlu \
--privileged \
--shm-size=128g \
--ipc=host \
--network=host \
--pid=host \
-v "$HOME:$HOME" \
-w "$HOME" \
"$XLLM_IMAGE" \
/bin/bash

The container remains after exit. Re-enter it with docker start -ai xllm-mlu when stopped, or docker exec -it xllm-mlu bash while it is running.

Run inside the container. If your image already includes the xLLM wheel and supports xllm serve, skip the source installation. Otherwise, clone the repository (or enter your existing checkout), initialize its submodules, and build and install the wheel for this backend:

Terminal window
git clone https://github.com/xLLM-AI/xllm.git
cd xllm
# Install pre-commit on first use
pip install pre-commit
pre-commit install
git submodule update --init --recursive
python setup.py install --device mlu

You can also install directly with python setup.py install; the installer automatically detects the backend from the current PyTorch environment when --device is omitted.

python setup.py build only builds the artifacts; the unified command requires the installed wheel and its packaged server binary. Check the installed command before continuing:

Terminal window
command -v xllm
xllm serve --help

Run inside the container after replacing MODEL_PATH with your weights directory. This example starts one device. Adjust the backend settings using the target model recipe and CLI Reference.

#!/bin/bash
set -e
export MLU_VISIBLE_DEVICES=0
MODEL_PATH="/path/to/model/Qwen3-8B"
PORT=8010
MASTER_NODE_ADDR="127.0.0.1:9748"
NNODES=1
xllm serve \
--model="$MODEL_PATH" \
--model_id=Qwen3-8B \
--port="$PORT" \
--nnodes="$NNODES" \
--master_node_addr="$MASTER_NODE_ADDR" \
--block_size=16

For two devices on one machine, replace the device selection and NNODES lines in the script with:

Terminal window
export MLU_VISIBLE_DEVICES=0,1
NNODES=2

Run xllm serve once. The launcher assigns ranks and per-rank ports, writes log/node_0.log, log/node_1.log, and so on, and stays in the foreground. Do not pass --node_rank or wrap the command in a per-device loop.

In another shell, wait for model loading to finish and query the rank-0 service (port 8010 in this example):

Terminal window
curl http://127.0.0.1:8010/health
curl http://127.0.0.1:8010/v1/models

The model list should include Qwen3-8B, as set by --model_id. Use that ID in requests to the OpenAI-Compatible APIs. If you change PORT, update the request URLs as well.

  • Replace XLLM_IMAGE with your available MLU image. Follow the image provider’s requirements for driver libraries and device access.
  • The launch example uses --block_size=16. Model-specific optimizations, including Context Parallelism below, require their own compatible model and topology.
Report issue