From fe6b4e9547f0322783d4fb39e2395752e4160d2d Mon Sep 17 00:00:00 2001 From: Davanum Srinivas Date: Thu, 6 Feb 2025 15:46:39 -0500 Subject: [PATCH] Enable NVIDIA Plugin automatically for AL2023 GPU AMIs Signed-off-by: Davanum Srinivas --- pkg/eks/tasks.go | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/pkg/eks/tasks.go b/pkg/eks/tasks.go index 831249ee22..edc0328251 100644 --- a/pkg/eks/tasks.go +++ b/pkg/eks/tasks.go @@ -339,19 +339,21 @@ func (c *ClusterProvider) ClusterTasksForNodeGroups(cfg *api.ClusterConfig, inst for _, ng := range cfg.NodeGroups { clusterRequiresNeuronDevicePlugin = clusterRequiresNeuronDevicePlugin || api.HasInstanceType(ng, instanceutils.IsNeuronInstanceType) - // Only AL2 requires the NVIDIA device plugin + // Only AL2/AL2023 requires the NVIDIA device plugin clusterRequiresNvidiaDevicePlugin = clusterRequiresNvidiaDevicePlugin || (api.HasInstanceType(ng, instanceutils.IsNvidiaInstanceType) && - ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2) + ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2 || + ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2023) efaEnabled = efaEnabled || api.IsEnabled(ng.EFAEnabled) } for _, ng := range cfg.ManagedNodeGroups { clusterRequiresNeuronDevicePlugin = clusterRequiresNeuronDevicePlugin || api.HasInstanceTypeManaged(ng, instanceutils.IsNeuronInstanceType) - // Only AL2 requires the NVIDIA device plugin + // Only AL2/AL2023 requires the NVIDIA device plugin clusterRequiresNvidiaDevicePlugin = clusterRequiresNvidiaDevicePlugin || (api.HasInstanceTypeManaged(ng, instanceutils.IsNvidiaInstanceType) && - ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2) + ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2 || + ng.GetAMIFamily() == api.NodeImageFamilyAmazonLinux2023) efaEnabled = efaEnabled || api.IsEnabled(ng.EFAEnabled) } if clusterRequiresNeuronDevicePlugin {