Skip to content

Commit ec59922

Browse files
committed
feat: add npu pipeline
Add a test CI pipeline based on NPU Signed-off-by: yangzeyu <532183776@qq.com>
1 parent 141a107 commit ec59922

1 file changed

Lines changed: 46 additions & 0 deletions

File tree

.buildkite/pipeline-npu.yaml

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,46 @@
1+
# buildkite-agent pipeline upload .buildkite/pipeline-npu.yaml
2+
3+
# This pipeline is for running NPU tests on PRs .
4+
steps:
5+
- label: ":gear: npu test steps"
6+
key: npu-test-steps
7+
steps:
8+
- label: ":pytest: NPU tests"
9+
key: npu-tests
10+
agents:
11+
queue: "ascend-a3"
12+
resource_class: "npu-2"
13+
plugins:
14+
- kubernetes:
15+
podSpec:
16+
containers:
17+
- image: "quay.io/ascend/vime:vime-latest"
18+
volumeMounts:
19+
- name: devshm
20+
mountPath: /dev/shm
21+
- name: hf-cache
22+
mountPath: /root/.cache/huggingface
23+
env:
24+
- name: HF_HOME
25+
value: /root/.cache/huggingface
26+
- name: VIME_TEST_ENABLE_INFINITE_RUN
27+
value: "false"
28+
# Expand additional pod_env variables here as needed, e.g.:
29+
# - name: VIME_TEST_USE_DEEPEP
30+
# value: "0"
31+
volumes:
32+
- name: devshm
33+
emptyDir:
34+
medium: Memory
35+
- name: hf-cache
36+
hostPath:
37+
path: /mnt/hf-cache
38+
type: DirectoryOrCreate
39+
timeout_in_minutes: 60
40+
#
41+
command: |
42+
'PR="${BUILDKITE_PULL_REQUEST:-false}"'
43+
'[ "$PR" = "false" ] && PR="non-pr"'
44+
'export GITHUB_COMMIT_NAME="${BUILDKITE_COMMIT}_${PR}"'
45+
"pip install -e . --no-deps --break-system-packages"
46+
f"python print('test-npu')"

0 commit comments

Comments
 (0)