From 6c1fba6a6d3fc15c29f80b9670a8cfed3250bbd5 Mon Sep 17 00:00:00 2001
From: Sam Stoelinga <sammiestoel@gmail.com>
Date: Fri, 22 Aug 2025 22:14:30 -0700
Subject: [PATCH] Kueue support jobsets that specify cpu and memory

---
 src/xpk/core/kueue.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/xpk/core/kueue.py b/src/xpk/core/kueue.py
index 257ed2bf1..7eb7310af 100644
--- a/src/xpk/core/kueue.py
+++ b/src/xpk/core/kueue.py
@@ -493,10 +493,17 @@ def get_kueue_covered_resources_config(
     A string of Kueue covered resources configuration.
   """
   config_format = """
-  - coveredResources: ["{resource_type}"]
+  - coveredResources:
+    - cpu
+    - memory
+    - "{resource_type}"
     flavors:
     - name: {cluster_hardware_name}
       resources:
+      - name: cpu
+        nominalQuota: 999999999999
+      - name: memory
+        nominalQuota: 99999999Ti
       - name: "{resource_type}"
         nominalQuota: {total_chips}
   """