nvproxy: Add 535.261.03 and 570.172.08 drivers.

Also fix the `tools/gpu:main checksum` command to output the checksum of both
architectures.

PiperOrigin-RevId: 792488148

Author: ayushr2

pkg/sentry/devices/nvproxy/version.go

@@ -707,7 +707,8 @@ func Init() {
 		v535_183_06 := addDriverABI(535, 183, 06, "c7bb0a0569c5347845479ed4e3e4d885c6ee3b8adf068c3401cdf754d5ba3d3b", ChecksumNoDriver, v535_183_01)
 		v535_216_01 := addDriverABI(535, 216, 01, "5ddea1147810012e33967c3181341bcd6624bd3d654c63f845df833b4ece6af7", "4869ae0345b5892b2a50aed566c8226d3e07813d1190aa466feba5e9e21b33b9", v535_183_06)
 		v535_230_02 := addDriverABI(535, 230, 02, "20cca9118083fcc8083158466e9cb2b616a7922206bcb7296b1fa5cc9af2e0fd", "ea000e6ff481f55e9bfedbea93b739368c635fe4be6156fdad560524ac7f363b", v535_216_01)
-		_ = addDriverABI(535, 247, 01, "c250e686494cb0c1b5eeea58ba2003707510b2766df05b06ba20b11b3445466b", "bd8ea5c3747a588ff1a29b4f59300d2eba69402a605cb95fce10a30f535993d0", v535_230_02)
+		v535_247_01 := addDriverABI(535, 247, 01, "c250e686494cb0c1b5eeea58ba2003707510b2766df05b06ba20b11b3445466b", "bd8ea5c3747a588ff1a29b4f59300d2eba69402a605cb95fce10a30f535993d0", v535_230_02)
+		_ = addDriverABI(535, 261, 03, "d74b61d11e9c9b9052f4042d6ec4437f13d1def30e964e232d47e5d659d11d68", "9a412d3ac01c99d2ca02100a7139597fce8804c52bf533d11b60437286834a93", v535_247_01)
 
 		// 545.23.06 is an intermediate unqualified version from the main branch.
 		v545_23_06 := func() *driverABI {
@@ -892,6 +893,10 @@ func Init() {
 		v570_124_06 := addDriverABI(570, 124, 06, "1818c90657d17e510de9fa032385ff7e99063e848e901cb4636ee71c8b339313", ChecksumNoDriver, v570_86_15)
 		v570_133_20 := addDriverABI(570, 133, 20, "1253d17b1528e8a24bf1f34a8ac6591c924b98ad7a32344bde253aa622ac1605", ChecksumNoDriver, v570_124_06)
 
+		// The following exist on the "570" branch. They branched off the main
+		// branch at 570.133.20.
+		_ = addDriverABI(570, 172, 8, "0256867e082caf93d7b25fa7c8e69b316062a9c6c72c6e228fad7b238c6fa17d", "15547216f2b514ace7724a5ab4c3327669904a41cafb8d4d9048d3c9b60963d8", v570_133_20)
+
 		// 575.51.02 is an intermediate unqualified version from the main branch.
 		v575_51_02 := func() *driverABI {
 			abi := v570_133_20()

tools/gpu/drivers/install_driver.go

@@ -38,6 +38,7 @@ const (
 	nvidiaBaseURLX86_64 = "https://us.download.nvidia.com/tesla/%s/NVIDIA-Linux-x86_64-%s.run"
 	nvidiaARM64BaseURL  = "https://us.download.nvidia.com/XFree86/aarch64/%s/NVIDIA-Linux-aarch64-%s.run"
 
+	archAMD64 = "amd64"
 	archARM64 = "arm64"
 )
 
@@ -46,10 +47,13 @@ func init() {
 }
 
 func getNvidiaBaseURL(driverVersion, arch string) string {
-	if arch == archARM64 {
+	switch arch {
+	case archARM64:
 		return fmt.Sprintf(nvidiaARM64BaseURL, driverVersion, driverVersion)
+	case archAMD64:
+		return fmt.Sprintf(nvidiaBaseURLX86_64, driverVersion, driverVersion)
 	}
-	return fmt.Sprintf(nvidiaBaseURLX86_64, driverVersion, driverVersion)
+	panic(fmt.Sprintf("unsupported arch: %q", arch))
 }
 
 // Installer handles the logic to install drivers.
@@ -337,7 +341,7 @@ func tryToPrintFailureLogs() {
 
 // ValidateChecksum validates the checksum of the driver.
 func ValidateChecksum(ctx context.Context, version string, checksums nvproxy.Checksums) error {
-	for _, arch := range []string{"x86_64", archARM64} {
+	for _, arch := range []string{archAMD64, archARM64} {
 		wantChecksum := checksums.X86_64()
 		if arch == archARM64 {
 			wantChecksum = checksums.Arm64()

tools/gpu/main.go

@@ -105,7 +105,7 @@ func main() {
 		}
 
 		var eg errgroup.Group
-		for _, arch := range []string{"amd64", "amd64"} {
+		for _, arch := range []string{"amd64", "arm64"} {
 			eg.Go(func() error {
 				installer, err := drivers.NewInstaller(*checksumVersion, false)
 				if err != nil {