chore: set GOMEMLIMIT in app, parquet (#261)

wandb · Nov 15, 2024 · 973d79d · 973d79d
1 parent 0561520
commit 973d79d
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 1 deletion.
diff --git a/charts/operator-wandb/Chart.yaml b/charts/operator-wandb/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 name: operator-wandb
 description: A Helm chart for deploying W&B to Kubernetes
 type: application
-version: 0.18.17
+version: 0.18.18
 appVersion: 1.0.0
 icon: https://wandb.ai/logo.svg
 

diff --git a/charts/operator-wandb/charts/app/templates/_deployment.tpl b/charts/operator-wandb/charts/app/templates/_deployment.tpl
@@ -96,6 +96,10 @@ spec:
               containerPort: 8125
               protocol: TCP
           env:
+            - name: GOMEMLIMIT
+              valueFrom:
+                resourceFieldRef:
+                  resource: limits.memory
             - name: GLUE_ENABLED
               value: "{{ .glueSingletonEnabled }}"
             {{- if .onlyService }}

diff --git a/charts/operator-wandb/charts/parquet/templates/cron.yaml b/charts/operator-wandb/charts/parquet/templates/cron.yaml
@@ -58,6 +58,10 @@ spec:
                   subPath: redis_ca.pem
                 {{- end }}
               env:
+                - name: GOMEMLIMIT
+                  valueFrom:
+                    resourceFieldRef:
+                      resource: limits.memory
                 - name: GORILLA_GLUE_EXECUTE
                   value: "true"
                 - name: GORILLA_GLUE_EXECUTE_TASK_NAME

diff --git a/charts/operator-wandb/charts/parquet/templates/deployment.yaml b/charts/operator-wandb/charts/parquet/templates/deployment.yaml
@@ -70,6 +70,10 @@ spec:
               containerPort: 8087
               protocol: TCP
           env:
+            - name: GOMEMLIMIT
+              valueFrom:
+                resourceFieldRef:
+                  resource: limits.memory
             {{- if ne .Values.traceRatio 0.0 }}
             - name: GORILLA_TRACER
               value: "otlp+grpc://{{ .Release.Name }}-otel-daemonset:4317?trace_ratio={{ .Values.traceRatio }}"