Cx Organization

.github/actions/gh_tests/action.yml

@@ -0,0 +1,80 @@
+name: GH Tests
+description: "Run quick or full tests with a single configurable action."
+inputs:
+  mode:
+    description: "Test mode: quick (only serial), medium (1 and 4 ranks), and full (1 to 4)."
+    required: false
+    default: "full"
+  sha:
+    description: "Commit SHA to checkout before running tests."
+    required: false
+    default: "${{ github.sha }}"
+  py-version:
+    description: "Python version for the test run."
+    required: false
+    default: "3.14"
+  mpi:
+    description: "MPI implementation to install."
+    required: false
+    default: "mpich"
+  install-options:
+    description: "Package install target/options for heat."
+    required: false
+    default: ".[dev]"
+  pytorch-version:
+    description: "Pinned torch/torchvision/torchaudio versions."
+    required: false
+    default: "torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0"
+runs:
+  using: "composite"
+  steps:
+    - name: Setup MPI
+      uses: mpi4py/setup-mpi@3969f247e8fceef153418744f9d9ee6fdaeda29f # v1.2.0
+      with:
+        mpi: ${{ inputs.mpi }}
+    - name: "Set up Python"
+      uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405 # v6.2.0
+      with:
+        python-version: "${{ inputs.py-version }}"
+    - name: Install uv and set the Python version
+      uses: astral-sh/setup-uv@08807647e7069bb48b6ef5acd8ec9567f424441b # v8.1.0
+      with:
+        enable-cache: true
+    # Installations
+    - name: Install heat
+      if: ${{ inputs.mode == 'full' || inputs.mode == 'medium' }}
+      shell: bash
+      env:
+        UV_SYSTEM_PYTHON: 1
+      run: |
+        uv pip install pytest flaky
+        uv pip install ${{ inputs.pytorch-version }} --extra-index-url https://download.pytorch.org/whl/cpu
+        uv pip install -e ${{ inputs.install-options }}
+    - name: Install quick test dependencies
+      if: ${{ inputs.mode == 'quick' }}
+      shell: bash
+      env:
+        UV_SYSTEM_PYTHON: 1
+      run: |
+        uv pip install pytest flaky
+        uv pip install .
+    # Running tests
+    - name: Run serial tests
+      shell: bash
+      run: |
+        mpirun -n 1 pytest -vv
+    - name: Run tests on 2 ranks
+      if: ${{ inputs.mode == 'full'}}
+      shell: bash
+      run: |
+        mpirun -n 2 pytest -vv
+    - name: Run tests on 3 ranks
+      if: ${{ inputs.mode == 'full'}}
+      shell: bash
+      run: |
+        mpirun -n 3 pytest -vv
+    - name: Run tests on 4 ranks
+      if: ${{ inputs.mode == 'full' || inputs.mode == 'medium' }}
+      shell: bash
+      run: |
+        mpirun -n 4 pytest -vv

.github/actions/trigger_codebase_benchmarks/action.yml

@@ -0,0 +1,45 @@
+name: Trigger Codebase Benchmarks
+description: "Trigger benchmarks on Codebase for performance testing and analysis."
+inputs:
+  branch:
+    description: "The branch to trigger benchmarks on. Defaults to 'main'."
+    required: false
+    default: "main"
+  sha:
+    description: "The commit SHA to trigger benchmarks on. Defaults to the current commit."
+    required: false
+    default: "${{ github.sha }}"
+  pr_number:
+    description: "The pull request number to trigger benchmarks on. Defaults to the current PR number."
+    required: false
+    default: ""
+  author:
+    description: "The author of the benchmarks. Defaults to the PR assignee or 'heat_team'."
+    required: false
+    default: "${{ github.event.pull_request.assignee.login || 'heat_team' }}"
+  trigger_token:
+    description: "The trigger token used to start the Codebase pipeline."
+    required: true
+runs:
+  using: "composite"
+  steps:
+    - name: Trigger benchmarks
+      shell: bash
+      env:
+        AUTHOR: ${{ inputs.author}}
+        BRANCH: ${{ inputs.branch }}
+        API_TOKEN: ${{ inputs.trigger_token }}
+        SHA: ${{ inputs.sha }}
+        PR_NUMBER: ${{ inputs.pr_number }}
+      run: |
+        SHORT_SHA=$(git rev-parse --short $SHA)
+        curl -s -X POST \
+          --fail-with-body \
+          -F "token=$API_TOKEN" \
+          -F "ref=pipeline/cb/base" \
+          -F "variables[SHA]=$SHA" \
+          -F "variables[SHORT_SHA]=${SHORT_SHA}" \
+          -F "variables[BRANCH]=$BRANCH" \
+          -F "variables[PR]=$PR_NUMBER" \
+          -F "variables[AUTHOR]=${AUTHOR}" \
+          https://codebase.helmholtz.cloud/api/v4/projects/20697/trigger/pipeline

.github/actions/trigger_codebase_tests/action.yml

@@ -0,0 +1,36 @@
+name: Trigger Codebase Tests
+description: "Trigger tests on Codebase for performance testing and analysis."
+inputs:
+  branch:
+    description: "The branch to trigger tests on. Defaults to 'main'."
+    required: false
+    default: "main"
+  sha:
+    description: "The commit SHA to trigger tests on. Defaults to the current commit."
+    required: false
+    default: "${{ github.sha }}"
+  pr_number:
+    description: "The pull request number to trigger tests on. Defaults to the current PR number."
+    required: false
+    default: ""
+  trigger_token:
+    description: "The trigger token used to start the Codebase pipeline."
+    required: true
+runs:
+  using: "composite"
+  steps:
+    - name: Trigger codebase tests
+      shell: bash
+      env:
+        API_TOKEN: ${{ inputs.trigger_token }}
+        BRANCH: ${{ inputs.branch }}
+        PR_NUMBER: ${{ inputs.pr_number }}
+      run: |
+        curl -s -X POST \
+          --fail-with-body \
+          -F "token=$API_TOKEN" \
+          -F "ref=pipeline/ci/base" \
+          -F "variables[BRANCH]=$BRANCH" \
+          -F "variables[PR]=$PR_NUMBER" \
+          -F "variables[SHA]=${{ inputs.sha }}" \
+          https://codebase.helmholtz.cloud/api/v4/projects/20697/trigger/pipeline

.github/workflows/array-api.yml

@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         python-version: ['3.11', '3.12', '3.13', '3.14']
-        mpi: [ 'openmpi' ]
+        mpi: [ 'mpich' ]
 
     steps:
     - name: Checkout