GitList

Browse code

daemon: fix bug in nvidia device registration

Signed-off-by: Tibor Vass <tibor@docker.com>

Tibor Vass authored on 2019/03/26 07:40:19
Showing 2 changed files

daemon/nvidia_linux.go index f0a6478..c136634 100644
pkg/capabilities/caps.go index c0deb13..4679a7a 100644

@@ -39,8 +39,8 @@ func init() {
 		capset:     capset,
 		updateSpec: setNvidiaGPUs,
 	}
-	for c := range capset {
-		nvidiaDriver.capset[c] = struct{}{}
+	for c := range allNvidiaCaps {
+		nvidiaDriver.capset[string(c)] = struct{}{}
 	}
 	registerDeviceDriver("nvidia", nvidiaDriver)
 }

pkg/capabilities/caps.go

History View file @ 739adb4

@@ -19,5 +19,6 @@ anyof:
 		}
 		return andList
 	}
+	// match anything
 	return nil
 }

...	...	@@ -19,5 +19,6 @@ anyof:
19	19	}
20	20	return andList
21	21	}
	22	+ // match anything
22	23	return nil
23	24	}

...	...	@@ -39,8 +39,8 @@ func init() {
39	39	capset: capset,
40	40	updateSpec: setNvidiaGPUs,
41	41	}
42		- for c := range capset {
43		- nvidiaDriver.capset[c] = struct{}{}
	42	+ for c := range allNvidiaCaps {
	43	+ nvidiaDriver.capset[string(c)] = struct{}{}
44	44	}
45	45	registerDeviceDriver("nvidia", nvidiaDriver)
46	46	}