diff --git a/.github/workflows/cifar_benchmark.yaml b/.github/workflows/cifar_benchmark.yaml index 5888e4daf5..ab3525b27e 100644 --- a/.github/workflows/cifar_benchmark.yaml +++ b/.github/workflows/cifar_benchmark.yaml @@ -21,6 +21,7 @@ on: options: - "m6i.metal" - "u-6tb1.112xlarge" + - "hpc7a.96xlarge" num_samples: description: Number of samples to use default: "3" @@ -42,6 +43,12 @@ env: AGENT_TOOLSDIRECTORY: /opt/hostedtoolcache RUNNER_TOOL_CACHE: /opt/hostedtoolcache + +# We need to use other settings than the CI here to be able to launch benchmarks +# on AWS Ireland that has the hpc7a (which are the fastest machines atm). +# We might clash with TFHE-rs benchmarks on hpc7a since they launch benches on all +# updates of main. + # Jobs jobs: start-cifar-runner: @@ -61,17 +68,17 @@ jobs: with: aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }} aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }} - aws-region: ${{ secrets.AWS_REGION }} + aws-region: ${{ secrets.AWS_BENCH_REGION }} - name: Start CIFAR-10 8-bit runner id: start-cifar10-8bit-runner uses: machulav/ec2-github-runner@2c4d1dcf2c54673ed3bfd194c4b6919ed396a209 with: mode: start github-token: ${{ secrets.EC2_RUNNER_BOT_TOKEN }} - ec2-image-id: ${{ secrets.AWS_EC2_AMI }} + ec2-image-id: ${{ secrets.AWS_BENCH_EC2_AMI }} ec2-instance-type: ${{ github.event.inputs.instance_type }} - subnet-id: ${{ secrets.AWS_EC2_SUBNET_ID }} - security-group-id: ${{ secrets.AWS_EC2_SECURITY_GROUP_ID }} + subnet-id: ${{ secrets.AWS_BENCH_EC2_SUBNET_ID }} + security-group-id: ${{ secrets.AWS_BENCH_EC2_SECURITY_GROUP_ID }} aws-resource-tags: > [ {"Key": "Name", "Value": "cml-benchmark-cifar10"},