Merge pull request #590 from elezar/fix-mig-strategy-reference

Default to mig-strategy none if not specified
NVIDIA · Mar 15, 2024 · 32bb029 · 32bb029
2 parents 78b5fad + 93f161c
commit 32bb029
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cmd/mps-control-daemon/main.go b/cmd/mps-control-daemon/main.go
@@ -66,6 +66,12 @@ func main() {
 			Destination: &config.configFile,
 			EnvVars:     []string{"CONFIG_FILE"},
 		},
+		&cli.StringFlag{
+			Name:    "mig-strategy",
+			Value:   spec.MigStrategyNone,
+			Usage:   "the desired strategy for exposing MIG devices on GPUs that support it:\n\t\t[none | single | mixed]",
+			EnvVars: []string{"MIG_STRATEGY"},
+		},
 	}
 	c.Flags = config.flags