From 987f978bc855e5947f7e192466e5e37a3725f68b Mon Sep 17 00:00:00 2001 From: Davanum Srinivas Date: Thu, 6 Feb 2025 16:26:05 -0500 Subject: [PATCH] Enable NVIDIA Plugin automatically for AL2023 GPU AMIs (#8199) 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 {