From d216bae8fb45a1cfad14ebb52c9940311bbe5180 Mon Sep 17 00:00:00 2001
From: lusoal <lucasdu@amazon.com>
Date: Tue, 17 Oct 2023 09:42:14 -0500
Subject: [PATCH] Changed Karpenter provisioner name to match the correct
 instance type

---
 ai-ml/jupyterhub/addons.tf                                  | 6 +++---
 .../helm/jupyterhub/jupyterhub-values-cognito.yaml          | 2 +-
 .../jupyterhub/helm/jupyterhub/jupyterhub-values-dummy.yaml | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/ai-ml/jupyterhub/addons.tf b/ai-ml/jupyterhub/addons.tf
index a700e4ada..c55166b6f 100755
--- a/ai-ml/jupyterhub/addons.tf
+++ b/ai-ml/jupyterhub/addons.tf
@@ -187,12 +187,12 @@ module "eks_blueprints_addons" {
       ]
     }
     karpenter-resources-mig = {
-      name        = "karpenter-resources-mig"
-      description = "A Helm chart for karpenter GPU based resources - compatible with GPU MIG"
+      name        = "karpenter-resources-gpu"
+      description = "A Helm chart for karpenter GPU based resources - compatible with P4d instances"
       chart       = "${path.module}/helm/karpenter-resources"
       values = [
         <<-EOT
-          name: gpu-mig
+          name: gpu
           clusterName: ${module.eks.cluster_name}
           instanceSizes: ["24xlarge"]
           instanceFamilies: ["p4d"]
diff --git a/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-cognito.yaml b/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-cognito.yaml
index 4ddc3686c..386e0e42e 100755
--- a/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-cognito.yaml
+++ b/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-cognito.yaml
@@ -196,7 +196,7 @@ singleuser:
         image: cschranz/gpu-jupyter:v1.5_cuda-11.6_ubuntu-20.04_python-only
         node_selector:
           node.kubernetes.io/instance-type: p4d.24xlarge
-          karpenter.sh/provisioner-name: gpu-p4d
+          karpenter.sh/provisioner-name: gpu
         tolerations:
           - key: "nvidia.com/gpu"
             operator: "Exists"
diff --git a/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-dummy.yaml b/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-dummy.yaml
index 792ded86d..9064e1d47 100755
--- a/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-dummy.yaml
+++ b/ai-ml/jupyterhub/helm/jupyterhub/jupyterhub-values-dummy.yaml
@@ -177,7 +177,7 @@ singleuser:
         image: cschranz/gpu-jupyter:v1.5_cuda-11.6_ubuntu-20.04_python-only
         node_selector:
           node.kubernetes.io/instance-type: p4d.24xlarge
-          karpenter.sh/provisioner-name: gpu-p4d
+          karpenter.sh/provisioner-name: gpu
         tolerations:
           - key: "nvidia.com/gpu"
             operator: "Exists"