From de7f30328e812ee8f582a612c779cbe510849ede Mon Sep 17 00:00:00 2001
From: Paramadon <siprmp@amazon.com>
Date: Thu, 9 May 2024 11:31:16 -0400
Subject: [PATCH] dcgm exporter is up and running

---
 .github/workflows/daemonset.yaml        | 157 ++++++++++++++++++++++++
 integration-tests/terraform/eks/main.tf |   2 +-
 2 files changed, 158 insertions(+), 1 deletion(-)
 create mode 100644 .github/workflows/daemonset.yaml

diff --git a/.github/workflows/daemonset.yaml b/.github/workflows/daemonset.yaml
new file mode 100644
index 000000000..8dfde833f
--- /dev/null
+++ b/.github/workflows/daemonset.yaml
@@ -0,0 +1,157 @@
+# Please edit the object below. Lines beginning with a '#' will be ignored,
+# and an empty file will abort the edit. If an error occurs while saving this file will be
+# reopened with the relevant failures.
+#
+apiVersion: apps/v1
+kind: DaemonSet
+metadata:
+  annotations:
+    deprecated.daemonset.template.generation: "1"
+  creationTimestamp: "2024-05-01T21:01:08Z"
+  generation: 1
+  labels:
+    k8s-app: dcgm-exporter
+    version: v1
+  name: dcgm-exporter
+  namespace: amazon-cloudwatch
+  resourceVersion: "1704573"
+  uid: ea691d02-d5a7-47bf-a58f-3bd1fd5e4de6
+spec:
+  revisionHistoryLimit: 10
+  selector:
+    matchLabels:
+      k8s-app: dcgm-exporter
+  template:
+    metadata:
+      creationTimestamp: null
+      labels:
+        k8s-app: dcgm-exporter
+        version: v1
+    spec:
+      affinity:
+        nodeAffinity:
+          requiredDuringSchedulingIgnoredDuringExecution:
+            nodeSelectorTerms:
+              - matchExpressions:
+                  - key: node.kubernetes.io/instance-type
+                    operator: In
+                    values:
+                      - p2.xlarge
+                      - p2.8xlarge
+                      - p2.16xlarge
+                      - p3.2xlarge
+                      - p3.8xlarge
+                      - p3.16xlarge
+                      - p3dn.24xlarge
+                      - p4d.24xlarge
+                      - p4de.24xlarge
+                      - p5.48xlarge
+                      - g3s.xlarge
+                      - g3.4xlarge
+                      - g3.8xlarge
+                      - g3.16xlarge
+                      - g4dn.xlarge
+                      - g4dn.2xlarge
+                      - g4dn.4xlarge
+                      - g4dn.8xlarge
+                      - g4dn.16xlarge
+                      - g4dn.12xlarge
+                      - g4dn.metal
+                      - g4ad.xlarge
+                      - g4ad.2xlarge
+                      - g4ad.4xlarge
+                      - g4ad.8xlarge
+                      - g4ad.16xlarge
+                      - g5.xlarge
+                      - g5.2xlarge
+                      - g5.4xlarge
+                      - g5.8xlarge
+                      - g5.16xlarge
+                      - g5.xlarge
+                      - g5.2xlarge
+                      - g5.4xlarge
+                      - g5.8xlarge
+                      - g5.16xlarge
+                      - g5.12xlarge
+                      - g5.24xlarge
+                      - g5.48xlarge
+                      - g5g.xlarge
+                      - g5g.2xlarge
+                      - g5g.4xlarge
+                      - g5g.8xlarge
+                      - g5g.16xlarge
+                      - g5g.metal
+      containers:
+        - args:
+            - --web-config-file=/etc/dcgm-exporter/web-config.yaml
+          env:
+            - name: DCGM_EXPORTER_KUBERNETES
+              value: "true"
+            - name: DCGM_EXPORTER_LISTEN
+              value: :9400
+            - name: DCGM_EXPORTER_COLLECTORS
+              value: /etc/dcgm-exporter/dcp-metrics-included.csv
+            - name: NODE_NAME
+              valueFrom:
+                fieldRef:
+                  apiVersion: v1
+                  fieldPath: spec.nodeName
+          image: 602401143452.dkr.ecr.us-east-1.amazonaws.com/eks/observability/dcgm-exporter:3.3.3-3.3.1-ubuntu22.04
+          imagePullPolicy: IfNotPresent
+          name: dcgm-exporter
+          ports:
+            - containerPort: 9400
+              name: metrics
+              protocol: TCP
+          resources:
+            limits:
+              cpu: 500m
+              memory: 250Mi
+            requests:
+              cpu: 250m
+              memory: 128Mi
+          terminationMessagePath: /dev/termination-log
+          terminationMessagePolicy: File
+          volumeMounts:
+            - mountPath: /var/lib/kubelet/pod-resources
+              name: pod-gpu-resources
+              readOnly: true
+            - mountPath: /etc/dcgm-exporter/
+              name: dcgm-config
+            - mountPath: /etc/amazon-cloudwatch-observability-dcgm-cert
+              name: dcgmtls
+              readOnly: true
+          dnsPolicy: ClusterFirst
+          nodeSelector:
+            kubernetes.io/os: linux
+          restartPolicy: Always
+          schedulerName: default-scheduler
+          securityContext: {}
+          serviceAccount: dcgm-exporter-service-acct
+          serviceAccountName: dcgm-exporter-service-acct
+          terminationGracePeriodSeconds: 30
+          volumes:
+          - name: dcgmtls
+            secret:
+              defaultMode: 420
+              items:
+                - key: tls.crt
+                  path: server.crt
+                - key: tls.key
+                  path: server.key
+              secretName: amazon-cloudwatch-observability-agent-cert
+          - hostPath:
+              path: /var/lib/kubelet/pod-resources
+              type: ""
+            name: pod-gpu-resources
+          - configMap:
+              defaultMode: 420
+              name: dcgm-exporter-config-map
+            name: dcgm-config
+      updateStrategy:
+        rollingUpdate:
+          maxSurge: 0
+          maxUnavailable: 1
+        type: RollingUpdate
+    status:
+      currentNumberScheduled: 1
\ No newline at end of file
diff --git a/integration-tests/terraform/eks/main.tf b/integration-tests/terraform/eks/main.tf
index 2eff0b5a4..92db23eb8 100644
--- a/integration-tests/terraform/eks/main.tf
+++ b/integration-tests/terraform/eks/main.tf
@@ -46,7 +46,7 @@ resource "aws_eks_node_group" "this" {
     min_size     = 1
   }
 
-  ami_type       = "AL2_x86_64"
+  ami_type       = "AL2_x86_64_GPU"
   capacity_type  = "ON_DEMAND"
   disk_size      = 20
   instance_types = ["g4dn.xlarge"]