diff --git a/.buildkite/pipeline-npu.yaml b/.buildkite/pipeline-npu.yaml index 177f5396f..1d43dad2b 100644 --- a/.buildkite/pipeline-npu.yaml +++ b/.buildkite/pipeline-npu.yaml @@ -2,7 +2,8 @@ steps: - label: ":buildkit: Build and Push NPU Test Image" key: image-build agents: - queue: "linux-aarch64-a2b3-1" + queue: "ascend-a3" + resource_class: "npu-2" plugins: - kubernetes: metadata: @@ -84,7 +85,8 @@ steps: key: upload-ready-pipeline if: build.branch != "main" && build.pull_request.labels includes "ready" agents: - queue: "linux-aarch64-a2b3-1" + queue: "ascend-a3" + resource_class: "npu-2" command: | buildkite-agent pipeline upload .buildkite/test-npu-ready.yaml \ No newline at end of file diff --git a/.buildkite/test-npu-ready.yaml b/.buildkite/test-npu-ready.yaml index d56424a77..88a295a48 100644 --- a/.buildkite/test-npu-ready.yaml +++ b/.buildkite/test-npu-ready.yaml @@ -2,7 +2,8 @@ steps: - label: "Simple Unit Test" depends_on: upload-ready-pipeline agents: - queue: "linux-aarch64-a2b3-1" + queue: "ascend-a3" + resource_class: "npu-2" image: "swr.cn-southwest-2.myhuaweicloud.com/modelfoundry/vllm-omni-ci-npu:${BUILDKITE_COMMIT}" plugins: - kubernetes: