From c479a82442224af8add8bea79dc1a3ca1b012c02 Mon Sep 17 00:00:00 2001
From: github-actions <github-actions@github.com>
Date: Sun, 15 Sep 2024 01:21:21 +0000
Subject: [PATCH] Auto commit - CI build

---
 amazon-eks-ami/nodeadm/internal/kubelet/eni-max-pods.txt  | 1 +
 .../templates/al2023/runtime/gpu/nvidia-kmod-load.service | 8 ++++----
 amazon-eks-ami/templates/shared/runtime/eni-max-pods.txt  | 1 +
 3 files changed, 6 insertions(+), 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
index 4fabd347..b984d3d1 100644
--- 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
index 6659620a..d67806b6 100755
--- 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
index 4fabd347..b984d3d1 100644
--- 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