Skip to content

Commit d498a2d

Browse files
committed
Remove support for nvidia driver capabilities
This chanage updates the device spec to allow for a driver OR capabilities to be specified. It also removes the capabilities from the gpu spec and indicates that the gpu capability is implied. Signed-off-by: Evan Lezar <[email protected]>
1 parent a0d3b94 commit d498a2d

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

schema/compose-spec.json

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1226,8 +1226,17 @@
12261226
},
12271227
"additionalProperties": false,
12281228
"patternProperties": {"^x-": {}},
1229-
"required": [
1230-
"capabilities"
1229+
"oneOf": [
1230+
{
1231+
"required": [
1232+
"capabilities"
1233+
]
1234+
},
1235+
{
1236+
"required": [
1237+
"driver"
1238+
]
1239+
}
12311240
]
12321241
}
12331242
},
@@ -1245,10 +1254,6 @@
12451254
"items": {
12461255
"type": "object",
12471256
"properties": {
1248-
"capabilities": {
1249-
"$ref": "#/definitions/list_of_strings",
1250-
"description": "List of capabilities the GPU needs to have (e.g., 'compute', 'utility')."
1251-
},
12521257
"count": {
12531258
"type": ["string", "integer"],
12541259
"description": "Number of GPUs to use."
@@ -1259,7 +1264,7 @@
12591264
},
12601265
"driver": {
12611266
"type": "string",
1262-
"description": "GPU driver to use (e.g., 'nvidia')."
1267+
"description": "GPU driver to use (e.g., 'nvidia'). If this is not specified any available driver with the 'gpu' capability will be selected."
12631268
},
12641269
"options": {
12651270
"$ref": "#/definitions/list_or_dict",

0 commit comments

Comments
 (0)