Skip to content

ci: add npu on device#116

Open
fiskrt wants to merge 2 commits into
huawei-csl:mainfrom
fiskrt:feat-ci-on-npu
Open

ci: add npu on device#116
fiskrt wants to merge 2 commits into
huawei-csl:mainfrom
fiskrt:feat-ci-on-npu

Conversation

@fiskrt
Copy link
Copy Markdown
Collaborator

@fiskrt fiskrt commented Apr 21, 2026

Add a NPU CI

Compliments the current CI pipeline which runs on every commit (even outside of main branch and in PRs) with a NPU CI which only runs when a commit is pushed to branch main and takes approx 25 mins.

Runs inside of a docker image called pto-dsl-ci:{hash of dockerfile} so it caches image.

Once inside the docker, it runs:

python -m pip install --no-cache-dir -e .
python -c "import torch; print(torch.ones(2).npu())"
pytest -v
python examples/validate_all_examples.py

TODO:

  • add icon in readme that shows if NPU tests are green/red
  • Improve docker run settings

@fiskrt fiskrt requested a review from learning-chip April 21, 2026 14:41
@MirkoDeVita98
Copy link
Copy Markdown
Collaborator

nice, we can also setup a bot (e.g. renovate) that autobumps version of ptoas, commit of pto-isa in the CI yml file, creates a PR and runs tests. That to ensure w always have latest upated ptoas verified with npu tests/examples and we do not have to manually do it in each PR.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants