Skip to content

Commit 1bb60af

Browse files
committed
Add cpu and memory to covered resources
1 parent 329a0f8 commit 1bb60af

File tree

2 files changed

+19
-11
lines changed

2 files changed

+19
-11
lines changed

src/xpk/core/kueue.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,7 @@ def get_kueue_covered_resources_config(
469469
A string of Kueue covered resources configuration.
470470
"""
471471
config_format = """
472-
- coveredResources: ["{resource_type}"]
472+
- coveredResources: ["{resource_type}", "cpu", "memory"]
473473
flavors:
474474
- name: {cluster_hardware_name}
475475
resources:
@@ -479,6 +479,14 @@ def get_kueue_covered_resources_config(
479479
nominalQuota: 99999999999
480480
- name: "memory"
481481
nominalQuota: 9999999Ti
482+
- name: cpu-user
483+
resources:
484+
- name: "{resource_type}"
485+
nominalQuota: 0
486+
- name: "cpu"
487+
nominalQuota: 480
488+
- name: "memory"
489+
nominalQuota: 2000G
482490
"""
483491
config_string = config_format.format(
484492
cluster_hardware_name=cluster_hardware_name,

src/xpk/core/pathways.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -39,16 +39,16 @@ def add_pw_resource_flavors(args):
3939

4040
def add_pw_resources_to_kueue(args):
4141
"""Add resource flavors required for Pathways, to the cluster queue."""
42-
resources_yaml = """- coveredResources: ["cpu", "memory"]
43-
flavors:
44-
- name: cpu-user
45-
resources:
46-
- name: "cpu"
47-
nominalQuota: 480
48-
- name: "memory"
49-
nominalQuota: 2000G"""
50-
if args.enable_pathways:
51-
return resources_yaml
42+
# resources_yaml = """- coveredResources: ["cpu", "memory"]
43+
# flavors:
44+
# - name: cpu-user
45+
# resources:
46+
# - name: "cpu"
47+
# nominalQuota: 480
48+
# - name: "memory"
49+
# nominalQuota: 2000G"""
50+
# if args.enable_pathways:
51+
# return resources_yaml
5252
return ''
5353

5454

0 commit comments

Comments
 (0)