From ef410e665ba9c9bd98d23ca6c0e61fd91dce8887 Mon Sep 17 00:00:00 2001 From: Jamie Liu Date: Tue, 18 Jul 2023 15:15:38 -0700 Subject: [PATCH] Pass NV2080_CTRL_CMD_MC_SERVICE_INTERRUPTS through nvproxy. Fixes #9176 PiperOrigin-RevId: 549125072 --- pkg/abi/nvgpu/ctrl.go | 3 ++- pkg/sentry/devices/nvproxy/frontend.go | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/pkg/abi/nvgpu/ctrl.go b/pkg/abi/nvgpu/ctrl.go index 04547ccd1..30938b692 100644 --- a/pkg/abi/nvgpu/ctrl.go +++ b/pkg/abi/nvgpu/ctrl.go @@ -190,7 +190,8 @@ type NV2080_CTRL_GR_GET_INFO_PARAMS struct { // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080mc.h: const ( - NV2080_CTRL_CMD_MC_GET_ARCH_INFO = 0x20801701 + NV2080_CTRL_CMD_MC_GET_ARCH_INFO = 0x20801701 + NV2080_CTRL_CMD_MC_SERVICE_INTERRUPTS = 0x20801702 ) // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080nvlink.h: diff --git a/pkg/sentry/devices/nvproxy/frontend.go b/pkg/sentry/devices/nvproxy/frontend.go index 7c1df0998..2e4747512 100644 --- a/pkg/sentry/devices/nvproxy/frontend.go +++ b/pkg/sentry/devices/nvproxy/frontend.go @@ -582,6 +582,7 @@ func rmControl(fi *frontendIoctlState) (uintptr, error) { nvgpu.NV2080_CTRL_CMD_GR_GET_TPC_MASK, nvgpu.NV2080_CTRL_CMD_GSP_GET_FEATURES, nvgpu.NV2080_CTRL_CMD_MC_GET_ARCH_INFO, + nvgpu.NV2080_CTRL_CMD_MC_SERVICE_INTERRUPTS, nvgpu.NV2080_CTRL_CMD_NVLINK_GET_NVLINK_STATUS, nvgpu.NV2080_CTRL_CMD_PERF_BOOST, nvgpu.NV2080_CTRL_CMD_RC_GET_WATCHDOG_INFO,