File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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')"
You can’t perform that action at this time.
0 commit comments