Port test-gpu.yml to Open-Athena/ec2 runner

.github/workflows/test-gpu.yml

@@ -9,18 +9,21 @@ on:
   workflow_dispatch:
 
 permissions:
-  id-token: write # This is required for interacting with AWS via OIDC
-  contents: read # This is required for actions/checkout
+  id-token: write # Required for interacting with AWS via OIDC
+  contents: read # Required for actions/checkout (normally set by default, but must explicitly specify when defining a custom `permissions` block.
+
 jobs:
-  start-aws-runner:
-    uses: ./.github/workflows/aws-runner-start.yml
+  ec2:
+    uses: Open-Athena/ec2-gha/.github/workflows/runner.yml@v2
+    with:
+      ec2_instance_type: g6.xlarge
+      ec2_image_id: ami-0365bff494b18bf93 # Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.7 (Ubuntu 22.04) in oa-ci-dev
+      ec2_root_device_size: '+1' # AMI's reported VolumeSize (40GB) +1GB; leaves headroom for test I/O; instance runs out of disk otherwise
     secrets:
-      AWS_ROLE: ${{ secrets.AWS_ROLE }}
       GH_SA_TOKEN: ${{ secrets.GH_SA_TOKEN }}
-
   run-tests:
-    needs: start-aws-runner
-    runs-on: ${{ fromJSON(needs.start-aws-runner.outputs.instances) }}
+    needs: ec2
+    runs-on: ${{ needs.ec2.outputs.id }}
     steps:
       - name: Check out code
         uses: actions/checkout@v2
@@ -48,15 +51,3 @@ jobs:
 
       - name: Run pytest
         run: uv run --locked pytest -n 2 -v -m "not manual and cuda"
-
-  stop-aws-runner:
-    uses: ./.github/workflows/aws-runner-stop.yml
-    needs:
-      - start-aws-runner
-      - run-tests
-    if: ${{ always() }}
-    with:
-      instance_mapping: ${{ needs.start-aws-runner.outputs.mapping }}
-    secrets:
-      AWS_ROLE: ${{ secrets.AWS_ROLE }}
-      GH_SA_TOKEN: ${{ secrets.GH_SA_TOKEN }}

.secrets.baseline

@@ -133,5 +133,5 @@
       }
     ]
   },
-  "generated_at": "2025-06-30T17:27:31Z"
+  "generated_at": "2025-08-08T20:05:06Z"
 }