Skip to content

[DOC] Update the metax image readme #511

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.

Sign up for GitHub

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

Merged
zhzhcookie merged 2 commits into from
Apr 12, 2026
Merged

[DOC] Update the metax image readme #511

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
59cd7e0
[DOC] Update the metax image readme
zhzhcookie Apr 12, 2026
1fb5530
[DOC] Update the metax image readme
zhzhcookie Apr 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/hcu-delivery.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ jobs:
shell: bash
run: |
set -x
TRITON_VER="3.0"
TRITON_VER="3.1"
CHIP=""
export FLAGTREE_BACKEND=hcu
WHL_VER=${{ inputs.WHL_VER }}
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/metax-delivery.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ jobs:
shell: bash
run: |
set -x
TRITON_VER="3.1"
TRITON_VER="3.0"
CHIP=""
export FLAGTREE_BACKEND=metax
WHL_VER=${{ inputs.WHL_VER }}
Expand Down
32 changes: 15 additions & 17 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -139,23 +139,21 @@ python3 -m pip uninstall -y triton # Repeat the cmd until fully uninstalled
RES="--index-url=https://resource.flagos.net/repository/flagos-pypi-hosted/simple"
```

|Backend |Install command<br>(The version corresponds to the git tag)|Triton<br>ver.|Python<br>ver.|libc.so &<br>libstdc++.so|
|:---------|:---------|:---------|:---------|:---------|
|nvidia |python3 -m pip install flagtree===0.5.0 $RES |3.6|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3 -m pip install flagtree==0.5.0+3.5 $RES |3.5|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3 -m pip install flagtree==0.4.0+3.3 $RES |3.3|3.10<br>3.11<br>3.12|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|nvidia |python3 -m pip install flagtree==0.5.0+3.1 $RES |3.1|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|iluvatar |python3 -m pip install flagtree==0.5.1+iluvatar3.1 $RES |3.1|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|mthreads |python3 -m pip install flagtree==0.5.1+mthreads3.1 $RES |3.1|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|metax |python3 -m pip install flagtree==0.5.1+metax3.1 $RES |3.1|3.12|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|metax |python3 -m pip install flagtree==0.5.1+metax3.1 $RES |3.1|3.10|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|hcu |python3 -m pip install flagtree==0.5.0+hcu3.0 $RES |3.0|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|ascend |python3 -m pip install flagtree==0.5.0+ascend3.2 $RES |3.2|3.11|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|tsingmicro|python3 -m pip install flagtree==0.5.0+tsingmicro3.3 $RES |3.3|3.10|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|aipu |python3 -m pip install flagtree==0.5.0+aipu3.3 $RES |3.3|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|sunrise |python3 -m pip install flagtree==0.4.0+sunrise3.4 $RES |3.4|3.10|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|enflame<br>GCU300(S60) |python3 -m pip install flagtree==0.5.0+enflame.gcu300.3.5 $RES|3.3|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|enflame<br>GCU400(L600)|python3 -m pip install flagtree==0.5.0+enflame.gcu400.3.5 $RES|3.5|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|Backend |Install command<br>(The version corresponds to the git tag)|Triton<br>ver.|libc.so &<br>libstdc++.so|
|:---------|:---------|:---------|:---------|
|nvidia |python3.12 -m pip install flagtree===0.5.0 $RES |3.6|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3.12 -m pip install flagtree==0.5.0+3.5 $RES |3.5|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3.12 -m pip install flagtree==0.4.0+3.3 $RES |3.3|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|nvidia |python3.12 -m pip install flagtree==0.5.0+3.1 $RES |3.1|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|iluvatar |python3.10 -m pip install flagtree==0.5.1+iluvatar3.1 $RES |3.1|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|mthreads |python3.10 -m pip install flagtree==0.5.1+mthreads3.1 $RES |3.1|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|metax |python3.12 -m pip install flagtree==0.5.1+metax3.0 $RES |3.0|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|hcu |python3.10 -m pip install flagtree==0.5.1+hcu3.1 $RES |3.1|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|ascend |python3.11 -m pip install flagtree==0.5.0+ascend3.2 $RES |3.2|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|tsingmicro|python3.10 -m pip install flagtree==0.5.0+tsingmicro3.3 $RES |3.3|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|aipu |python3.10 -m pip install flagtree==0.5.0+aipu3.3 $RES |3.3|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|sunrise |python3.10 -m pip install flagtree==0.4.0+sunrise3.4 $RES |3.4|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|enflame |python3.12 -m pip install flagtree==0.5.0+enflame3.5 $RES |3.5|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|

Historical versions of flagtree can be found at https://resource.flagos.net/#browse/search/pypi/=assets.attributes.pypi.description%3Dflagtree

Expand Down
32 changes: 15 additions & 17 deletions README_cn.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -139,23 +139,21 @@ python3 -m pip uninstall -y triton # Repeat the cmd until fully uninstalled
RES="--index-url=https://resource.flagos.net/repository/flagos-pypi-hosted/simple"
```

