Use ACI's Big Containers in SNP CI

.azure-pipelines-templates/deploy_aci.yml

@@ -61,12 +61,14 @@ jobs:
           python3 scripts/azure_deployment/arm_template.py deploy aci \
             --subscription-id $(CCF_AZURE_SUBSCRIPTION_ID) \
             --resource-group ccf-aci \
-            --region northeurope \
+            --region eastus \
             --aci-type dynamic-agent \
             --deployment-name ci-$(Build.BuildNumber) \
             --aci-image ccfmsrc.azurecr.io/ccf/ci:pr-`git rev-parse HEAD` \
             --managed-identity $(CCF_SNP_CI_MANAGED_IDENTITY_ID) \
             --ports 22 \
+            --memory-gb 64 \
+            --core-count 16 \
             --aci-setup-timeout 300 \
             --aci-private-key-b64 $(sshKey) \
             --out ~/aci_ips
@@ -80,6 +82,18 @@ jobs:
           ACR_REGISTRY_RESOURCE_NAME: ccfmsrc
           CCF_AZURE_SUBSCRIPTION_ID: $(CCF_AZURE_SUBSCRIPTION_ID)
 
+      - script: |
+          set -ex
+          set -o pipefail
+          mapfile -t IP_ADDR_LIST <<< $(cat ~/ipAddresses | awk '{print $2}')
+          ssh agent@${IP_ADDR_LIST[0]} -o "StrictHostKeyChecking=no" -o ConnectTimeout=100 '
+          nproc --all
+          lscpu
+          cat /proc/cpuinfo 
+          '
+        name: print_cpu_stats
+        displayName: "Print CPU stats"
+
       - script: |
           set -ex
           set -o pipefail

.snpcc_canary

@@ -3,6 +3,5 @@
    O     \     o      |    /     |
 /-xXx--//-----x=x--/-xXx--/---x-/--->>>--/
 ...
-/\/\d(-_-)b/\/\/
-----vmpl---
-
+/\/\d(-_-)b/\/\
+----vmpl----

scripts/azure_deployment/arm_aci.py

@@ -110,15 +110,17 @@ def make_dev_container_command(args):
     ]
 
 
-def make_dev_container(id, name, image, command, ports, with_volume):
+def make_dev_container(
+    id, name, image, command, ports, with_volume, memory_gb, core_count
+):
     t = {
         "name": f"{name}-{id}",
         "properties": {
             "image": image,
             "command": command,
             "ports": [{"protocol": "TCP", "port": p} for p in ports],
             "environmentVariables": [],
-            "resources": {"requests": {"memoryInGB": 16, "cpu": 4}},
+            "resources": {"requests": {"memoryInGB": memory_gb, "cpu": core_count}},
         },
     }
     if with_volume:
@@ -163,6 +165,18 @@ def parse_aci_args(parser: ArgumentParser) -> Namespace:
         action="append",
         default=[],
     )
+    parser.add_argument(
+        "--memory-gb",
+        help="How many GB of memory to request for the container",
+        type=int,
+        default=16,
+    )
+    parser.add_argument(
+        "--core-count",
+        help="How many CPU cores to request for the container",
+        type=int,
+        default=4,
+    )
 
     # SEV-SNP options
     parser.add_argument(
@@ -245,6 +259,8 @@ def make_aci_deployment(args: Namespace) -> Deployment:
                 command,
                 args.ports,
                 with_volume=True,
+                memory_gb=args.memory_gb,
+                core_count=args.core_count,
             )
         ]