first commit

aws · Oct 6, 2023 · 7d11e7b · 7d11e7b
1 parent 8bb221d
commit 7d11e7b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/aws-cdk-lib/aws-eks/lib/cluster.ts b/packages/aws-cdk-lib/aws-eks/lib/cluster.ts
@@ -1800,10 +1800,16 @@ export class Cluster extends ClusterBase {
  * @param options options for creating a new nodegroup
  */
  public addNodegroupCapacity(id: string, options?: NodegroupOptions): Nodegroup {
- return new Nodegroup(this, `Nodegroup${id}`, {
+ const mng = new Nodegroup(this, `Nodegroup${id}`, {
  cluster: this,
  ...options,
  });
+
+ if (options?.instanceTypes?.some(instanceType => nodeTypeForInstanceType(instanceType) === NodeType.INFERENTIA)) {
+ this.addNeuronDevicePlugin();
+ }
+
+ return mng;
  }
 
  /**