diff --git a/pkg/abi/nvgpu/ctrl.go b/pkg/abi/nvgpu/ctrl.go index dd39e8891..c07432042 100644 --- a/pkg/abi/nvgpu/ctrl.go +++ b/pkg/abi/nvgpu/ctrl.go @@ -311,6 +311,11 @@ const ( NV2080_CTRL_CMD_GR_GET_SM_ISSUE_RATE_MODIFIER = 0x20801230 ) +// From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080grmgr.h: +const ( + NV2080_CTRL_CMD_GRMGR_GET_GR_FS_INFO = 0x20803801 +) + // From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080gsp.h: const ( NV2080_CTRL_CMD_GSP_GET_FEATURES = 0x20803601 @@ -387,7 +392,8 @@ const ( // From src/common/sdk/nvidia/inc/ctrl/ctrl906f.h: const ( - NV906F_CTRL_CMD_RESET_CHANNEL = 0x906f0102 + NV906F_CTRL_GET_CLASS_ENGINEID = 0x906f0101 + NV906F_CTRL_CMD_RESET_CHANNEL = 0x906f0102 ) // From src/common/sdk/nvidia/inc/ctrl/ctrl90e6.h: diff --git a/pkg/sentry/devices/nvproxy/version.go b/pkg/sentry/devices/nvproxy/version.go index 222264701..9c9d6935e 100644 --- a/pkg/sentry/devices/nvproxy/version.go +++ b/pkg/sentry/devices/nvproxy/version.go @@ -255,6 +255,7 @@ func Init() { 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_GRMGR_GET_GR_FS_INFO: 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, @@ -270,6 +271,7 @@ func Init() { nvgpu.NV83DE_CTRL_CMD_DEBUG_SET_EXCEPTION_MASK: rmControlSimple, nvgpu.NV83DE_CTRL_CMD_DEBUG_READ_ALL_SM_ERROR_STATES: rmControlSimple, nvgpu.NV83DE_CTRL_CMD_DEBUG_CLEAR_ALL_SM_ERROR_STATES: rmControlSimple, + nvgpu.NV906F_CTRL_GET_CLASS_ENGINEID: rmControlSimple, nvgpu.NV906F_CTRL_CMD_RESET_CHANNEL: rmControlSimple, nvgpu.NV90E6_CTRL_CMD_MASTER_GET_VIRTUAL_FUNCTION_ERROR_CONT_INTR_MASK: rmControlSimple, nvgpu.NVC36F_CTRL_GET_CLASS_ENGINEID: rmControlSimple,