test

Author: yhtang

.github/workflows/_runner_ondemand_slurm.yaml

@@ -44,7 +44,7 @@ jobs:
         id: meta
         shell: bash -x -e {0}
         run: |
-          JOB_NAME=${{ inputs.NAME }}-$(printf "%04x%04x" $RANDOM $RANDOM)
+          JOB_NAME=${{ inputs.NAME }}-$(printf "%8x" $((RANDOM<<16+RANDOM)))
           LOG_FILE=/nfs/cluster/${JOB_NAME}.log
           for var in JOB_NAME LOG_FILE; do
             echo "$var=${!var}" >> $GITHUB_OUTPUT
@@ -59,7 +59,7 @@ jobs:
             sbatch --parsable \
           <<"EOF"
           #!/bin/bash
-          #SBATCH --job-name=${{steps.meta.outputs.JOB_NAME }}
+          #SBATCH --job-name=${{ steps.meta.outputs.JOB_NAME }}
           #SBATCH --exclusive
           #SBATCH --nodes=1
           #SBATCH --tasks=1
@@ -85,7 +85,7 @@ jobs:
             --gpus all \
             --privileged \
             -v /runner \
-            -e RUNNER_NAME="${{ inputs.NAME }}" \
+            -e RUNNER_NAME="${{ steps.meta.outputs.JOB_NAME }}" \
             -e RUNNER_LABELS="${{ inputs.LABELS }}" \
             -e RUNNER_REPO="${{ github.repository }}" \
             -e RUNNER_TOKEN="${RUNNER_TOKEN}" \

.github/workflows/_sandbox.yaml

@@ -1,41 +1,32 @@
 name: "~Sandbox"
 
 on:
-  workflow_dispatch:
+  push:
 
 jobs:
-  sandbox:
-    runs-on: ubuntu-22.04
-    steps:
-      - name: Login to GitHub Container Registry
-        uses: docker/login-action@v3
-        with:
-          registry: ghcr.io
-          username: ${{ github.repository_owner }}
-          password: ${{ secrets.GITHUB_TOKEN }}
+  runner:
+    uses: ./.github/workflows/_runner_ondemand_slurm.yaml
+    with:
+      NAME: "TEST"
+      LABELS: "A100,${{ github.run_id }}"
+      TIME: "01:00:00"
+    secrets: inherit
 
-      - name: Print usage
-        run: |
-          cat << EOF
-          This is an empty workflow file located in the main branch of your
-          repository. It serves as a testing ground for new GitHub Actions on
-          development branches before merging them to the main branch. By
-          defining and overloading this workflow on your development branch,
-          you can test new actions without affecting your main branch, ensuring
-          a smooth integration process once the changes are ready to be merged.
+  jax-unit-test:
+    strategy:
+      fail-fast: false
+      matrix:
+        GPU_ARCH: [V100, A100]
+    # ensures A100 job lands on dedicated runner for this particular job
+    runs-on:
+      - self-hosted
+      - "${{ matrix.GPU_ARCH }}"
+      - "${{ matrix.GPU_ARCH == 'A100' && github.run_id || '' }}"
+    env:
+      BADGE_FILENAME_FULL: ${{ inputs.BADGE_FILENAME }}-${{ matrix.GPU_ARCH }}.json
+    steps:
+      - name: Print environment variables
+        run: env
 
-          Usage:
-          
-          1. In your development branch, modify the sandbox.yml workflow file
-             to include the new actions you want to test. Make sure to commit
-             the changes to the development branch.
-          2. Navigate to the 'Actions' tab in your repository, select the
-             '~Sandbox' workflow, and choose your development branch from the
-             branch dropdown menu. Click on 'Run workflow' to trigger the
-             workflow on your development branch.
-          3. Once you have tested and verified the new actions in the Sandbox
-             workflow, you can incorporate them into your main workflow(s) and
-             merge the development branch into the main branch. Remember to
-             revert the changes to the sandbox.yml file in the main branch to
-             keep it empty for future testing.
-          EOF
+      - name: Print GPU information
+        run: nvidia-smi