CI: pin Ubuntu 24.04 MPICH for arm64 wheel tests

Host arm64 mpich/libmpich12 4.2.0-5.1 debs on nrn-ci-deps ci-deps-v1 and
select amd64 vs arm64 in install_mpich_noble.sh so ubuntu-24.04-arm wheel
jobs no longer try to dpkg amd64 packages (LP#2072338 pin).

Author: nrnhines

ci/deps/MANIFEST.yml

@@ -49,6 +49,28 @@ assets:
       - ci/deps/install_mpich_noble.sh
     notes: Runtime library package for mpich-noble-4.2.0-5.1.
 
+  - id: mpich-noble-4.2.0-5.1-arm64
+    file: mpich_4.2.0-5.1_arm64.deb
+    sha256: c790342026ca5786bdad889af5b2c6f4c249bb07968aaf5435feb6b2771df452
+    upstream_url: https://launchpad.net/ubuntu/+source/mpich/4.2.0-5.1/+build/28285883/+files/mpich_4.2.0-5.1_arm64.deb
+    managed: true
+    consumers:
+      - .github/workflows/wheels-template.yml
+      - ci/deps/install_mpich_noble.sh
+    notes: >
+      arm64 counterpart of mpich-noble-4.2.0-5.1 (Launchpad build 28285883).
+      Pair with libmpich12-noble-4.2.0-5.1-arm64. Used by ubuntu-24.04-arm wheel tests.
+
+  - id: libmpich12-noble-4.2.0-5.1-arm64
+    file: libmpich12_4.2.0-5.1_arm64.deb
+    sha256: 4fdbdcdcf23a36f3f7fcd2f2fe2fdaff24629a107fa9f0c6904dd1c3aa662ae1
+    upstream_url: https://launchpad.net/ubuntu/+source/mpich/4.2.0-5.1/+build/28285883/+files/libmpich12_4.2.0-5.1_arm64.deb
+    managed: true
+    consumers:
+      - .github/workflows/wheels-template.yml
+      - ci/deps/install_mpich_noble.sh
+    notes: Runtime library package for mpich-noble-4.2.0-5.1-arm64.
+
   # --- P0/P1: static readline/ncurses sources (GNU mirrors flake) ---
   - id: ncurses-6.4-src
     file: ncurses-6.4.tar.gz

ci/deps/README.md

@@ -76,7 +76,7 @@ There is no automatic upgrade. `check-upstream.sh` is awareness only.
 
 Hosted on **nrn-ci-deps** release **`ci-deps-v1`**:
 
-- `mpich_4.2.0-5.1_amd64.deb` + `libmpich12_4.2.0-5.1_amd64.deb` — Ubuntu 24.04 wheel tests (LP#2072338)
+- `mpich_4.2.0-5.1_{amd64,arm64}.deb` + `libmpich12_4.2.0-5.1_{amd64,arm64}.deb` — Ubuntu 24.04 wheel tests (LP#2072338); `install_mpich_noble.sh` selects by `dpkg --print-architecture`
 - `ncurses-6.4.tar.gz`, `readline-8.3.tar.gz` — Mac static readline (`build_static_readline_osx.bash`)
 - `readline-7.0.tar.gz`, `ncurses-6.4.tar.gz` — manylinux wheel image (`packaging/python/Dockerfile`)
 - `automake-1.16.5.tar.xz` — Ubuntu MUSIC path in `neuron-ci.yml`

ci/deps/install_mpich_noble.sh

@@ -30,6 +30,26 @@ if [[ "$(id -u)" -ne 0 ]]; then
   exit 1
 fi
 
+arch="$(dpkg --print-architecture)"
+case "${arch}" in
+  amd64)
+    mpich_id="mpich-noble-4.2.0-5.1"
+    lib_id="libmpich12-noble-4.2.0-5.1"
+    mpich_deb="mpich_4.2.0-5.1_amd64.deb"
+    lib_deb="libmpich12_4.2.0-5.1_amd64.deb"
+    ;;
+  arm64)
+    mpich_id="mpich-noble-4.2.0-5.1-arm64"
+    lib_id="libmpich12-noble-4.2.0-5.1-arm64"
+    mpich_deb="mpich_4.2.0-5.1_arm64.deb"
+    lib_deb="libmpich12_4.2.0-5.1_arm64.deb"
+    ;;
+  *)
+    echo "error: unsupported architecture '${arch}' for pinned mpich (need amd64 or arm64)" >&2
+    exit 1
+    ;;
+esac
+
 WORKDIR="$(mktemp -d -t nrn-mpich-noble.XXXXXX)"
 cleanup() { rm -rf "${WORKDIR}"; }
 trap cleanup EXIT
