Skip to content

Commit 329a0f8

Browse files
committed
Add tpu resouce flavor to kueue
1 parent 348388a commit 329a0f8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/xpk/core/kueue.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -475,6 +475,10 @@ def get_kueue_covered_resources_config(
475475
resources:
476476
- name: "{resource_type}"
477477
nominalQuota: {total_chips}
478+
- name: "cpu"
479+
nominalQuota: 99999999999
480+
- name: "memory"
481+
nominalQuota: 9999999Ti
478482
"""
479483
config_string = config_format.format(
480484
cluster_hardware_name=cluster_hardware_name,

0 commit comments

Comments
 (0)