Merge pull request #127 from OneideLuizSchneider/feature/new-ami-rele…

…ase-09-15-24-0b4a59eda543ee0caa14b778d978f13c11bdc913 New AMI version for EKS - Auto-PR
OneideLuizSchneider · Sep 15, 2024 · 64e4a2e · 64e4a2e
2 parents 0b4a59e + c479a82
commit 64e4a2e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/amazon-eks-ami/nodeadm/internal/kubelet/eni-max-pods.txt b/amazon-eks-ami/nodeadm/internal/kubelet/eni-max-pods.txt
@@ -567,6 +567,7 @@ p3dn.24xlarge 737
 p4d.24xlarge 737
 p4de.24xlarge 737
 p5.48xlarge 100
+p5e.48xlarge 100
 r3.2xlarge 58
 r3.4xlarge 234
 r3.8xlarge 234

diff --git a/amazon-eks-ami/templates/al2023/runtime/gpu/nvidia-kmod-load.service b/amazon-eks-ami/templates/al2023/runtime/gpu/nvidia-kmod-load.service
@@ -1,9 +1,9 @@
 [Unit]
 Description=Loading NVIDIA kernel modules
-# load modules after cloud-init
-After=cloud-final.service
-Requires=cloud-final.service
-Before=nvidia-persistenced.service
+# the script needs to use IMDS, so wait for the network to be up to avoid any flakiness
+After=network-online.target
+Wants=network-online.target
+Before=nvidia-fabricmanager.service nvidia-persistenced.service
 
 [Service]
 Type=oneshot

diff --git a/amazon-eks-ami/templates/shared/runtime/eni-max-pods.txt b/amazon-eks-ami/templates/shared/runtime/eni-max-pods.txt
@@ -567,6 +567,7 @@ p3dn.24xlarge 737
 p4d.24xlarge 737
 p4de.24xlarge 737
 p5.48xlarge 100
+p5e.48xlarge 100
 r3.2xlarge 58
 r3.4xlarge 234
 r3.8xlarge 234