From 15de55a7f57802d6e897ddef4723c15dbf1705f2 Mon Sep 17 00:00:00 2001 From: Anthony Cui Date: Thu, 27 Jun 2024 14:43:37 -0700 Subject: [PATCH] Add support for 550.90.07 Nvidia driver. Ran through all the CUDA tests on T4, L4, and A100 GPUs. All tests passed without changes. PiperOrigin-RevId: 647452614 --- pkg/sentry/devices/nvproxy/version.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pkg/sentry/devices/nvproxy/version.go b/pkg/sentry/devices/nvproxy/version.go index 106554e86..f0131d9bd 100644 --- a/pkg/sentry/devices/nvproxy/version.go +++ b/pkg/sentry/devices/nvproxy/version.go @@ -372,7 +372,8 @@ func Init() { abi.uvmIoctl[nvgpu.UVM_MAP_EXTERNAL_ALLOCATION] = uvmIoctlHasFrontendFD[nvgpu.UVM_MAP_EXTERNAL_ALLOCATION_PARAMS_V550] return abi }) - _ = addDriverABI(550, 54, 15, "2e859ae5f912a9a47aaa9b2d40a94a14f6f486b5d3b67c0ddf8b72c1c9650385", v550_54_14) + v550_54_15 := addDriverABI(550, 54, 15, "2e859ae5f912a9a47aaa9b2d40a94a14f6f486b5d3b67c0ddf8b72c1c9650385", v550_54_14) + _ = addDriverABI(550, 90, 07, "51acf579d5a9884f573a1d3f522e7fafa5e7841e22a9cec0b4bbeae31b0b9733", v550_54_15) }) }