allow multi gpu to be specified via ctr

crosbymichael · crosbymichael · commit 09d78bb6b9e8 · 2021-06-21T23:49:43.000-04:00
Signed-off-by: Michael Crosby &lt;michael@thepasture.io&gt;
diff --git a/cmd/ctr/commands/commands.go b/cmd/ctr/commands/commands.go
@@ -147,7 +147,7 @@ var (
 			Name:  "pid-file",
 			Usage: "file path to write the task's pid",
 		},
-		cli.IntFlag{
+		cli.IntSliceFlag{
 			Name:  "gpus",
 			Usage: "add gpus to the container",
 		},
diff --git a/cmd/ctr/commands/run/run_unix.go b/cmd/ctr/commands/run/run_unix.go
@@ -250,7 +250,7 @@ func NewContainer(ctx gocontext.Context, client *containerd.Client, context *cli
 			}))
 		}
 		if context.IsSet("gpus") {
-			opts = append(opts, nvidia.WithGPUs(nvidia.WithDevices(context.Int("gpus")), nvidia.WithAllCapabilities))
+			opts = append(opts, nvidia.WithGPUs(nvidia.WithDevices(context.IntSlice("gpus")...), nvidia.WithAllCapabilities))
 		}
 		if context.IsSet("allow-new-privs") {
 			opts = append(opts, oci.WithNewPrivileges)

Original file line number	Diff line number	Diff line change
`@@ -250,7 +250,7 @@ func NewContainer(ctx gocontext.Context, client containerd.Client, context cli`
`250`	`250`	`}))`
`251`	`251`	`}`
`252`	`252`	`if context.IsSet("gpus") {`
`253`		`- opts = append(opts, nvidia.WithGPUs(nvidia.WithDevices(context.Int("gpus")), nvidia.WithAllCapabilities))`
	`253`	`+ opts = append(opts, nvidia.WithGPUs(nvidia.WithDevices(context.IntSlice("gpus")...), nvidia.WithAllCapabilities))`
`254`	`254`	`}`
`255`	`255`	`if context.IsSet("allow-new-privs") {`
`256`	`256`	`opts = append(opts, oci.WithNewPrivileges)`