From 3a86eaf0d88793211063074219a98ed7b60b1951 Mon Sep 17 00:00:00 2001 From: Ayush Ranjan Date: Thu, 8 Feb 2024 20:22:44 -0800 Subject: [PATCH] Add NV2080_CTRL_CMD_GR_GET_SM_ISSUE_RATE_MODIFIER to nvproxy. Fixes #9846 PiperOrigin-RevId: 605506597 --- pkg/abi/nvgpu/ctrl.go | 15 ++++++++------- pkg/sentry/devices/nvproxy/version.go | 1 + 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/pkg/abi/nvgpu/ctrl.go b/pkg/abi/nvgpu/ctrl.go index 419d99444..b854e1e2e 100644 --- a/pkg/abi/nvgpu/ctrl.go +++ b/pkg/abi/nvgpu/ctrl.go @@ -185,13 +185,14 @@ const ( // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080gr.h: const ( - NV2080_CTRL_CMD_GR_GET_INFO = 0x20801201 - NV2080_CTRL_CMD_GR_SET_CTXSW_PREEMPTION_MODE = 0x20801210 - NV2080_CTRL_CMD_GR_GET_CTX_BUFFER_SIZE = 0x20801218 - NV2080_CTRL_CMD_GR_GET_GLOBAL_SM_ORDER = 0x2080121b - NV2080_CTRL_CMD_GR_GET_CAPS_V2 = 0x20801227 - NV2080_CTRL_CMD_GR_GET_GPC_MASK = 0x2080122a - NV2080_CTRL_CMD_GR_GET_TPC_MASK = 0x2080122b + NV2080_CTRL_CMD_GR_GET_INFO = 0x20801201 + NV2080_CTRL_CMD_GR_SET_CTXSW_PREEMPTION_MODE = 0x20801210 + NV2080_CTRL_CMD_GR_GET_CTX_BUFFER_SIZE = 0x20801218 + NV2080_CTRL_CMD_GR_GET_GLOBAL_SM_ORDER = 0x2080121b + NV2080_CTRL_CMD_GR_GET_CAPS_V2 = 0x20801227 + NV2080_CTRL_CMD_GR_GET_GPC_MASK = 0x2080122a + NV2080_CTRL_CMD_GR_GET_TPC_MASK = 0x2080122b + NV2080_CTRL_CMD_GR_GET_SM_ISSUE_RATE_MODIFIER = 0x20801230 ) // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080gsp.h: diff --git a/pkg/sentry/devices/nvproxy/version.go b/pkg/sentry/devices/nvproxy/version.go index 56a19fd16..c4e138688 100644 --- a/pkg/sentry/devices/nvproxy/version.go +++ b/pkg/sentry/devices/nvproxy/version.go @@ -246,6 +246,7 @@ func Init() { nvgpu.NV2080_CTRL_CMD_GR_GET_CAPS_V2: rmControlSimple, nvgpu.NV2080_CTRL_CMD_GR_GET_GPC_MASK: rmControlSimple, nvgpu.NV2080_CTRL_CMD_GR_GET_TPC_MASK: rmControlSimple, + nvgpu.NV2080_CTRL_CMD_GR_GET_SM_ISSUE_RATE_MODIFIER: rmControlSimple, nvgpu.NV2080_CTRL_CMD_GSP_GET_FEATURES: rmControlSimple, nvgpu.NV2080_CTRL_CMD_MC_GET_ARCH_INFO: rmControlSimple, nvgpu.NV2080_CTRL_CMD_MC_SERVICE_INTERRUPTS: rmControlSimple,