From 53f976dede42b4e0071e22681933f28f28f137d6 Mon Sep 17 00:00:00 2001 From: Ayush Ranjan Date: Thu, 21 Mar 2024 09:25:53 -0700 Subject: [PATCH] Qualify Nvidia driver 535.161.07. Tested on a T4 GPU with driver version 535.161.07: ``` $ docker run --runtime=runsc --rm -it --gpus all nvcr.io/nvidia/k8s/cuda-sample:vectoradd-cuda11.7.1-ubi8 [Vector addition of 50000 elements] Copy input data from the host memory to the CUDA device CUDA kernel launch with 196 blocks of 256 threads Copy output data from the CUDA device to the host memory Test PASSED Done ``` PiperOrigin-RevId: 617867263 --- pkg/sentry/devices/nvproxy/version.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/pkg/sentry/devices/nvproxy/version.go b/pkg/sentry/devices/nvproxy/version.go index 4c896b9a9..2cff97585 100644 --- a/pkg/sentry/devices/nvproxy/version.go +++ b/pkg/sentry/devices/nvproxy/version.go @@ -315,17 +315,18 @@ func Init() { } }) - // 535.104.12 does not exist on the main branch. It branched off the main + // 535.104.12 exists on the "535.104.12" branch. It branched off the main // branch at 535.104.05. _ = addDriverABI(535, 104, 12, "ffc2d89e233d2427edb1ff5f436028a94b3ef86e78f97e088e11d905c82e8001", v535_104_05) // 535.113.01 is an intermediate unqualified version from the main branch. v535_113_01 := v535_104_05 - // The following do not exist on the main branch. They branched off the - // main branch at 535.113.01. + // The following exist on the "535" branch. They branched off the main + // branch at 535.113.01. v535_129_03 := addDriverABI(535, 129, 03, "e6dca5626a2608c6bb2a046cfcb7c1af338b9e961a7dd90ac09bb8a126ff002e", v535_113_01) - _ = addDriverABI(535, 154, 05, "7e95065caa6b82de926110f14827a61972eb12c200e863a29e9fb47866eaa898", v535_129_03) + v535_154_05 := addDriverABI(535, 154, 05, "7e95065caa6b82de926110f14827a61972eb12c200e863a29e9fb47866eaa898", v535_129_03) + _ = addDriverABI(535, 161, 07, "edc527f1dcfa0212a3bf815ebf302d45ef9663834a41e11a851dd38da159a8cd", v535_154_05) // 545.23.06 is an intermediate unqualified version from the main branch. v545_23_06 := func() *driverABI {