AWS returns the arch as arm64 so account for that too

clusterinthecloud · Aug 10, 2020 · 31ea2db · 31ea2db
1 parent 3b9e5fd
commit 31ea2db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/roles/slurm/files/citc_aws.py b/roles/slurm/files/citc_aws.py
@@ -75,7 +75,7 @@ def create_node_config(client, hostname: str, nodespace: Dict[str, str], ssh_key
     shape = features["shape"]
     if features["arch"] == "x86_64":
         arch = "x86_64"
-    elif features["arch"] == "aarch64":
+    elif features["arch"] in {"aarch64", "arm64"}:
         arch = "arm64"  # This is what AWS calls aarch64
     else:
         raise ValueError(f"'{shape}' architecture ({features['arch']}) not recognised")