Test linux arm ci

Author: metab0t

.github/actions/setup_optimizers_linux/action.yml

@@ -16,6 +16,14 @@ inputs:
   CHECK_LICENSE: 
     description: "..."
     required: true
+  ARCH:
+    description: "..."
+    required: true
+    type: choice
+    default: "X64"
+    options:
+      - "X64"
+      - "ARM64"
 
 runs:
   using: "composite"
@@ -32,29 +40,51 @@ runs:
         cache-name: cache-installers-linux
       with:
         path: ~/installers
-        key: ${{ runner.os }}-build-${{ env.cache-name }}-${{ hashFiles('optimizer_version.toml') }}
+        key: ${{ runner.os }}--${{ runner.arch }}-build-${{ env.cache-name }}-${{ hashFiles('optimizer_version.toml') }}
         restore-keys: |
-          ${{ runner.os }}-build-${{ env.cache-name }}-
+          ${{ runner.os }}--${{ runner.arch }}-build-${{ env.cache-name }}-
 
-    - if: ${{ steps.cache-installers-linux.outputs.cache-hit != 'true' }}
+    - if: ${{ (steps.cache-installers-linux.outputs.cache-hit != 'true') && (inputs.ARCH == 'X64') }}
       shell: bash
-      name: Download Installers
+      name: Download X64 Installers
       run: |
         curl -L -o ~/installers/gurobi.tar.gz https://packages.gurobi.com/11.0/gurobi11.0.0_linux64.tar.gz
         curl -L -o ~/installers/copt.tar.gz https://pub.shanshu.ai/download/copt/7.2.2/linux64/CardinalOptimizer-7.2.2-lnx64.tar.gz
         curl -L -o ~/installers/mosek.tar.bz2 https://download.mosek.com/stable/10.2.0/mosektoolslinux64x86.tar.bz2
         curl -L -o ~/installers/idaes-solvers.tar.gz https://github.com/IDAES/idaes-ext/releases/download/3.4.2/idaes-solvers-ubuntu2204-x86_64.tar.gz
 
-    - name: Setup Gurobi Installation
+    - if: ${{ (steps.cache-installers-linux.outputs.cache-hit != 'true') && (inputs.ARCH == 'ARM64') }}
+      shell: bash
+      name: Download ARM64 Installers
+      run: |
+        curl -L -o ~/installers/gurobi.tar.gz https://packages.gurobi.com/11.0/gurobi11.0.0_armlinux64.tar.gz
+        curl -L -o ~/installers/copt.tar.gz https://pub.shanshu.ai/download/copt/7.2.2/linux64/CardinalOptimizer-7.2.2-aarch64_lnx.tar.gz
+        curl -L -o ~/installers/mosek.tar.bz2 https://download.mosek.com/stable/10.2.0/mosektoolslinuxaarch64.tar.bz2
+        curl -L -o ~/installers/idaes-solvers.tar.gz https://github.com/IDAES/idaes-ext/releases/download/3.4.2/idaes-lib-ubuntu2204-aarch64.tar.gz
+
+    - name: Setup Gurobi Installation Home
+      if: ${{ inputs.ARCH == 'X64' }}
       shell: bash
-      env:
-        GUROBI_WLS: ${{ inputs.GUROBI_WLS }}
       run: |
         tar xfz ~/installers/gurobi.tar.gz -C ~/
         ls ~/gurobi1100/linux64
         # set environment variables
         export GUROBI_HOME="${HOME}/gurobi1100/linux64"
         echo "GUROBI_HOME=${GUROBI_HOME}" >> $GITHUB_ENV
+    - name: Setup Gurobi Installation Home
+      if: ${{ inputs.ARCH == 'ARM64' }}
+      shell: bash
+      run: |
+        tar xfz ~/installers/gurobi.tar.gz -C ~/
+        ls ~/gurobi1100/armlinux64
+        # set environment variables
+        export GUROBI_HOME="${HOME}/gurobi1100/armlinux64"
+        echo "GUROBI_HOME=${GUROBI_HOME}" >> $GITHUB_ENV
+    - name: Setup Gurobi Installation
+      shell: bash
+      env:
+        GUROBI_WLS: ${{ inputs.GUROBI_WLS }}
+      run: |
         echo "PATH=${PATH}:${GUROBI_HOME}/bin" >> $GITHUB_ENV
         echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${GUROBI_HOME}/lib" >> $GITHUB_ENV
         echo $GUROBI_HOME
@@ -91,16 +121,29 @@ runs:
       run: |
         copt_cmd -c "quit"
 
-    - name: Setup MOSEK Installation
+    - name: Setup MOSEK Installation Env
+      if: ${{ inputs.ARCH == 'X64' }}
       shell: bash
-      env:
-        MOSEK_LICENSE: ${{ inputs.MOSEK_LICENSE }}
       run: |
         tar jxf ~/installers/mosek.tar.bz2 -C ~/
         ls ~/mosek
         # set environment variables
         export MOSEK_10_2_BINDIR="${HOME}/mosek/10.2/tools/platform/linux64x86/bin"
         echo "MOSEK_10_2_BINDIR=${MOSEK_10_2_BINDIR}" >> $GITHUB_ENV
+    - name: Setup MOSEK Installation Env
+      if: ${{ inputs.ARCH == 'ARM64' }}
+      shell: bash
+      run: |
+        tar jxf ~/installers/mosek.tar.bz2 -C ~/
+        ls ~/mosek
+        # set environment variables
+        export MOSEK_10_2_BINDIR="${HOME}/mosek/10.2/tools/platform/linuxaarch64/bin"
+        echo "MOSEK_10_2_BINDIR=${MOSEK_10_2_BINDIR}" >> $GITHUB_ENV
+    - name: Setup MOSEK Installation
+      shell: bash
+      env:
+        MOSEK_LICENSE: ${{ inputs.MOSEK_LICENSE }}
+      run: |
         echo "PATH=${PATH}:${MOSEK_10_2_BINDIR}" >> $GITHUB_ENV
         echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${MOSEK_10_2_BINDIR}" >> $GITHUB_ENV
         echo $MOSEK_10_2_BINDIR

.github/workflows/linux-build.yml

@@ -9,7 +9,7 @@ on:
 jobs:
 
   linux_build:
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
 
     # Only allow one build at a time otherwise we can run out of licenses
     concurrency:
@@ -18,6 +18,7 @@ jobs:
     strategy:
       fail-fast: true
       matrix:
+        os: [ubuntu-latest, ubuntu-24.04-arm64]
         # python-version: ["3.9", "3.10", "3.11", "3.12"]
         python-version: ["3.9", "3.12"]
 
@@ -38,6 +39,7 @@ jobs:
           MOSEK_LICENSE: ${{ secrets.MOSEK_LICENSE }}
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           CHECK_LICENSE: true
+          ARCH: ${{ runner.arch }}
 
       - name: Build
         run: |