Updated tf functions to actually work, removing tf plan

massdriver-cloud · May 7, 2024 · 2f08291 · 2f08291
1 parent 4152193
commit 2f08291
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/core-services/nvidia_gpu.tf b/core-services/nvidia_gpu.tf
@@ -1,5 +1,9 @@
+locals {
+  has_gpu = length(coalesce([for ng in var.node_groups : length(regexall("^p[0-9]\\..*", ng.instance_type)) > 1 ? "gpu" : ""])) > 0
+}
+
 resource "kubernetes_daemonset" "nvidia" {
-  count = length([for ng in var.node_groups : ng if can(regex("^[p0-9]\\..*", ng.instance_type))]) > 0 ? 1 : 0
+  count = local.has_gpu ? 1 : 0
   metadata {
     name      = "nvidia-device-plugin-daemonset"
     namespace = kubernetes_namespace_v1.md-core-services.metadata.0.name

diff --git a/core-services/tf.plan b/core-services/tf.plan
diff --git a/custom-resources/tf.plan b/custom-resources/tf.plan
diff --git a/src/main.tf b/src/main.tf
@@ -59,7 +59,7 @@ resource "aws_eks_node_group" "node_group" {
   }
 
   dynamic "taint" {
-    for_each = [for ng in var.node_groups : ng if can(regex("^p[0-9]\\..*", ng.instance_type))]
+    for_each = length(regexall("^p[0-9]\\..*", each.value.instance_type)) > 0 ? toset(["gpu"]) : toset([])
     content {
       key    = "sku"
       value  = "gpu"