chore: Update cifar workflow to be able to run on hpc7a

zama-ai · Feb 2, 2024 · aebed11 · aebed11
1 parent 060d851
commit aebed11
Showing 1 changed file with 11 additions and 4 deletions.
diff --git 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"},