feat: Add AL2023 ARM64 NVIDIA variants (#3369)

ndbaker1 · web-flow · commit 715d42bf1467 · 2025-06-09T17:27:17.000-05:00
diff --git a/modules/_user_data/main.tf b/modules/_user_data/main.tf
@@ -34,6 +34,7 @@ locals {
     AL2023_ARM_64_STANDARD     = "al2023"
     AL2023_x86_64_NEURON       = "al2023"
     AL2023_x86_64_NVIDIA       = "al2023"
+    AL2023_ARM_64_NVIDIA       = "al2023"
   }
   # Try to use `ami_type` first, but fall back to current, default behavior
   # TODO - will be removed in v21.0
diff --git a/modules/eks-managed-node-group/main.tf b/modules/eks-managed-node-group/main.tf
@@ -367,6 +367,7 @@ locals {
     AL2023_ARM_64_STANDARD     = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/standard/recommended/release_version"
     AL2023_x86_64_NEURON       = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/neuron/recommended/release_version"
     AL2023_x86_64_NVIDIA       = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/nvidia/recommended/release_version"
+    AL2023_ARM_64_NVIDIA       = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/nvidia/recommended/release_version"
   }
 
   # The Windows SSM params currently do not have a release version, so we have to get the full output JSON blob and parse out the release version
diff --git a/modules/self-managed-node-group/main.tf b/modules/self-managed-node-group/main.tf
@@ -28,6 +28,7 @@ locals {
     AL2023_ARM_64_STANDARD     = "al2023"
     AL2023_x86_64_NEURON       = "al2023"
     AL2023_x86_64_NVIDIA       = "al2023"
+    AL2023_ARM_64_NVIDIA       = "al2023"
   }
 
   user_data_type = local.ami_type_to_user_data_type[var.ami_type]
@@ -51,6 +52,7 @@ locals {
     AL2023_ARM_64_STANDARD     = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/standard/recommended/image_id"
     AL2023_x86_64_NEURON       = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/neuron/recommended/image_id"
     AL2023_x86_64_NVIDIA       = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/nvidia/recommended/image_id"
+    AL2023_ARM_64_NVIDIA       = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/nvidia/recommended/image_id"
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@ locals {`
`34`	`34`	`AL2023_ARM_64_STANDARD = "al2023"`
`35`	`35`	`AL2023_x86_64_NEURON = "al2023"`
`36`	`36`	`AL2023_x86_64_NVIDIA = "al2023"`
	`37`	`+ AL2023_ARM_64_NVIDIA = "al2023"`
`37`	`38`	`}`
`38`	`39`	# Try to use `ami_type` first, but fall back to current, default behavior
`39`	`40`	`# TODO - will be removed in v21.0`
Original file line number	Diff line number	Diff line change
`@@ -367,6 +367,7 @@ locals {`
`367`	`367`	`AL2023_ARM_64_STANDARD = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/standard/recommended/release_version"`
`368`	`368`	`AL2023_x86_64_NEURON = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/neuron/recommended/release_version"`
`369`	`369`	`AL2023_x86_64_NVIDIA = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/nvidia/recommended/release_version"`
	`370`	`+ AL2023_ARM_64_NVIDIA = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/nvidia/recommended/release_version"`
`370`	`371`	`}`
`371`	`372`
`372`	`373`	`# The Windows SSM params currently do not have a release version, so we have to get the full output JSON blob and parse out the release version`
Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@ locals {`
`28`	`28`	`AL2023_ARM_64_STANDARD = "al2023"`
`29`	`29`	`AL2023_x86_64_NEURON = "al2023"`
`30`	`30`	`AL2023_x86_64_NVIDIA = "al2023"`
	`31`	`+ AL2023_ARM_64_NVIDIA = "al2023"`
`31`	`32`	`}`
`32`	`33`
`33`	`34`	`user_data_type = local.ami_type_to_user_data_type[var.ami_type]`
`@@ -51,6 +52,7 @@ locals {`
`51`	`52`	`AL2023_ARM_64_STANDARD = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/standard/recommended/image_id"`
`52`	`53`	`AL2023_x86_64_NEURON = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/neuron/recommended/image_id"`
`53`	`54`	`AL2023_x86_64_NVIDIA = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/x86_64/nvidia/recommended/image_id"`
	`55`	`+ AL2023_ARM_64_NVIDIA = "/aws/service/eks/optimized-ami/${local.ssm_cluster_version}/amazon-linux-2023/arm64/nvidia/recommended/image_id"`
`54`	`56`	`}`
`55`	`57`	`}`
`56`	`58`