feat(eks): add nodegroup gpu integ test

aws · Nov 22, 2024 · f696b9d · f696b9d
1 parent 7f82fd5
commit f696b9d
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/packages/@aws-cdk-testing/framework-integ/test/aws-eks/test/integ.eks-cluster.ts b/packages/@aws-cdk-testing/framework-integ/test/aws-eks/test/integ.eks-cluster.ts
@@ -75,6 +75,8 @@ class EksClusterStack extends Stack {
 
     this.assertNodeGroupCustomAmi();
 
+    this.assertNodeGroupGpu();
+
     this.assertSimpleManifest();
 
     this.assertManifestWithoutValidation();
@@ -277,6 +279,19 @@ class EksClusterStack extends Stack {
       nodeRole: this.cluster.defaultCapacity ? this.cluster.defaultCapacity.role : undefined,
     });
   }
+  private assertNodeGroupGpu() {
+    // add a GPU nodegroup
+    this.cluster.addNodegroupCapacity('extra-ng-gpu', {
+      instanceTypes: [
+        new ec2.InstanceType('p2.xlarge'),
+        new ec2.InstanceType('g5.xlarge'),
+        new ec2.InstanceType('g6e.xlarge'),
+      ],
+      minSize: 1,
+      // reusing the default capacity nodegroup instance role when available
+      nodeRole: this.cluster.defaultCapacity ? this.cluster.defaultCapacity.role : undefined,
+    });
+  }
   private assertSpotCapacity() {
     // spot instances (up to 10)
     this.cluster.addAutoScalingGroupCapacity('spot', {