|后端 |安装命令（版本号对应 git tag）|Triton<br>版本|Python<br>版本|libc.so & libstdc++.so|
|:---------|:---------|:---------|:---------|:---------|
|nvidia |python3 -m pip install flagtree===0.5.0 $RES |3.6|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3 -m pip install flagtree==0.5.0+3.5 $RES |3.5|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3 -m pip install flagtree==0.4.0+3.3 $RES |3.3|3.10<br>3.11<br>3.12|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|nvidia |python3 -m pip install flagtree==0.5.0+3.1 $RES |3.1|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|iluvatar |python3 -m pip install flagtree==0.5.1+iluvatar3.1 $RES |3.1|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|mthreads |python3 -m pip install flagtree==0.5.1+mthreads3.1 $RES |3.1|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|metax |python3 -m pip install flagtree==0.5.1+metax3.1 $RES |3.1|3.12|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|metax |python3 -m pip install flagtree==0.5.1+metax3.1 $RES |3.1|3.10|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|hcu |python3 -m pip install flagtree==0.5.0+hcu3.0 $RES |3.0|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|ascend |python3 -m pip install flagtree==0.5.0+ascend3.2 $RES |3.2|3.11|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|tsingmicro|python3 -m pip install flagtree==0.5.0+tsingmicro3.3 $RES |3.3|3.10|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|aipu |python3 -m pip install flagtree==0.5.0+aipu3.3 $RES |3.3|3.10|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|sunrise |python3 -m pip install flagtree==0.4.0+sunrise3.4 $RES |3.4|3.10|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|enflame<br>GCU300(S60) |python3 -m pip install flagtree==0.5.0+enflame.gcu300.3.5 $RES|3.3|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|enflame<br>GCU400(L600)|python3 -m pip install flagtree==0.5.0+enflame.gcu400.3.5 $RES|3.5|3.12|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|后端 |安装命令（版本号对应 git tag）|Triton<br>版本|libc.so & libstdc++.so|
|:---------|:---------|:---------|:---------|
|nvidia |python3.12 -m pip install flagtree===0.5.0 $RES |3.6|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3.12 -m pip install flagtree==0.5.0+3.5 $RES |3.5|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|nvidia |python3.12 -m pip install flagtree==0.4.0+3.3 $RES |3.3|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|nvidia |python3.12 -m pip install flagtree==0.5.0+3.1 $RES |3.1|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|iluvatar |python3.10 -m pip install flagtree==0.5.1+iluvatar3.1 $RES |3.1|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|mthreads |python3.10 -m pip install flagtree==0.5.1+mthreads3.1 $RES |3.1|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|metax |python3.12 -m pip install flagtree==0.5.1+metax3.0 $RES |3.0|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|hcu |python3.10 -m pip install flagtree==0.5.1+hcu3.1 $RES |3.1|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|ascend |python3.11 -m pip install flagtree==0.5.0+ascend3.2 $RES |3.2|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|tsingmicro|python3.10 -m pip install flagtree==0.5.0+tsingmicro3.3 $RES |3.3|GLIBC_2.30<br>GLIBCXX_3.4.28<br>CXXABI_1.3.12|
|aipu |python3.10 -m pip install flagtree==0.5.0+aipu3.3 $RES |3.3|GLIBC_2.35<br>GLIBCXX_3.4.30<br>CXXABI_1.3.13|
|sunrise |python3.10 -m pip install flagtree==0.4.0+sunrise3.4 $RES |3.4|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|
|enflame |python3.12 -m pip install flagtree==0.5.0+enflame3.5 $RES |3.5|GLIBC_2.39<br>GLIBCXX_3.4.33<br>CXXABI_1.3.15|