@@ -39,15 +59,15 @@ trap cleanup EXIT
 # Override with NRN_CI_DEPS_SOURCE=local if debugging with a gitignored assets/ copy.
 export NRN_CI_DEPS_SOURCE="${NRN_CI_DEPS_SOURCE:-release}"
 
-"${SCRIPT_DIR}/fetch.sh" mpich-noble-4.2.0-5.1 "${WORKDIR}"
-"${SCRIPT_DIR}/fetch.sh" libmpich12-noble-4.2.0-5.1 "${WORKDIR}"
+"${SCRIPT_DIR}/fetch.sh" "${mpich_id}" "${WORKDIR}"
+"${SCRIPT_DIR}/fetch.sh" "${lib_id}" "${WORKDIR}"
 
 # Ensure base packages exist (headers etc.) then overwrite with pinned debs.
 apt-get install -y -qq mpich libmpich-dev || true
 dpkg --install \
-  "${WORKDIR}/libmpich12_4.2.0-5.1_amd64.deb" \
-  "${WORKDIR}/mpich_4.2.0-5.1_amd64.deb"
+  "${WORKDIR}/${lib_deb}" \
+  "${WORKDIR}/${mpich_deb}"
 
-echo "installed pinned mpich 4.2.0-5.1 for Ubuntu 24.04"
+echo "installed pinned mpich 4.2.0-5.1 for Ubuntu 24.04 (${arch})"
 mpichversion 2>/dev/null || true
 mpirun.mpich --version 2>/dev/null || mpirun --version 2>/dev/null || true

docs/install/ci_deps.md

@@ -39,12 +39,16 @@ ci/deps/fetch.sh --list
 
 # Download a managed asset from nrn-ci-deps (default for install helpers)
 NRN_CI_DEPS_SOURCE=release ci/deps/fetch.sh mpich-noble-4.2.0-5.1 /tmp/out
+# arm64 pin (ubuntu-24.04-arm wheel tests):
+# NRN_CI_DEPS_SOURCE=release ci/deps/fetch.sh mpich-noble-4.2.0-5.1-arm64 /tmp/out
 
 # Ubuntu 24.04 wheel tests: install pinned MPICH from the Release
+# (selects amd64 or arm64 debs via dpkg --print-architecture)
 ci/deps/install_mpich_noble.sh
 
 # Compare pins to upstream (awareness only; does not auto-upgrade)
 ci/deps/check-upstream.sh mpich-noble-4.2.0-5.1
+# ci/deps/check-upstream.sh mpich-noble-4.2.0-5.1-arm64
 ```
 
 Promote a new pin (files stay local/transient; then upload to **nrn-ci-deps**):
@@ -60,7 +64,8 @@ rm -f ci/deps/assets/*   # optional; assets/ is gitignored
 
 On **nrn-ci-deps** release **`ci-deps-v1`**:
 
-- Ubuntu 24.04 MPICH packages for wheel tests (LP#2072338)
+- Ubuntu 24.04 MPICH packages for wheel tests (LP#2072338), **amd64 and arm64**
+
 - GNU **ncurses** / **readline** sources for Mac static readline and the manylinux Dockerfile
 - **automake** 1.16.5 for the Ubuntu MUSIC job in `neuron-ci`