From d38677ae5b1904e4b4b3020138283f58c785d711 Mon Sep 17 00:00:00 2001
From: Hossein Zamanian <129794814+hossein-zamanian@users.noreply.github.com>
Date: Sun, 3 Dec 2023 11:51:29 -0500
Subject: [PATCH] chore(eks): Add missing ami types for node groups

---
 packages/aws-cdk-lib/aws-eks/lib/managed-nodegroup.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/aws-cdk-lib/aws-eks/lib/managed-nodegroup.ts b/packages/aws-cdk-lib/aws-eks/lib/managed-nodegroup.ts
index b59b941318323..4d6b29993832f 100644
--- a/packages/aws-cdk-lib/aws-eks/lib/managed-nodegroup.ts
+++ b/packages/aws-cdk-lib/aws-eks/lib/managed-nodegroup.ts
@@ -42,6 +42,10 @@ export enum NodegroupAmiType {
    * Bottlerocket(x86-64)
    */
   BOTTLEROCKET_X86_64 = 'BOTTLEROCKET_x86_64',
+  /**
+   * Bottlerocket Nvidia(x86-64)
+   */
+  BOTTLEROCKET_x86_64_NVIDIA = 'BOTTLEROCKET_x86_64_NVIDIA',
   /**
    * Windows Core 2019 (x86-64)
    */
@@ -550,7 +554,7 @@ const x8664AmiTypes: NodegroupAmiType[] = [NodegroupAmiType.AL2_X86_64, Nodegrou
 const windowsAmiTypes: NodegroupAmiType[] = [NodegroupAmiType.WINDOWS_CORE_2019_X86_64,
   NodegroupAmiType.WINDOWS_CORE_2022_X86_64, NodegroupAmiType.WINDOWS_FULL_2019_X86_64,
   NodegroupAmiType.WINDOWS_FULL_2022_X86_64];
-const gpuAmiTypes: NodegroupAmiType[] = [NodegroupAmiType.AL2_X86_64_GPU];
+const gpuAmiTypes: NodegroupAmiType[] = [NodegroupAmiType.AL2_X86_64_GPU, NodegroupAmiType.BOTTLEROCKET_x86_64_NVIDIA];
 
 /**
  * This function check if the instanceType is GPU instance.