[ssbuffer](fix) Remove fallback by name for certain kernels #1
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: Release Triton Ascend Images | |
| on: | |
| workflow_dispatch: | |
| push: | |
| branches: [main] | |
| paths: | |
| - 'docker/Dockerfile' | |
| - 'requirements.txt' | |
| - 'requirements_dev.txt' | |
| - '.github/workflows/build-docker-image.yml' | |
| pull_request: | |
| branches: [main] | |
| paths: | |
| - 'docker/Dockerfile' | |
| - 'requirements.txt' | |
| - 'requirements_dev.txt' | |
| - '.github/workflows/build-docker-image.yml' | |
| concurrency: | |
| group: build-docker-${{ github.ref }} | |
| cancel-in-progress: true | |
| permissions: | |
| contents: read | |
| actions: write | |
| jobs: | |
| build-images: | |
| name: Build ${{ matrix.cann_version }}-${{ matrix.chip_type }}-${{ matrix.os }}-py${{ matrix.python_version }}-${{ matrix.platforms }} | |
| # Use buildkit runner matching the target platform. | |
| runs-on: ${{ matrix.platforms == 'amd64' && 'linux-amd64-cpu-4' || 'linux-aarch64-cpu-4' }} | |
| container: | |
| image: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:9.0.0-a3-ubuntu22.04-py3.12 | |
| timeout-minutes: 360 | |
| strategy: | |
| fail-fast: false | |
| # Limit parallelism: too many concurrent builds overwhelm the shared | |
| # buildkitd workers and SWR (auth timeouts / connection resets). | |
| max-parallel: 6 | |
| matrix: | |
| cann_version: [8.5.0, 9.0.0-beta.2] | |
| chip_type: [910b, a3] | |
| os: [ubuntu22.04] | |
| python_version: ["3.10", "3.11"] | |
| platforms: [amd64, arm64] | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v4 | |
| with: | |
| symlinks: true | |
| - name: Login to Quay.io | |
| if: github.event_name == 'push' | |
| uses: docker/login-action@v3 | |
| with: | |
| registry: quay.io | |
| username: ${{ secrets.QUAY_USER }} | |
| password: ${{ secrets.QUAY_PASSWD }} | |
| - name: Generate tags | |
| id: tags | |
| run: | | |
| git config --global --add safe.directory "$GITHUB_WORKSPACE" | |
| CANN_TAG="${{ matrix.cann_version }}-${{ matrix.chip_type }}-${{ matrix.os }}-py${{ matrix.python_version }}" | |
| GIT_COMMIT_SHORT="$(git rev-parse --short HEAD)" | |
| TAG_COMMIT="${CANN_TAG}-${GIT_COMMIT_SHORT}" | |
| TAG_LATEST="${CANN_TAG}-latest" | |
| echo "cann_tag=${CANN_TAG}" >> $GITHUB_OUTPUT | |
| echo "tag_commit=${TAG_COMMIT}" >> $GITHUB_OUTPUT | |
| echo "tag_latest=${TAG_LATEST}" >> $GITHUB_OUTPUT | |
| - name: Build and push | |
| uses: docker/build-push-action@v7 | |
| with: | |
| context: . | |
| file: docker/Dockerfile | |
| push: ${{ github.event_name == 'push' }} | |
| tags: | | |
| quay.io/ascend/triton:${{ steps.tags.outputs.tag_commit }}-${{ matrix.platforms }} | |
| quay.io/ascend/triton:${{ steps.tags.outputs.tag_latest }}-${{ matrix.platforms }} | |
| build-args: | | |
| CANN_BASE_IMAGE=swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/cann:${{ steps.tags.outputs.cann_tag }} | |
| APTMIRROR=http://cache-service.nginx-pypi-cache.svc.cluster.local:8081 | |
| PIP_INDEX_URL=http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple | |
| PIP_TRUSTED_HOST=cache-service.nginx-pypi-cache.svc.cluster.local | |
| TORCH_INDEX_URL=http://cache-service.nginx-pypi-cache.svc.cluster.local/whl/cpu | |
| cache-from: type=registry,ref=swr.cn-southwest-2.myhuaweicloud.com/modelfoundry/triton-buildcache:${{ matrix.cann_version }}-${{ matrix.platforms }} | |
| cache-to: type=registry,ref=swr.cn-southwest-2.myhuaweicloud.com/modelfoundry/triton-buildcache:${{ matrix.cann_version }}-${{ matrix.platforms }},mode=max |