From 9a8a48455b67e57c50c42030844e445f6acdda0c Mon Sep 17 00:00:00 2001
From: Evan Lezar <elezar@nvidia.com>
Date: Wed, 18 Jun 2025 18:15:28 +0200
Subject: [PATCH] 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 NVIDIA capabilitied
from the documentation of the gpus.capabilities field.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
---
 schema/compose-spec.json | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/schema/compose-spec.json b/schema/compose-spec.json
index 946024bd..3d113b4d 100644
--- a/schema/compose-spec.json
+++ b/schema/compose-spec.json
@@ -1226,8 +1226,17 @@
         },
         "additionalProperties": false,
         "patternProperties": {"^x-": {}},
-        "required": [
-          "capabilities"
+        "oneOf": [
+          {
+            "required": [
+              "capabilities"
+            ]
+          },
+          {
+            "required": [
+              "driver"
+            ]
+          }
         ]
       }
     },
@@ -1247,7 +1256,7 @@
             "properties": {
               "capabilities": {
                 "$ref": "#/definitions/list_of_strings",
-                "description": "List of capabilities the GPU needs to have (e.g., 'compute', 'utility')."
+                "description": "List of capabilities the GPU needs to have."
               },
               "count": {
                 "type": ["string", "integer"],
@@ -1259,7 +1268,7 @@
               },
               "driver": {
                 "type": "string",
-                "description": "GPU driver to use (e.g., 'nvidia')."
+                "description": "GPU driver to use (e.g., 'nvidia'). If this is not specified any available driver with the 'gpu' capability will be selected."
               },
               "options": {
                 "$ref": "#/definitions/list_or_dict",