feat: Add support for specifying elastic inference accelerator (#1176)

terraform-aws-modules · Apr 19, 2021 · 26d40d2 · 26d40d2
1 parent 2e1651d
commit 26d40d2
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/examples/launch_templates/main.tf b/examples/launch_templates/main.tf
@@ -60,5 +60,12 @@ module "eks" {
       asg_desired_capacity = 1
       public_ip            = true
     },
+    {
+      name                          = "worker-group-3"
+      instance_type                 = "t2.large"
+      asg_desired_capacity          = 1
+      public_ip                     = true
+      elastic_inference_accelerator = "eia2.medium"
+    },
   ]
 }
diff --git a/local.tf b/local.tf
@@ -94,6 +94,7 @@ locals {
     spot_instance_pools                      = 10                                                   # "Number of Spot pools per availability zone to allocate capacity. EC2 Auto Scaling selects the cheapest Spot pools and evenly allocates Spot capacity across the number of Spot pools that you specify."
     spot_max_price                           = ""                                                   # Maximum price per unit hour that the user is willing to pay for the Spot instances. Default is the on-demand price
     max_instance_lifetime                    = 0                                                    # Maximum number of seconds instances can run in the ASG. 0 is unlimited.
+    elastic_inference_accelerator            = null                                                 # Type of elastic inference accelerator to be attached. Example values are eia1.medium, eia2.large, etc.
   }
 
   workers_group_defaults = merge(

diff --git a/workers_launch_template.tf b/workers_launch_template.tf
@@ -290,6 +290,18 @@ resource "aws_launch_template" "workers_launch_template" {
     "instance_type",
     local.workers_group_defaults["instance_type"],
   )
+
+  dynamic "elastic_inference_accelerator" {
+    for_each = lookup(
+      var.worker_groups_launch_template[count.index],
+      "elastic_inference_accelerator",
+      local.workers_group_defaults["elastic_inference_accelerator"]
+    ) != null ? [lookup(var.worker_groups_launch_template[count.index], "elastic_inference_accelerator", local.workers_group_defaults["elastic_inference_accelerator"])] : []
+    content {
+      type = elastic_inference_accelerator.value
+    }
+  }
+
   key_name = lookup(
     var.worker_groups_launch_template[count.index],
     "key_name",