[Refactor] Adjustments to moe_comm_method selection process #3973
Workflow file for this run
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: 'image / Ubuntu / a3' | |
| # This is a docker build check and publish job: | |
| # 1. PR Triggered docker image build check | |
| # - is for image build check | |
| # - Enable on main/*-dev branch | |
| # - push: ${{ github.event_name != 'pull_request' }} ==> false | |
| # 2. branches push trigger image publish | |
| # - is for branch/dev/nightly image | |
| # - commits are merge into main/*-dev ==> vllm-ascend:main / vllm-ascend:*-dev | |
| # 3. tags push trigger image publish | |
| # - is for final release image | |
| # - Publish when tag with v* (pep440 version) ===> vllm-ascend:v1.2.3-a3|vllm-ascend:v1.2.3rc1-a3 | |
| on: | |
| pull_request: | |
| branches: | |
| - 'main' | |
| - '*-dev' | |
| paths: | |
| - '.github/workflows/image_a3_ubuntu.yml' | |
| - 'Dockerfile.a3' | |
| - 'vllm_ascend/**' | |
| - 'setup.py' | |
| - 'pyproject.toml' | |
| - 'requirements.txt' | |
| - 'cmake/**' | |
| - 'CMakeLists.txt' | |
| - 'csrc/**' | |
| types: [ labeled ] | |
| push: | |
| # Publish image when tagging, the Dockerfile in tag will be build as tag image | |
| branches: | |
| - 'main' | |
| - '*-dev' | |
| tags: | |
| - 'v*' | |
| paths: | |
| - '.github/workflows/image_a3_ubuntu.yml' | |
| - 'Dockerfile.a3' | |
| - 'vllm_ascend/**' | |
| - 'setup.py' | |
| - 'pyproject.toml' | |
| - 'requirements.txt' | |
| - 'cmake/**' | |
| - 'CMakeLists.txt' | |
| - 'csrc/**' | |
| # only cancel in-progress runs of the same workflow | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| build: | |
| name: vllm-ascend image build | |
| # Only arm64 build on openEuler arm64, only amd64 build on Ubuntu amd64 | |
| # Push event or PR with both 'ready' and 'ready-for-test' labels | |
| runs-on: ubuntu-latest | |
| if: ${{ github.event_name == 'push' || (contains(github.event.pull_request.labels.*.name, 'ready') && contains(github.event.pull_request.labels.*.name, 'ready-for-test')) }} | |
| steps: | |
| - uses: actions/checkout@v4 | |
| with: | |
| fetch-depth: 0 | |
| persist-credentials: false | |
| - name: Print | |
| run: | | |
| lscpu | |
| - name: Docker meta | |
| id: meta | |
| uses: docker/metadata-action@v5 | |
| with: | |
| # TODO(yikun): add more hub image and a note on release policy for container image | |
| images: | | |
| quay.io/ascend/vllm-ascend | |
| # Note for test case | |
| # https://github.com/marketplace/actions/docker-metadata-action#typeref | |
| # 1. branch job pulish per main/*-dev branch commits | |
| # 2. main and dev pull_request is build only, so the tag pr-N-a3 is fine | |
| # 3. only pep440 matched tag will be published: | |
| # - v0.7.1 --> v0.7.1-a3 | |
| # - pre/post/dev: v0.7.1rc1-a3/v0.7.1rc1-a3/v0.7.1rc1.dev1-a3/v0.7.1.post1-a3, no latest | |
| # which follow the rule from vLLM with prefix v | |
| # TODO(yikun): the post release might be considered as latest release | |
| tags: | | |
| type=ref,event=branch,suffix=-a3 | |
| type=ref,event=pr,suffix=-a3 | |
| type=pep440,pattern={{raw}},suffix=-a3 | |
| flavor: | |
| latest=false | |
| - name: Free up disk space | |
| uses: jlumbroso/free-disk-space@54081f138730dfa15788a46383842cd2f914a1be # v1.3.1 | |
| with: | |
| tool-cache: true | |
| docker-images: false | |
| - name: Build - Set up QEMU | |
| uses: docker/setup-qemu-action@v3 | |
| - name: Build - Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v3 | |
| - name: Publish - Login to Quay Container Registry | |
| if: ${{ github.event_name == 'push' && github.repository_owner == 'vllm-project' }} | |
| uses: docker/login-action@v3 | |
| with: | |
| registry: quay.io | |
| username: ${{ vars.QUAY_USERNAME }} | |
| password: ${{ secrets.QUAY_PASSWORD }} | |
| - name: Build and push a3 | |
| uses: docker/build-push-action@v6 | |
| with: | |
| platforms: >- | |
| ${{ | |
| github.event_name == 'push' && github.repository_owner == 'vllm-project' && | |
| 'linux/amd64,linux/arm64' || | |
| 'linux/amd64' | |
| }} | |
| # use the current repo path as the build context, ensure .git is contained | |
| context: . | |
| file: Dockerfile.a3 | |
| # only trigger when tag, branch/main push | |
| push: ${{ github.event_name == 'push' && github.repository_owner == 'vllm-project' }} | |
| labels: ${{ steps.meta.outputs.labels }} | |
| tags: ${{ steps.meta.outputs.tags }} | |
| build-args: | | |
| PIP_INDEX_URL=https://pypi.org/simple | |
| provenance: false | |