Skip to content

Commit 1cbe135

Browse files
committed
Merge branch 'sgl-cmake2' of https://github.com/1329009851/sgl-kernel-npu into sgl-cmake2
* 'sgl-cmake2' of https://github.com/1329009851/sgl-kernel-npu: CI execution requirements for separating a2 and a3 (sgl-project#367) Fix the bug that total expert num greater than 256 or local expert num is less than 8 (sgl-project#364) adapt ant moving to A2 single machine (sgl-project#362) reset ci -- run test mixed running for experts on a2. (sgl-project#365) Revert "Build the deepep package with the chip model included. (sgl-project#274)" (sgl-project#363) fix:buffer control (sgl-project#361) Build the deepep package with the chip model included. (sgl-project#274) bugfix wrong packages build dir (sgl-project#360) bump version to 2026.02.01 (sgl-project#359) Cover the workflows cases on a3 (sgl-project#321) release follows naming convention (sgl-project#356) Modify notifydispatch to support DEEPEP_NORMAL_LONG_SEQ_ROUND up to 128. (sgl-project#352) fix the hanging bug (sgl-project#355) [Bugfix] Fix build script working with cann 8.5.0 (sgl-project#354) Modify the description of DeepEP in the README file. (sgl-project#348) Revert "Add scripts for building CMake files (sgl-project#344)" (sgl-project#353) Add scripts for building CMake files (sgl-project#344) Support x86_64 and aarch64 binary release (sgl-project#325) add function for deep-ep tests (sgl-project#301) [Doc] Improved README.md content and English grammar and integrated the DeepWiki badge for Ask AI (sgl-project#345)
2 parents 65f10e5 + f2f86d0 commit 1cbe135

39 files changed

+3250
-876
lines changed

.github/workflows/build_and_release.yml

Lines changed: 27 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -11,23 +11,30 @@ on:
1111
paths:
1212
- '.github/workflows/build_and_release.yml'
1313
- 'config.ini'
14+
- 'build.sh'
1415

1516
jobs:
1617

1718
build-and-release:
18-
strategy:
19-
matrix:
20-
os: [ubuntu-24.04-arm]
21-
hardware: [910b, a3]
22-
cann_version: [8.5.0]
23-
runs-on: ${{ matrix.os }}
24-
container:
25-
image: quay.io/ascend/cann:${{ matrix.cann_version }}-${{ matrix.hardware }}-ubuntu22.04-py3.11
26-
permissions:
27-
contents: write
2819
defaults:
2920
run:
3021
shell: bash
22+
permissions:
23+
contents: write
24+
strategy:
25+
matrix:
26+
arch: [x86_64, aarch64]
27+
npu_hardware: [910b, a3]
28+
cann_version: [8.3.rc2, 8.5.0]
29+
# Ref: https://docs.github.com/en/actions/reference/runners/github-hosted-runners#standard-github-hosted-runners-for-public-repositories
30+
include:
31+
- arch: x86_64
32+
runner: ubuntu-24.04
33+
- arch: aarch64
34+
runner: ubuntu-24.04-arm
35+
runs-on: ${{ matrix.runner }}
36+
container:
37+
image: quay.io/ascend/cann:${{ matrix.cann_version }}-${{ matrix.npu_hardware }}-ubuntu22.04-py3.11
3138

3239
steps:
3340
- name: Checkout repository
@@ -39,13 +46,11 @@ jobs:
3946
4047
- name: Install dependency
4148
run: |
42-
env
4349
bash scripts/npu_ci_install_dependency.sh
4450
4551
- name: Build SGL-Kernel-NPU
4652
run: |
4753
env -i PATH=${PATH} bash --login -c "
48-
env
4954
export LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/runtime/lib64/stub:${LD_LIBRARY_PATH} && \
5055
./build.sh -a kernels
5156
"
@@ -58,15 +63,15 @@ jobs:
5863
"
5964
6065
- name: Build DeepEP for 910b
61-
if: matrix.hardware == '910b'
66+
if: matrix.npu_hardware == '910b'
6267
run: |
6368
env -i PATH=${PATH} bash --login -c "
6469
export LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/runtime/lib64/stub:${LD_LIBRARY_PATH} && \
6570
./build.sh -a deepep2
6671
"
6772
6873
- name: Build DeepEP for a3
69-
if: matrix.hardware == 'a3'
74+
if: matrix.npu_hardware == 'a3'
7075
run: |
7176
env -i PATH=${PATH} bash --login -c "
7277
export LD_LIBRARY_PATH=${ASCEND_HOME_PATH}/runtime/lib64/stub:${LD_LIBRARY_PATH} && \
@@ -75,12 +80,18 @@ jobs:
7580
7681
- name: Pack up packages
7782
run: |
78-
zip -v "sgl-kernel-npu_${{ github.event.release.tag_name }}_${{ matrix.cann_version }}_${{ matrix.hardware }}.zip" output/*.whl
83+
export TORCH_DEVICE_BACKEND_AUTOLOAD=0
84+
TORCH_VERSION=$(python3 -c "import torch; print(torch.__version__)" | awk -F'+' '{print $1}')
85+
PYTHON_VERSION=$(python3 -c "import platform; print(platform.python_version())" | awk -F'.' '{print $1$2}')
86+
PACKAGE_NAME="sgl-kernel-npu-${{ github.event.release.tag_name }}-torch${TORCH_VERSION}-py${PYTHON_VERSION}-cann${{ matrix.cann_version }}-${{ matrix.npu_hardware }}-${{ matrix.arch }}.zip"
87+
cd output/
88+
zip -v ${PACKAGE_NAME} ./*.whl
89+
ls -al
7990
8091
- name: Upload Release Artifacts
8192
if: github.event_name != 'pull_request'
8293
uses: softprops/action-gh-release@v2
8394
env:
8495
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
8596
with:
86-
files: ./*.zip
97+
files: ./output/*.zip

0 commit comments

Comments
 (0)