Add support for Ubuntu24

Author: hgreebe

CHANGELOG.md

@@ -3,6 +3,7 @@ CHANGELOG
 3.13.0
 ------
 **ENHANCEMENTS**
+- Add support Ubuntu2404
 
 **CHANGES**
 

cli/src/pcluster/config/cluster_config.py

@@ -32,9 +32,7 @@
     DefaultUserHomeType,
 )
 from pcluster.config.common import Imds as TopLevelImds
-from pcluster.config.common import (
-    Resource,
-)
+from pcluster.config.common import Resource
 from pcluster.constants import (
     CIDR_ALL_IPS,
     CW_ALARMS_ENABLED_DEFAULT,

cli/src/pcluster/constants.py

@@ -23,7 +23,17 @@
 
 SUPPORTED_SCHEDULERS = ["slurm", "awsbatch"]
 SCHEDULERS_SUPPORTING_IMDS_SECURED = ["slurm"]
-SUPPORTED_OSES = ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "rhel8", "rocky8", "rhel9", "rocky9"]
+SUPPORTED_OSES = [
+    "alinux2",
+    "alinux2023",
+    "ubuntu2004",
+    "ubuntu2204",
+    "ubuntu2404",
+    "rhel8",
+    "rocky8",
+    "rhel9",
+    "rocky9",
+]
 SUPPORTED_OSES_FOR_SCHEDULER = {"slurm": SUPPORTED_OSES, "awsbatch": ["alinux2", "alinux2023"]}
 UNSUPPORTED_OSES_FOR_MICRO_NANO = ["ubuntu2004", "ubuntu2204", "rhel8", "rocky8", "rhel9", "rocky9"]
 UNSUPPORTED_OSES_FOR_DCV = ["alinux2023"]

cli/src/pcluster/resources/imagebuilder/parallelcluster.yaml

@@ -90,6 +90,8 @@ phases:
                 OS='ubuntu2004'
               elif [ `echo "${!RELEASE}" | grep '^ubuntu\.22'` ]; then
                 OS='ubuntu2204'
+              elif [ `echo "${!RELEASE}" | grep '^ubuntu\.24'` ]; then
+                OS='ubuntu2404'
               elif [ `echo "${!RELEASE}" | grep '^rhel\.8'` ]; then
                 OS='rhel8'
               elif [ `echo "${!RELEASE}" | grep '^rocky\.8'` ]; then
@@ -175,7 +177,7 @@ phases:
                 # This component only supports aarch64 CPUs on Amazon Linux 2, Ubuntu2004, Ubuntu2204, RHEL8, Rocky8, RHEL9 and Rocky9
                 ARCH=$(uname -m)
                 if [[ `echo ${!ARCH}` == 'aarch64' ]]; then
-                  if [ `echo "${!RELEASE}" | grep -Ev '^(amzn\.2|amzn\.2023|ubuntu\.20\.04|ubuntu\.22\.04|rhel\.8|rocky\.8|rhel\.9|rocky\.9)'` ]; then
+                  if [ `echo "${!RELEASE}" | grep -Ev '^(amzn\.2|amzn\.2023|ubuntu\.20\.04|ubuntu\.22\.04|ubuntu\.24\.04|rhel\.8|rocky\.8|rhel\.9|rocky\.9)'` ]; then
                     echo "This component does not support '${!RELEASE}' on ARM64 CPUs. Failing build."
                     exit {{ FailExitCode }}
                   fi

cli/src/pcluster/resources/imagebuilder/parallelcluster_test.yaml

@@ -42,6 +42,8 @@ phases:
                 OS='ubuntu2004'
               elif [ `echo "${RELEASE}" | grep '^ubuntu\.22'` ]; then
                 OS='ubuntu2204'
+              elif [ `echo "${!RELEASE}" | grep '^ubuntu\.24'` ]; then
+                OS='ubuntu2404'
               elif [ `echo "${RELEASE}" | grep '^rhel\.8'` ]; then
                 OS='rhel8'
               elif [ `echo "${RELEASE}" | grep '^rocky\.8'` ]; then

cli/src/pcluster/resources/imagebuilder/parallelcluster_validate.yaml

@@ -42,6 +42,8 @@ phases:
                 OS='ubuntu2004'
               elif [ `echo "${RELEASE}" | grep '^ubuntu\.22'` ]; then
                 OS='ubuntu2204'
+              elif [ `echo "${!RELEASE}" | grep '^ubuntu\.24'` ]; then
+                OS='ubuntu2404'
               elif [ `echo "${RELEASE}" | grep '^rhel\.8'` ]; then
                 OS='rhel8'
               elif [ `echo "${RELEASE}" | grep '^rocky\.8'` ]; then

cli/src/pcluster/resources/imagebuilder/update_and_reboot.yaml

@@ -43,6 +43,8 @@ phases:
                 OS='ubuntu2004'
               elif [ `echo "${!RELEASE}" | grep '^ubuntu\.22'` ]; then
                 OS='ubuntu2204'
+              elif [ `echo "${!RELEASE}" | grep '^ubuntu\.24'` ]; then
+                OS='ubuntu2404'
               elif [ `echo "${!RELEASE}" | grep '^rhel\.8'` ]; then
                 OS='rhel8'
               elif [ `echo "${!RELEASE}" | grep '^rocky\.8'` ]; then
@@ -91,7 +93,7 @@ phases:
               # This component only supports aarch64 CPUs on Amazon Linux 2, Ubuntu2004, Ubuntu2204, RHEL8, Rocky8, RHEL9 and Rocky9
               ARCH=$(uname -m)
               if [[ `echo ${!ARCH}` == 'aarch64' ]]; then
-                if [ `echo "${!RELEASE}" | grep -Ev '^(amzn\.2|ubuntu\.20\.04|ubuntu\.22\.04|rhel\.8|rocky\.8|rhel\.9|rocky\.9)'` ]; then
+                if [ `echo "${!RELEASE}" | grep -Ev '^(amzn\.2|ubuntu\.20\.04|ubuntu\.22\.04|ubuntu\.24\.04|rhel\.8|rocky\.8|rhel\.9|rocky\.9)'` ]; then
                   echo "This component does not support '${!RELEASE}' on ARM64 CPUs. Failing build."
                   exit {{ FailExitCode }}
                 fi

cli/tests/pcluster/test_utils.py

@@ -88,11 +88,11 @@ def test_generate_random_prefix():
     [
         (
             "x86_64",
-            ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "rhel8", "rocky8", "rhel9", "rocky9"],
+            ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "ubuntu2404", "rhel8", "rocky8", "rhel9", "rocky9"],
         ),
         (
             "arm64",
-            ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "rhel8", "rocky8", "rhel9", "rocky9"],
+            ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "ubuntu2404", "rhel8", "rocky8", "rhel9", "rocky9"],
         ),
     ],
 )
@@ -108,7 +108,7 @@ def test_get_supported_os_for_architecture(architecture, supported_oses):
     [
         (
             "slurm",
-            ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "rhel8", "rocky8", "rhel9", "rocky9"],
+            ["alinux2", "alinux2023", "ubuntu2004", "ubuntu2204", "ubuntu2404", "rhel8", "rocky8", "rhel9", "rocky9"],
         ),
         ("awsbatch", ["alinux2", "alinux2023"]),
     ],