flagtree 历史版本可以在 https://resource.flagos.net/#browse/search/pypi/=assets.attributes.pypi.description%3Dflagtree 查询

Expand Down
22 changes: 11 additions & 11 deletions documents/install.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
[[中文版](./install_cn.md)|English]

# 💫 NVIDIA & AMD [nvidia](/third_party/nvidia/) & [amd](/third_party/amd/)
## 💫 NVIDIA & AMD [nvidia](/third_party/nvidia/) & [amd](/third_party/amd/)

- Based on Triton 3.1/3.2/3.3/3.4/3.5/3.6, x64

## 1. Environment for build and run
### 1. Environment for build and run

### 1.1 Use the preinstalled image (for Triton 3.6)
#### 1.1 Use the preinstalled image (for Triton 3.6)

If you use this preinstalled image, you do not need to perform the later step 1.x unless you need to install a version of FlagTree other than the one for Triton 3.6.
If your network connection is available, you also do not need to perform the later step 1.x, because dependencies will be fetched automatically during the build.
Expand All @@ -33,7 +33,7 @@ docker run -dit \
docker exec -it ${CONTAINER} /bin/bash
```

### 1.2 Manually download the LLVM
#### 1.2 Manually download the LLVM

```shell
cd ${YOUR_LLVM_DOWNLOAD_DIR}
Expand Down Expand Up @@ -71,7 +71,7 @@ export LLVM_INCLUDE_DIRS=$LLVM_SYSPATH/include
export LLVM_LIBRARY_DIR=$LLVM_SYSPATH/lib
```

### 1.3 Manually download the Triton dependencies
#### 1.3 Manually download the Triton dependencies

```shell
cd ${YOUR_CODE_DIR}/FlagTree
Expand Down Expand Up @@ -101,9 +101,9 @@ sh python/scripts/unpack_triton_build_deps.sh ./build-deps-triton_3.6.x-linux-x6
After executing the above script, the original ~/.triton directory will be renamed, and a new ~/.triton directory will be created to store the pre-downloaded packages.
Note that the script will prompt for manual confirmation during execution.

## 2. Installation Commands
### 2. Installation Commands

### 2.1 Source-free Installation
#### 2.1 Source-free Installation

```shell
# Note: First install PyTorch, then execute the following commands
Expand All @@ -118,7 +118,7 @@ python3 -m pip install flagtree===0.5.0 $RES
python3 -m pip show flagtree
```

### 2.2 Build from Source
#### 2.2 Build from Source

```shell
apt update; apt install zlib1g zlib1g-dev libxml2 libxml2-dev
Expand All @@ -137,13 +137,13 @@ MAX_JOBS=32 python3 -m pip install . --no-build-isolation -v
unset LLVM_SYSPATH LLVM_INCLUDE_DIRS LLVM_LIBRARY_DIR
```

## 3. Testing and validation
### 3. Testing and validation

