add flag to disable gpu to lotus-bench

2019-12-02 11:25:10 -08:00 · 2019-12-02 11:25:10 -08:00 · 22f96e626b
commit 22f96e626b
parent 5beb7f4a48
1 changed files with 7 additions and 0 deletions
--- a/cmd/lotus-bench/main.go
+++ b/cmd/lotus-bench/main.go
@ -64,8 +64,15 @@ func main() {
 				Name:  "sector-size",
 				Value: 1024,
 			},
+			&cli.BoolFlag{
+				Name:  "no-gpu",
+				Usage: "disable gpu usage for the benchmark run",
+			},
 		},
 		Action: func(c *cli.Context) error {
+			if c.Bool("no-gpu") {
+				os.Setenv("BELLMAN_NO_GPU", "1")
+			}
 			sdir, err := homedir.Expand(c.String("storage-dir"))
 			if err != nil {
 				return err