diff --git a/audit/cmd/bee/main.go b/audit/cmd/bee/main.go index cfdff8f..01b77bc 100644 --- a/audit/cmd/bee/main.go +++ b/audit/cmd/bee/main.go @@ -8,6 +8,7 @@ import ( "io" "log/slog" "os" + "os/exec" "runtime/debug" "strconv" "strings" @@ -88,6 +89,8 @@ func run(args []string, stdout, stderr io.Writer) (exitCode int) { return runBenchmark(args[1:], stdout, stderr) case "bee-worker": return runBeeWorker(args[1:], stdout, stderr) + case "gpu-bandwidth-groups": + return runGPUBandwidthGroups(args[1:], stdout, stderr) case "version", "--version", "-version": fmt.Fprintln(stdout, Version) return 0 @@ -111,6 +114,7 @@ func printRootUsage(w io.Writer) { bee run (bare name is looked up as scenarios/.json on removable media) bee benchmark nvidia [--profile standard|stability|overnight] bee bee-worker --export-dir `+app.DefaultExportDir+` --task-id TASK-001 + bee gpu-bandwidth-groups [--stage