Skip to content

Commit

Permalink
change how gpu enable/disable works
Browse files Browse the repository at this point in the history
  • Loading branch information
yourbuddyconner committed Sep 19, 2024
1 parent ea6f54d commit 7638370
Showing 1 changed file with 7 additions and 9 deletions.
16 changes: 7 additions & 9 deletions .github/workflows/adhoc.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,6 @@ name: Execute ZKVM-Perf
on:
workflow_dispatch:
inputs:
enable_gpu:
description: 'Enable GPU usage'
required: true
type: boolean
default: true
ami_id:
description: 'AMI ID'
required: true
Expand Down Expand Up @@ -61,7 +56,10 @@ jobs:
run: |
echo "matrix=$(jq -c . << EOF
{
"instance_type": ["g6.16xlarge", "r7i.16xlarge"]
"instance_config": [
{"type": "g6.16xlarge", "gpu": true},
{"type": "r7i.16xlarge", "gpu": false}
]
}
EOF
)" >> $GITHUB_OUTPUT
Expand Down Expand Up @@ -91,13 +89,13 @@ jobs:
mode: start
github-token: ${{ secrets.GH_PAT }}
ec2-image-id: ${{ inputs.ami_id }}
ec2-instance-type: ${{ matrix.instance_type }}
ec2-instance-type: ${{ matrix.instance_config.type }}
subnet-id: ${{ secrets.AWS_SUBNET_ID }}
security-group-id: ${{ secrets.AWS_SG_ID }}

perf:
needs: start-runner
name: Run ZKVM-Perf on ${{ matrix.instance_type }}
name: Run ZKVM-Perf on ${{ matrix.instance_config.type }}
runs-on: ${{ needs.start-runner.outputs.label }}
env:
CARGO_NET_GIT_FETCH_WITH_CLI: "true"
Expand Down Expand Up @@ -129,7 +127,7 @@ jobs:
- name: Run Tests (docker)
run: |
docker run ${{ inputs.enable_gpu && '--gpus all' || '' }} --platform linux/amd64 \
docker run ${{ matrix.instance_config.gpu && '--gpus all' || '' }} --platform linux/amd64 \
-v /var/run/docker.sock:/var/run/docker.sock \
-v ./benchmarks:/usr/src/app/benchmarks \
-e RUST_BACKTRACE=full \
Expand Down

0 comments on commit 7638370

Please sign in to comment.