Refer to [Tests of nvidia backend](https://github.com/flagos-ai/FlagTree/blob/triton_v3.6.x/.github/workflows/hopper-build-and-test.yml)

# Q&A
## Q&A

### Q: After installation, running the program reports: version GLIBC or GLIBCXX not found
#### Q: After installation, running the program reports: version GLIBC or GLIBCXX not found

A: Check which GLIBC / GLIBCXX versions are supported by libc.so.6 and libstdc++.so.6.0.30 in your environment:

Expand Down
18 changes: 9 additions & 9 deletions documents/install_aipu.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
[[中文版](./install_aipu_cn.md)|English]

# 💫 ARM China（安谋科技）[aipu](https://github.com/flagos-ai/FlagTree/tree/triton_v3.3.x/third_party/aipu/)
## 💫 ARM China（安谋科技）[aipu](https://github.com/flagos-ai/FlagTree/tree/triton_v3.3.x/third_party/aipu/)

- Based on Triton 3.3, x64/arm64

## 1. Build and run environment
### 1. Build and run environment

### 1.1 Use the preinstalled image (for the x64 CPU simulation environment)
#### 1.1 Use the preinstalled image (for the x64 CPU simulation environment)

If you use this preinstalled image, you do not need to perform the later step 1.x.
If your network connection is available, you also do not need to perform the later step 1.x, because dependencies will be fetched automatically during the build.
Expand All @@ -32,15 +32,15 @@ docker run -dit \
docker exec -it ${CONTAINER} /bin/bash
```

### 1.2 Manually download the FlagTree dependencies
#### 1.2 Manually download the FlagTree dependencies

```shell
mkdir -p ~/.flagtree/aipu; cd ~/.flagtree/aipu
wget https://baai-cp-web.ks3-cn-beijing.ksyuncs.com/trans/llvm-a66376b0-ubuntu-x64-clang16-lld16_v0.4.0.tar.gz
tar zxvf llvm-a66376b0-ubuntu-x64-clang16-lld16_v0.4.0.tar.gz
```

### 1.3 Manually download the Triton dependencies
#### 1.3 Manually download the Triton dependencies

```shell
cd ${YOUR_CODE_DIR}/FlagTree
Expand All @@ -52,9 +52,9 @@ sh python/scripts/unpack_triton_build_deps.sh ./build-deps-triton_3.3.x-linux-x6
After executing the above script, the original ~/.triton directory will be renamed, and a new ~/.triton directory will be created to store the pre-downloaded packages.
Note that the script will prompt for manual confirmation during execution.

## 2. Installation Commands
### 2. Installation Commands

### 2.1 Source-free Installation
#### 2.1 Source-free Installation

```shell
# Note: First install PyTorch, then execute the following commands
Expand All @@ -69,7 +69,7 @@ python3 -m pip install flagtree===0.5.0+aipu3.3 $RES
python3 -m pip show flagtree
```

### 2.2 Build from Source
#### 2.2 Build from Source

Before building, you need to execute `source ~/env_setup.sh`. The content of this script is as follows:

Expand All @@ -90,7 +90,7 @@ export FLAGTREE_BACKEND=aipu
MAX_JOBS=32 python3 -m pip install . --no-build-isolation -v
```

## 3. Testing and validation
### 3. Testing and validation

Before testing, you need to execute `source ~/env_setup.sh`. The content of this script is shown above.

Expand Down
18 changes: 9 additions & 9 deletions documents/install_aipu_cn.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
[中文版|[English](./install_aipu.md)]

# 💫 ARM China（安谋科技）[aipu](https://github.com/flagos-ai/FlagTree/tree/triton_v3.3.x/third_party/aipu/)
## 💫 ARM China（安谋科技）[aipu](https://github.com/flagos-ai/FlagTree/tree/triton_v3.3.x/third_party/aipu/)

- 对应的 Triton 版本为 3.3，基于 x64/arm64 平台

## 1. 构建及运行环境
### 1. 构建及运行环境

### 1.1 使用预装镜像（x64 cpu 模拟环境）
#### 1.1 使用预装镜像（x64 cpu 模拟环境）

使用该预装镜像，则不必执行后续步骤 1.x。
如果网络环境畅通，也不必执行后续步骤 1.x，依赖库会在构建时自动拉取。
Expand All @@ -32,15 +32,15 @@ docker run -dit \
docker exec -it ${CONTAINER} /bin/bash
```

### 1.2 手动下载 FlagTree 依赖库
#### 1.2 手动下载 FlagTree 依赖库

```shell
mkdir -p ~/.flagtree/aipu; cd ~/.flagtree/aipu
wget https://baai-cp-web.ks3-cn-beijing.ksyuncs.com/trans/llvm-a66376b0-ubuntu-x64-clang16-lld16_v0.4.0.tar.gz
tar zxvf llvm-a66376b0-ubuntu-x64-clang16-lld16_v0.4.0.tar.gz
```

### 1.3 手动下载 Triton 依赖库
#### 1.3 手动下载 Triton 依赖库

```shell
cd ${YOUR_CODE_DIR}/FlagTree
Expand All @@ -52,9 +52,9 @@ sh python/scripts/unpack_triton_build_deps.sh ./build-deps-triton_3.3.x-linux-x6
执行完上述脚本后，原有的 ~/.triton 目录将被重命名，新的 ~/.triton 目录会被创建并存放预下载包。
注意执行脚本过程中会提示手动确认。

## 2. 安装命令
### 2. 安装命令

### 2.1 免源码安装
#### 2.1 免源码安装

```shell
# Note: First install PyTorch, then execute the following commands
Expand All @@ -69,7 +69,7 @@ python3 -m pip install flagtree===0.5.0+aipu3.3 $RES
python3 -m pip show flagtree
```

### 2.2 从源码构建
#### 2.2 从源码构建

构建前需执行 `source ~/env_setup.sh`，该脚本内容如下：

Expand All @@ -90,7 +90,7 @@ export FLAGTREE_BACKEND=aipu
MAX_JOBS=32 python3 -m pip install . --no-build-isolation -v
```

## 3. 测试验证
### 3. 测试验证

测试前需执行 `source ~/env_setup.sh`，该脚本内容见上文。

Expand Down
20 changes: 10 additions & 10 deletions documents/install_ascend.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
[[中文版](./install_ascend_cn.md)|English]

# 💫 Huawei Ascend（华为昇腾）[ascend](https://github.com/flagos-ai/FlagTree/blob/triton_v3.2.x/third_party/ascend)
## 💫 Huawei Ascend（华为昇腾）[ascend](https://github.com/flagos-ai/FlagTree/blob/triton_v3.2.x/third_party/ascend)

- Based on Triton 3.2, aarch64
- Available for 910B/910C

## 1. Build and run environment
### 1. Build and run environment

### 1.1 Use the preinstalled image (910C)
#### 1.1 Use the preinstalled image (910C)

This preinstalled image is created by executing the later step 1.x based on [Dockerfile-ubuntu22.04-python3.11-ascend](/dockerfiles/Dockerfile-ubuntu22.04-python3.11-ascend) and installing FlagTree.
If you use this preinstalled image, you do not need to perform the later step 1.x for 910C, and for 910B you only need to perform step 1.2.
Expand Down Expand Up @@ -41,7 +41,7 @@ docker run -dit -u 0 --user=root \
docker exec -it ${CONTAINER} /bin/bash
```

### 1.2 Install CANN
#### 1.2 Install CANN

- Register an account at https://www.hiascend.com/developer/download/community/result?module=cann and download the corresponding `cann-toolkit` and `cann-ops` for your platform

Expand All @@ -57,15 +57,15 @@ chmod +x Ascend-cann-A3-ops_8.5.0_linux-aarch64.run
./Ascend-cann-A3-ops_8.5.0_linux-aarch64.run --install
```

### 1.3 Manually download the FlagTree dependencies
#### 1.3 Manually download the FlagTree dependencies

```shell
mkdir -p ~/.flagtree/ascend; cd ~/.flagtree/ascend
wget https://baai-cp-web.ks3-cn-beijing.ksyuncs.com/trans/llvm-a66376b0-ubuntu-aarch64-python311-compat_v0.3.0.tar.gz
tar zxvf llvm-a66376b0-ubuntu-aarch64-python311-compat_v0.3.0.tar.gz
```

### 1.4 Manually download the Triton dependencies
#### 1.4 Manually download the Triton dependencies

```shell
cd ${YOUR_CODE_DIR}/FlagTree
Expand All @@ -77,9 +77,9 @@ sh python/scripts/unpack_triton_build_deps.sh ./build-deps-triton_3.2.x-linux-aa
After executing the above script, the original ~/.triton directory will be renamed, and a new ~/.triton directory will be created to store the pre-downloaded packages.
Note that the script will prompt for manual confirmation during execution.

## 2. Installation Commands
### 2. Installation Commands

### 2.1 Source-free Installation
#### 2.1 Source-free Installation

```shell
# Note: First install PyTorch, then execute the following commands
Expand All @@ -94,7 +94,7 @@ python3 -m pip install flagtree===0.5.0+ascend3.2 $RES
python3 -m pip show flagtree
```

### 2.2 Build from Source
#### 2.2 Build from Source

```shell
apt update; apt install zlib1g zlib1g-dev libxml2 libxml2-dev
Expand All @@ -105,6 +105,6 @@ export FLAGTREE_BACKEND=ascend
MAX_JOBS=32 python3 -m pip install . --no-build-isolation -v
```

## 3. Testing and validation
### 3. Testing and validation

Refer to [Tests of ascend backend](https://github.com/flagos-ai/FlagTree/blob/triton_v3.2.x/.github/workflows/ascend-build-and-test.yml)
Loading
Loading