Remove services/ exclusion from pre-commit and fix linting issues (#362) #17
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: CI - Build CUDA Wheels | |
| on: | |
| push: | |
| tags: | |
| - "v*" | |
| permissions: | |
| contents: write | |
| jobs: | |
| build-wheels: | |
| runs-on: ${{ matrix.os }} | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| include: | |
| - os: ubuntu-24.04 | |
| arch: amd64 | |
| - os: ubuntu-24.04-arm | |
| arch: arm64 | |
| steps: | |
| - name: Checkout source | |
| uses: actions/checkout@v6 | |
| - name: Build wheel in Docker | |
| run: | | |
| cd kv_connectors/llmd_fs_backend | |
| if [[ "${{ matrix.arch }}" == "arm64" ]]; then | |
| PLATFORM="linux/arm64" | |
| else | |
| PLATFORM="linux/amd64" | |
| fi | |
| DOCKER_BUILDKIT=1 docker build \ | |
| --platform=${PLATFORM} \ | |
| --build-arg CUDA_VERSION=12.9.1 \ | |
| --build-arg torch_cuda_arch_list='8.7 8.9 9.0 10.0+PTX' \ | |
| --build-arg max_jobs=4 \ | |
| --tag llmd-fs-backend:build-${{ matrix.arch }} \ | |
| --target build \ | |
| --progress plain \ | |
| -f Dockerfile.wheel . | |
| mkdir -p wheel-release | |
| docker run --rm -v "$(pwd)/wheel-release:/artifacts_host" llmd-fs-backend:build-${{ matrix.arch }} \ | |
| bash -lc 'cp -v dist/*.whl /artifacts_host/ && chmod -R a+rw /artifacts_host' | |
| ls -lh wheel-release/ | |
| # ----------------------- | |
| # Upload artifacts | |
| # ----------------------- | |
| - name: Upload wheels as workflow artifact | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: cuda-wheels-${{ matrix.arch }} | |
| path: kv_connectors/llmd_fs_backend/wheel-release/*.whl | |
| attach-assets: | |
| needs: build-wheels | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Download wheel artifacts | |
| uses: actions/download-artifact@v4 | |
| with: | |
| path: release_wheels | |
| - name: Attach wheels to the tag release | |
| uses: softprops/action-gh-release@v2 | |
| with: | |
| tag_name: ${{ github.ref_name }} | |
| files: release_wheels/**/*.whl | |
| fail_on_unmatched_files: true | |
| draft: true | |
| prerelease: true |