File tree Expand file tree Collapse file tree 2 files changed +19
-11
lines changed Expand file tree Collapse file tree 2 files changed +19
-11
lines changed Original file line number Diff line number Diff line change @@ -469,7 +469,7 @@ def get_kueue_covered_resources_config(
469
469
A string of Kueue covered resources configuration.
470
470
"""
471
471
config_format = """
472
- - coveredResources: ["{resource_type}"]
472
+ - coveredResources: ["{resource_type}", "cpu", "memory" ]
473
473
flavors:
474
474
- name: {cluster_hardware_name}
475
475
resources:
@@ -479,6 +479,14 @@ def get_kueue_covered_resources_config(
479
479
nominalQuota: 99999999999
480
480
- name: "memory"
481
481
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
482
490
"""
483
491
config_string = config_format .format (
484
492
cluster_hardware_name = cluster_hardware_name ,
Original file line number Diff line number Diff line change @@ -39,16 +39,16 @@ def add_pw_resource_flavors(args):
39
39
40
40
def add_pw_resources_to_kueue (args ):
41
41
"""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
52
52
return ''
53
53
54
54
You can’t perform that action at this time.
0 commit comments