Function main

tools/gpu/main.go:79–159

()

}

func main() {
	ctx := context.Background()
	if len(os.Args) < 2 {
		printUsage()
		os.Exit(1)
	}
	nvproxy.Init()
	switch os.Args[1] {
	case installCmdStr:
		if err := installCmd.Parse(os.Args[2:]); err != nil {
			log.Warningf("%s failed with: %v", installCmdStr, err)
			os.Exit(1)
		}
		installer, err := drivers.NewInstaller(*version, *latest)
		if err != nil {
			log.Warningf("Failed to create installer: %v", err.Error())
			os.Exit(1)
		}
		if err := installer.MaybeInstall(ctx); err != nil {
			log.Warningf("Failed to install driver: %v", err.Error())
			os.Exit(1)
		}
	case checksumCmdStr:
		if err := checksumCmd.Parse(os.Args[2:]); err != nil {
			log.Warningf("%s failed with: %v", checksumCmdStr, err)
			os.Exit(1)
		}

		var eg errgroup.Group
		for _, arch := range []string{"amd64", "arm64"} {
			eg.Go(func() error {
				installer, err := drivers.NewInstaller(*checksumVersion, false)
				if err != nil {
					return fmt.Errorf("failed to create installer: %w", err)
				}
				checksum, err := installer.ChecksumDriver(ctx, arch)
				if err != nil {
					return fmt.Errorf("failed to get checksum on arch %q: %w", arch, err)
				}
				fmt.Printf("%s %s %s\n", arch, *checksumVersion, checksum)
				return nil
			})
		}
		if err := eg.Wait(); err != nil {
			log.Warningf("Failed to get checksum: %v", err.Error())
			os.Exit(1)
		}

	case validateChecksumCmdStr:
		if err := validateChecksumCmd.Parse(os.Args[2:]); err != nil {
			log.Warningf("%s failed with: %v", validateChecksumCmdStr, err)
			os.Exit(1)
		}

		var eg errgroup.Group

		nvproxy.ForEachSupportDriver(func(version nvconf.DriverVersion, checksums nvproxy.Checksums) {
			eg.Go(func() error {