diff --git a/.github/workflows/docker-bases.yml b/.github/workflows/docker-bases.yml index 31627c0fe1..1f24cec4de 100644 --- a/.github/workflows/docker-bases.yml +++ b/.github/workflows/docker-bases.yml @@ -24,18 +24,45 @@ jobs: ############## Basic gcc CPU ########################## ####################################################### deploy-cpu-bases: - name: "cpu-base" - runs-on: ubuntu-latest + name: cpu-${{ matrix.platform }} + runs-on: ${{ matrix.os }} env: DOCKER_BUILDKIT: "1" + REGISTRY_IMAGE: "devitocodes/bases:cpu-gcc" + + strategy: + fail-fast: false + + matrix: + name: [gcc-arm, gcc-amd64] + include: + - name: gcc-arm + platform: linux/arm + os: macos-latest + + - name: gcc-amd64 + platform: linux/amd64 + os: ubuntu-latest + steps: + - name: Prepare + run: | + platform=${{ matrix.platform }} + echo "PLATFORM_PAIR=${platform//\//-}" >> $GITHUB_ENV + - name: Checkout devito uses: actions/checkout@v4 - name: Check event name run: echo ${{ github.event_name }} + - name: Docker meta + id: meta + uses: docker/metadata-action@v5 + with: + images: ${{ env.REGISTRY_IMAGE }} + - name: Set up QEMU uses: docker/setup-qemu-action@v2 @@ -52,15 +79,74 @@ jobs: run: docker system prune -a -f - name: GCC image - uses: docker/build-push-action@v5 + id: build + uses: docker/build-push-action@v6 with: context: . file: './docker/Dockerfile.cpu' push: true target: 'gcc' build-args: 'arch=gcc' + platform: ${{ platform.platform }} tags: 'devitocodes/bases:cpu-gcc' + labels: ${{ steps.meta.outputs.labels }} + outputs: type=image,name=${{ env.REGISTRY_IMAGE }},push-by-digest=true,name-canonical=true,push=true + + - name: Export digest + run: | + mkdir -p /tmp/digests + digest="${{ steps.build.outputs.digest }}" + touch "/tmp/digests/${digest#sha256:}" + + - name: Upload digest + uses: actions/upload-artifact@v4 + with: + name: digests-${{ env.PLATFORM_PAIR }} + path: /tmp/digests/* + if-no-files-found: error + retention-days: 1 + + merge: + runs-on: ubuntu-latest + needs: + - deploy-cpu-bases + env: + DOCKER_BUILDKIT: "1" + REGISTRY_IMAGE: "devitocodes/bases:cpu-gcc" + + steps: + - name: Download digests + uses: actions/download-artifact@v4 + with: + path: /tmp/digests + pattern: digests-* + merge-multiple: true + + - name: Set up Docker Buildx + uses: docker/setup-buildx-action@v3 + + - name: Docker meta + id: meta + uses: docker/metadata-action@v5 + with: + images: ${{ env.REGISTRY_IMAGE }} + + - name: Login to Docker Hub + uses: docker/login-action@v3 + with: + username: ${{ secrets.DOCKER_USERNAME }} + password: ${{ secrets.DOCKER_PASSWORD }} + + - name: Create manifest list and push + working-directory: /tmp/digests + run: | + docker buildx imagetools create $(jq -cr '.tags | map("-t " + .) | join(" ")' <<< "$DOCKER_METADATA_OUTPUT_JSON") \ + $(printf '${{ env.REGISTRY_IMAGE }}@sha256:%s ' *) + + - name: Inspect image + run: | + docker buildx imagetools inspect ${{ env.REGISTRY_IMAGE }}:${{ steps.meta.outputs.version }} ####################################################### ############## Intel OneApi CPU ####################### diff --git a/setup.py b/setup.py index a9a33d98e4..692e40e0f5 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,17 @@ import versioneer import os -import pkg_resources +try: + import importlib.metadata as metadata + get_version = lambda x: metadata.version(x) + PkgNotFound = metadata.PackageNotFoundError + parse_version = lambda x: metadata.version(x) +except ImportError: + import pkg_resources + get_version = lambda x: pkg_resources.get_distribution(x).version + PkgNotFound = pkg_resources.DistributionNotFound + parse_version = lambda x: pkg_resources.parse_version(x) + from setuptools import setup, find_packages @@ -23,13 +33,13 @@ def numpy_compat(required): # Check if sympy is installed and enforce numpy version accordingly. # If sympy isn't installed, enforce sympy>=1.12.1 and numpy>=2.0 try: - sympy_version = pkg_resources.get_distribution("sympy").version - min_ver2 = pkg_resources.parse_version("1.12.1") - if pkg_resources.parse_version(sympy_version) < min_ver2: + sympy_version = get_version("sympy") + min_ver2 = parse_version("1.12.1") + if parse_version(sympy_version) < min_ver2: new_reqs.extend([f"numpy>{numpy_lb},<2.0", f"sympy=={sympy_version}"]) else: new_reqs.extend([f"numpy>=2.0,<{numpy_ub}", f"sympy=={sympy_version}"]) - except pkg_resources.DistributionNotFound: + except PkgNotFound: new_reqs.extend([f"sympy>=1.12.1,<{sympy_ub}", f"numpy>=2.0,<{numpy_ub}"]) return new_reqs