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
153 lines
3.9 KiB
Plaintext
153 lines
3.9 KiB
Plaintext
---
|
|
title: "MindSpore Models"
|
|
---
|
|
## Introduction
|
|
|
|
MindSpore is a high-performance AI framework optimized for Ascend NPUs. This doc guides users to run MindSpore models in SGLang.
|
|
|
|
## Requirements
|
|
|
|
MindSpore currently only supports Ascend NPU devices. Users need to first install Ascend CANN 8.5.
|
|
The CANN software packages can be downloaded from the [Ascend Official Website](https://www.hiascend.com).
|
|
|
|
## Supported Models
|
|
|
|
Currently, the following models are supported:
|
|
|
|
- **Qwen3**: Dense and MoE models
|
|
- **DeepSeek V3/R1**
|
|
- *More models coming soon...*
|
|
|
|
## Installation
|
|
|
|
> **Note**: Currently, MindSpore models are provided by an independent package `sgl-mindspore`. Support for MindSpore is built upon current SGLang support for Ascend NPU platform. Please first [install SGLang for Ascend NPU](../hardware-platforms/ascend-npus/ascend_npu) and then install `sgl-mindspore`:
|
|
|
|
```bash Install
|
|
git clone https://github.com/mindspore-lab/sgl-mindspore.git
|
|
cd sgl-mindspore
|
|
pip install -e .
|
|
```
|
|
|
|
|
|
## Run Model
|
|
|
|
Current SGLang-MindSpore supports Qwen3 and DeepSeek V3/R1 models. This doc uses Qwen3-8B as an example.
|
|
|
|
### Offline inference
|
|
|
|
Use the following script for offline inference:
|
|
|
|
```python Offline Infer
|
|
import sglang as sgl
|
|
|
|
# Initialize the engine with MindSpore backend
|
|
llm = sgl.Engine(
|
|
model_path="/path/to/your/model", # Local model path
|
|
device="npu", # Use NPU device
|
|
model_impl="mindspore", # MindSpore implementation
|
|
attention_backend="ascend", # Attention backend
|
|
tp_size=1, # Tensor parallelism size
|
|
dp_size=1 # Data parallelism size
|
|
)
|
|
|
|
# Generate text
|
|
prompts = [
|
|
"Hello, my name is",
|
|
"The capital of France is",
|
|
"The future of AI is"
|
|
]
|
|
|
|
sampling_params = {"temperature": 0, "top_p": 0.9}
|
|
outputs = llm.generate(prompts, sampling_params)
|
|
|
|
for prompt, output in zip(prompts, outputs):
|
|
print(f"Prompt: {prompt}")
|
|
print(f"Generated: {output['text']}")
|
|
print("---")
|
|
```
|
|
|
|
### Start server
|
|
|
|
Launch a server with MindSpore backend:
|
|
|
|
```bash Command
|
|
# Basic server startup
|
|
python3 -m sglang.launch_server \
|
|
--model-path /path/to/your/model \
|
|
--host 0.0.0.0 \
|
|
--device npu \
|
|
--model-impl mindspore \
|
|
--attention-backend ascend \
|
|
--tp-size 1 \
|
|
--dp-size 1
|
|
```
|
|
|
|
For distributed server with multiple nodes:
|
|
|
|
```bash Command
|
|
# Multi-node distributed server
|
|
python3 -m sglang.launch_server \
|
|
--model-path /path/to/your/model \
|
|
--host 0.0.0.0 \
|
|
--device npu \
|
|
--model-impl mindspore \
|
|
--attention-backend ascend \
|
|
--dist-init-addr 127.0.0.1:29500 \
|
|
--nnodes 2 \
|
|
--node-rank 0 \
|
|
--tp-size 4 \
|
|
--dp-size 2
|
|
```
|
|
|
|
## Troubleshooting
|
|
|
|
#### Debug Mode
|
|
|
|
Enable sglang debug logging by log-level argument.
|
|
|
|
```bash Debug Mode
|
|
python3 -m sglang.launch_server \
|
|
--model-path /path/to/your/model \
|
|
--host 0.0.0.0 \
|
|
--device npu \
|
|
--model-impl mindspore \
|
|
--attention-backend ascend \
|
|
--log-level DEBUG
|
|
```
|
|
|
|
Enable mindspore info and debug logging by setting environments.
|
|
|
|
```bash Command
|
|
export GLOG_v=1 # INFO
|
|
export GLOG_v=0 # DEBUG
|
|
```
|
|
|
|
#### Explicitly select devices
|
|
|
|
Use the following environment variable to explicitly select the devices to use.
|
|
|
|
```bash Command
|
|
export ASCEND_RT_VISIBLE_DEVICES=4,5,6,7 # to set device
|
|
```
|
|
|
|
#### Some communication environment issues
|
|
|
|
In case of some environment with special communication environment, users need set some environment variables.
|
|
|
|
```bash Disable LCCL
|
|
export MS_ENABLE_LCCL=off # current not support LCCL communication mode in SGLang-MindSpore
|
|
```
|
|
|
|
#### Some dependencies of protobuf
|
|
|
|
In case of some environment with special protobuf version, users need set some environment variables to avoid binary version mismatch.
|
|
|
|
```bash Command
|
|
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python # to avoid protobuf binary version mismatch
|
|
```
|
|
|
|
## Support
|
|
For MindSpore-specific issues:
|
|
|
|
- Refer to the [MindSpore documentation](https://www.mindspore.cn/)
|