Skip to content

Commit

Permalink
Add mig-strategy flag to mps-control-daemon
Browse files Browse the repository at this point in the history
This change makes the handling of the MIG strategy consistent with
GFD and the device plugin.

This also removes the need to explicitly check for nil values when querying
this value.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
  • Loading branch information
elezar committed Mar 13, 2024
1 parent 05d317d commit 93f161c
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions cmd/mps-control-daemon/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 93f161c

Please sign in to comment.