hanzoai/visor
2
0
Fork 0
mirror of https://github.com/hanzoai/visor.git synced 2026-09-24 17:04:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

nvproxy: Add 535.261.03 and 570.172.08 drivers.

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

PiperOrigin-RevId: 792488148
This commit is contained in:
Ayush Ranjan
2025-08-07 23:49:00 -07:00
committed by gVisor bot
parent 4d9f9bca1e
commit 5033dfba6b
3 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/sentry/devices/nvproxy/version.go
+6 -1
View File
@@ -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
+7 -3
View File
@@ -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
+1 -1
View File
@@ -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 {