RHOAIENG-28774: add arm64 image support in CI scripts and GitHub Actions workflows (red-hat-data-services#1330)

Author: jiridanek

.github/workflows/build-notebooks-TEMPLATE.yaml

@@ -31,7 +31,8 @@ name: Build & Publish Notebook Servers (TEMPLATE)
 
 jobs:
   build:
-    runs-on: ubuntu-24.04
+    # https://docs.github.com/en/actions/how-tos/using-github-hosted-runners/using-github-hosted-runners/about-github-hosted-runners#standard-github-hosted-runners-for-public-repositories
+    runs-on: ${{ inputs.platform == 'linux/arm64' && 'ubuntu-24.04-arm' || 'ubuntu-24.04' }}
     env:
       # Some pieces of code (image pulls, for example) in podman consult TMPDIR or default to /var/tmp
       TMPDIR: /home/runner/.local/share/containers/tmpdir
@@ -167,17 +168,34 @@ jobs:
         run: sudo apt-get -qq remove podman crun
 
       - uses: actions/cache@v4
+        # https://docs.github.com/en/actions/reference/variables-reference#default-environment-variables
+        # https://docs.github.com/en/actions/how-tos/writing-workflows/choosing-what-your-workflow-does/store-information-in-variables
         id: cached-linuxbrew
         with:
           path: /home/linuxbrew/.linuxbrew
-          key: linuxbrew
+          key: linuxbrew-${{ runner.os }}-${{ runner.arch }}
 
-      - name: Install podman
-        if: steps.cached-linuxbrew.outputs.cache-hit != 'true'
+      - name: Install podman (linux/amd64)
+        if: inputs.platform == 'linux/amd64' && steps.cached-linuxbrew.outputs.cache-hit != 'true'
         run: |
           /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
           /home/linuxbrew/.linuxbrew/bin/brew install podman
 
+      # Warning: Your CPU architecture (arm64) is not supported. We only support
+      # x86_64 CPU architectures. You will be unable to use binary packages (bottles).
+      #
+      # This is a Tier 2 configuration:
+      #  https://docs.brew.sh/Support-Tiers#tier-2
+      # Do not report any issues to Homebrew/* repositories!
+      # Read the above document instead before opening any issues or PRs.
+      - name: Install podman (linux/arm64)
+        if: inputs.platform == 'linux/arm64' && steps.cached-linuxbrew.outputs.cache-hit != 'true'
+        # Error: podman: no bottle available!
+        # If you're feeling brave, you can try to install from source with:
+        run: |
+          /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
+          /home/linuxbrew/.linuxbrew/bin/brew install --build-from-source podman
+
       - name: Add linuxbrew to PATH
         run: echo "/home/linuxbrew/.linuxbrew/bin/" >> $GITHUB_PATH
 

ci/cached-builds/gen_gha_matrix_jobs.py

@@ -21,8 +21,14 @@
 
 project_dir = pathlib.Path(__file__).parent.parent.parent.absolute()
 
+ARM64_COMPATIBLE = {
+    "codeserver-ubi9-python-3.11",
+    "codeserver-ubi9-python-3.12",
+}
+
 S390X_COMPATIBLE = {
     "runtime-minimal-ubi9-python-3.11",
+    "runtime-minimal-ubi9-python-3.12",
     # add more here
 }
 
@@ -57,6 +63,12 @@ class RhelImages(enum.Enum):
     INCLUDE_ONLY = "include-only"
 
 
+class Arm64Images(enum.Enum):
+    EXCLUDE = "exclude"
+    INCLUDE = "include"
+    ONLY = "only"
+
+
 class S390xImages(enum.Enum):
     EXCLUDE = "exclude"
     INCLUDE = "include"
@@ -82,6 +94,15 @@ def main() -> None:
         nargs="?",
         help="Whether to `include` rhel images or `exclude` them or `include-only` them",
     )
+    argparser.add_argument(
+        "--arm64-images",
+        type=Arm64Images,
+        choices=list(Arm64Images),
+        required=False,
+        default=Arm64Images.INCLUDE,
+        nargs="?",
+        help="Whether to include, exclude, or only include arm64 images",
+    )
     argparser.add_argument(
         "--s390x-images",
         type=S390xImages,
@@ -113,9 +134,12 @@ def main() -> None:
 
     targets_with_platform: list[tuple[str, str]] = []
     for target in targets:
-        if args.s390x_images != S390xImages.ONLY:
+        if args.s390x_images != S390xImages.ONLY or args.arm64_images != Arm64Images.ONLY:
             targets_with_platform.append((target, "linux/amd64"))
-        if args.s390x_images != S390xImages.EXCLUDE:
+        if args.arm64_images != Arm64Images.EXCLUDE and args.s390x_images != S390xImages.ONLY:
+            if target in ARM64_COMPATIBLE:
+                targets_with_platform.append((target, "linux/arm64"))
+        if args.s390x_images != S390xImages.EXCLUDE and args.arm64_images != Arm64Images.ONLY:
             # NOTE: hardcode the list of s390x-compatible Makefile targets in S390X_COMPATIBLE
             if target in S390X_COMPATIBLE:
                 targets_with_platform.append((target, "linux/s390x"))