chore: Add expconf slurm.gpu_type [FOUNDENG-338] (#5448)

Add configuation to enable Determined EE support for slurm.gpu_type.
determined-ai · Nov 17, 2022 · 781e754 · 781e754
1 parent c7d6923
commit 781e754
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 0 deletions.
diff --git a/harness/determined/common/schemas/expconf/_gen.py b/harness/determined/common/schemas/expconf/_gen.py
@@ -1522,6 +1522,13 @@
             "minimum": 1,
             "default": null
         },
+        "gpu_type": {
+            "type": [
+                "string",
+                "null"
+            ],
+            "default": null
+        },
         "sbatch_args": {
             "type": [
                 "array",

diff --git a/master/pkg/schemas/expconf/experiment_config.go b/master/pkg/schemas/expconf/experiment_config.go
@@ -188,6 +188,7 @@ func (l *LabelsV0) UnmarshalJSON(data []byte) error {
 // SlurmConfigV0 configures experiment resource usage.
 type SlurmConfigV0 struct {
 	RawSlotsPerNode *int     `json:"slots_per_node,omitempty"`
+	RawGpuType      *string  `json:"gpu_type,omitempty"`
 	RawSbatchArgs   []string `json:"sbatch_args,omitempty"`
 }
 

diff --git a/master/pkg/schemas/expconf/zgen_slurm_config_v0.go b/master/pkg/schemas/expconf/zgen_slurm_config_v0.go
diff --git a/master/pkg/schemas/zgen_schemas.go b/master/pkg/schemas/zgen_schemas.go
diff --git a/schemas/expconf/v0/hpc-cluster-slurm.json b/schemas/expconf/v0/hpc-cluster-slurm.json
@@ -14,6 +14,13 @@
             "minimum": 1,
             "default": null
         },
+        "gpu_type": {
+            "type": [
+                "string",
+                "null"
+            ],
+            "default": null
+        },
         "sbatch_args": {
             "type": [
                 "array",