Skip to content

use ascendai/cann image to build vnpu #81

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

Draft
DSFans2014 wants to merge 3 commits into
base: main
Choose a base branch
from
Draft

use ascendai/cann image to build vnpu #81

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
cef2e58
use cann image
DSFans2014 May 19, 2026
393a822
support x86
DSFans2014 May 19, 2026
e9438c0
update libvnpu
DSFans2014 May 22, 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
26 changes: 19 additions & 7 deletions .github/workflows/ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,15 @@ env:

jobs:
build-vnpu:
runs-on: ubuntu-22.04-arm
strategy:
matrix:
platform:
- { runner: "ubuntu-22.04-arm", arch: "arm64"}
- { runner: "ubuntu-22.04", arch: "amd64"}
runs-on: ${{ matrix.platform.runner }}
container:
image: ghcr.io/dsfans2014/ascend-device-plugin/ascend-lib:0.2
image: ascendai/cann:8.5.1

steps:
- uses: actions/checkout@v4
with:
Expand All @@ -37,11 +43,11 @@ jobs:
mkdir -p ./artifacts
cp libvnpu/target/release/limiter ./artifacts/
cp libvnpu/target/release/libvnpu.so ./artifacts/
echo "/hami-vnpu-core/libvnpu.so" > ./artifacts/ld.so.preload
echo "/usr/local/hami-vnpu-core-assets/libvnpu.so" > ./artifacts/ld.so.preload
- name: upload artifacts
uses: actions/upload-artifact@v4
with:
name: build-artifacts
name: build-artifacts-${{ matrix.platform.arch }}
if-no-files-found: error
path: ./artifacts/*

Expand All @@ -60,11 +66,17 @@ jobs:
id: branch-names
uses: tj-actions/branch-names@v8

- name: Download Artifacts
- name: Download ARM64 Artifacts
uses: actions/download-artifact@v4
with:
name: build-artifacts-arm64
path: ./lib/hami-vnpu-core/arm64/

- name: Download AMD64 Artifacts
uses: actions/download-artifact@v4
with:
name: build-artifacts
path: ./lib/hami-vnpu-core/
name: build-artifacts-amd64
path: ./lib/hami-vnpu-core/amd64/

- name: Docker Login
if: ${{ github.event_name == 'push' && (startsWith(github.ref, 'refs/tags/') || github.ref == 'refs/heads/main') }}
Expand Down
3 changes: 2 additions & 1 deletion Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,9 +19,10 @@ RUN go get huawei.com/npu-exporter/utils/logger@v0.0.0-00010101000000-0000000000
RUN make all

FROM $BASE_IMAGE
ARG TARGETARCH
ENV LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64/driver:/usr/local/Ascend/driver/lib64/common
COPY --from=build /build/ascend-device-plugin /usr/local/bin/ascend-device-plugin
COPY --from=build /build/lib/hami-vnpu-core/* /usr/local/hami-vnpu-core-assets/
COPY --from=build /build/lib/hami-vnpu-core/${TARGETARCH}/* /usr/local/hami-vnpu-core-assets/
RUN chmod +x /usr/local/hami-vnpu-core-assets/limiter

ENTRYPOINT ["ascend-device-plugin"]
2 changes: 1 addition & 1 deletion libvnpu
Open in desktop
Submodule libvnpu updated 13 files
+1 −1 crates/hook/Cargo.toml
+4 −58 crates/hook/src/hook.rs
+27 −0 crates/hook/src/lib.rs
+30 −0 crates/hook/src/signal_compat.rs
+57 −0 crates/limiter/src/config.rs
+25 −0 crates/limiter/src/lib.rs
+12 −14 crates/limiter/src/main.rs
+66 −55 crates/limiter/src/manager.rs
+0 −230 crates/limiter/src/shmem.rs
+55 −0 crates/limiter/src/shmem/futex.rs
+70 −0 crates/limiter/src/shmem/mod.rs
+86 −0 crates/limiter/src/shmem/setup.rs
+85 −133 crates/limiter/src/worker.rs
Loading