Making MPI operator optional for users

awslabs · Nov 6, 2023 · 0a46f3a · 0a46f3a
1 parent 1457d05
commit 0a46f3a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/ai-ml/trainium-inferentia/addons.tf b/ai-ml/trainium-inferentia/addons.tf
@@ -512,7 +512,8 @@ data "kubectl_file_documents" "mpi_operator_yaml" {
 }
 
 resource "kubectl_manifest" "mpi_operator" {
-  for_each   = data.kubectl_file_documents.mpi_operator_yaml.manifests
+  for_each   = var.enable_mpi_operator ? data.kubectl_file_documents.mpi_operator_yaml.manifests : {}
   yaml_body  = each.value
   depends_on = [module.eks.eks_cluster_id]
-}
+}
+
diff --git a/ai-ml/trainium-inferentia/variables.tf b/ai-ml/trainium-inferentia/variables.tf
@@ -41,4 +41,10 @@ variable "enable_amazon_prometheus" {
 variable "mpi_operator_version" {
   description = "The version of the MPI Operator to install"
   default     = "v0.4.0"
-}
+}
+
+variable "enable_mpi_operator" {
+  description = "Flag to enable the MPI Operator deployment"
+  type        = bool
+  default     = false
+}