introduce service.gpus

Signed-off-by: Nicolas De Loof <[email protected]>

loader/loader_test.go

@@ -2224,6 +2224,32 @@ services:
 	assert.ErrorContains(t, err, `capabilities is required`)
 }
 
+func TestServiceGpus(t *testing.T) {
+	p, err := loadYAML(`
+name: service-gpus
+services:
+  test:
+    image: redis:alpine
+    gpus:
+      - driver: nvidia
+      - driver: 3dfx
+        device_ids: ["voodoo2"]
+        capabilities: ["directX"]
+`)
+	assert.NilError(t, err)
+	assert.DeepEqual(t, p.Services["test"].Gpus, []types.DeviceRequest{
+		{
+			Driver: "nvidia",
+			Count:  -1,
+		},
+		{
+			Capabilities: []string{"directX"},
+			Driver:       "3dfx",
+			IDs:          []string{"voodoo2"},
+		},
+	})
+}
+
 func TestServicePullPolicy(t *testing.T) {
 	actual, err := loadYAML(`
 name: service-pull-policy

schema/compose-spec.json

@@ -267,6 +267,7 @@
         },
         "external_links": {"type": "array", "items": {"type": "string"}, "uniqueItems": true},
         "extra_hosts": {"$ref": "#/definitions/extra_hosts"},
+        "gpus": {"$ref": "#/definitions/gpus"},
         "group_add": {
           "type": "array",
           "items": {
@@ -649,6 +650,23 @@
       }
     },
 
+    "gpus": {
+      "id": "#/definitions/gpus",
+      "type": "array",
+      "items": {
+        "type": "object",
+        "properties": {
+          "capabilities": {"$ref": "#/definitions/list_of_strings"},
+          "count": {"type": ["string", "integer"]},
+          "device_ids": {"$ref": "#/definitions/list_of_strings"},
+          "driver":{"type": "string"},
+          "options":{"$ref": "#/definitions/list_or_dict"}
+        },
+        "additionalProperties": false,
+        "patternProperties": {"^x-": {}}
+      }
+    },
+
     "include": {
       "id": "#/definitions/include",
       "oneOf": [

transform/canonical.go

@@ -32,6 +32,7 @@ func init() {
 	transformers["services.*.deploy.resources.reservations.devices.*"] = transformDeviceRequest
 	transformers["services.*.env_file"] = transformEnvFile
 	transformers["services.*.extends"] = transformExtends
+	transformers["services.*.gpus.*"] = transformDeviceRequest
 	transformers["services.*.networks"] = transformServiceNetworks
 	transformers["services.*.volumes.*"] = transformVolumeMount
 	transformers["services.*.devices.*"] = transformDeviceMapping