Gets the NVidia extension options working, including marshalling.nvidia

1 files changed, 4 insertions, 0 deletions

diff --git a/cmd/gotop/main.go b/cmd/gotop/main.go
index 3a33535..94f03a8 100644
--- a/cmd/gotop/main.go
+++ b/cmd/gotop/main.go
@@ -77,6 +77,7 @@ func parseArgs() error {
 	opflag.BoolVarP(&conf.Mbps, "mbps", "", conf.Mbps, tr.Value("args.mbps"))
 	opflag.BoolVar(&conf.Test, "test", conf.Test, tr.Value("args.test"))
 	opflag.StringP("", "C", "", tr.Value("args.conffile"))
+	opflag.BoolVarP(&conf.Nvidia, "nvidia", "", conf.Nvidia, "Enable NVidia GPU support")
 	list := opflag.String("list", "", tr.Value("args.list"))
 	wc := opflag.Bool("write-config", false, tr.Value("args.write"))
 	opflag.SortFlags = false
@@ -139,6 +140,9 @@ func parseArgs() error {
 		}
 		os.Exit(0)
 	}
+	if conf.Nvidia {
+		conf.ExtensionVars["nvidia"] = "true"
+	}
 	if *wc {
 		path, err := conf.Write()
 		if err != nil {