tests.yaml

Author: ryan-williams

.github/workflows/test.yaml

@@ -1,9 +1,5 @@
 name: GPU tests
 on:
-  push:
-    branches: [main]
-  pull_request:
-    branches: [main]
   workflow_dispatch:
     inputs:
       instance_type:
@@ -18,6 +14,12 @@ on:
           - g6.xlarge    #  4 vCPUs, 16GB RAM,   L4 GPU
           - g6.2xlarge   #  8 vCPUs, 32GB RAM,   L4 GPU
           - g6.4xlarge   # 16 vCPUs, 64GB RAM,   L4 GPU
+  workflow_call:
+    inputs:
+      instance_type:
+        description: 'EC2 instance type'
+        required: true
+        type: string
 permissions:
   id-token: write
   contents: read

.github/workflows/tests.yaml

@@ -0,0 +1,70 @@
+name: GPU tests on multiple instance types
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+  workflow_dispatch:
+
+permissions:
+  id-token: write
+  contents: read
+
+jobs:
+  test-g5:
+    name: Test on g5.2xlarge (A10G)
+    uses: ./.github/workflows/test.yaml
+    with:
+      instance_type: g5.2xlarge
+    secrets: inherit
+
+  test-g6:
+    name: Test on g6.2xlarge (L4)
+    uses: ./.github/workflows/test.yaml
+    with:
+      instance_type: g6.2xlarge
+    secrets: inherit
+
+  summary:
+    name: Test Summary
+    needs: [test-g5, test-g6]
+    runs-on: ubuntu-latest
+    if: always()
+    steps:
+      - name: Summary
+        run: |
+          echo "## Test Results Summary" >> $GITHUB_STEP_SUMMARY
+          echo "" >> $GITHUB_STEP_SUMMARY
+          
+          # Check g5.2xlarge result
+          if [ "${{ needs.test-g5.result }}" == "success" ]; then
+            echo "✅ **g5.2xlarge (A10G)**: Tests passed" >> $GITHUB_STEP_SUMMARY
+          elif [ "${{ needs.test-g5.result }}" == "failure" ]; then
+            echo "❌ **g5.2xlarge (A10G)**: Tests failed" >> $GITHUB_STEP_SUMMARY
+          elif [ "${{ needs.test-g5.result }}" == "cancelled" ]; then
+            echo "⚠️ **g5.2xlarge (A10G)**: Tests cancelled" >> $GITHUB_STEP_SUMMARY
+          else
+            echo "⏭️ **g5.2xlarge (A10G)**: Tests skipped" >> $GITHUB_STEP_SUMMARY
+          fi
+          
+          # Check g6.2xlarge result
+          if [ "${{ needs.test-g6.result }}" == "success" ]; then
+            echo "✅ **g6.2xlarge (L4)**: Tests passed" >> $GITHUB_STEP_SUMMARY
+          elif [ "${{ needs.test-g6.result }}" == "failure" ]; then
+            echo "❌ **g6.2xlarge (L4)**: Tests failed" >> $GITHUB_STEP_SUMMARY
+          elif [ "${{ needs.test-g6.result }}" == "cancelled" ]; then
+            echo "⚠️ **g6.2xlarge (L4)**: Tests cancelled" >> $GITHUB_STEP_SUMMARY
+          else
+            echo "⏭️ **g6.2xlarge (L4)**: Tests skipped" >> $GITHUB_STEP_SUMMARY
+          fi
+          
+          echo "" >> $GITHUB_STEP_SUMMARY
+          echo "### GPU Details" >> $GITHUB_STEP_SUMMARY
+          echo "- **g5.2xlarge**: NVIDIA A10G (Ampere, compute capability 8.6)" >> $GITHUB_STEP_SUMMARY
+          echo "- **g6.2xlarge**: NVIDIA L4 (Ada Lovelace, compute capability 8.9)" >> $GITHUB_STEP_SUMMARY
+          
+          # Set overall status
+          if [ "${{ needs.test-g5.result }}" == "failure" ] || [ "${{ needs.test-g6.result }}" == "failure" ]; then
+            echo "::error::One or more test jobs failed"
+            exit 1
+          fi