From e91eb3cf2b8c27521c949bd88b509468351ae109 Mon Sep 17 00:00:00 2001 From: Ayush Ranjan Date: Wed, 31 Jan 2024 13:43:13 -0800 Subject: [PATCH] Add support for NVA06F_CTRL_CMD_GPFIFO_SCHEDULE in nvproxy. PiperOrigin-RevId: 603147485 --- pkg/abi/nvgpu/ctrl.go | 5 +++++ pkg/sentry/devices/nvproxy/version.go | 1 + 2 files changed, 6 insertions(+) diff --git a/pkg/abi/nvgpu/ctrl.go b/pkg/abi/nvgpu/ctrl.go index 26eb6d9a2..419d99444 100644 --- a/pkg/abi/nvgpu/ctrl.go +++ b/pkg/abi/nvgpu/ctrl.go @@ -273,6 +273,11 @@ const ( NVA06C_CTRL_CMD_PREEMPT = 0xa06c0105 ) +// From src/common/sdk/nvidia/inc/ctrl/ctrla06f/ctrla06fgpfifo.h: +const ( + NVA06F_CTRL_CMD_GPFIFO_SCHEDULE = 0xa06f0103 +) + // From src/common/sdk/nvidia/inc/ctrl/ctrlcb33.h: const ( NV_CONF_COMPUTE_CTRL_CMD_SYSTEM_GET_CAPABILITIES = 0xcb330101 diff --git a/pkg/sentry/devices/nvproxy/version.go b/pkg/sentry/devices/nvproxy/version.go index 90174f5a2..56a19fd16 100644 --- a/pkg/sentry/devices/nvproxy/version.go +++ b/pkg/sentry/devices/nvproxy/version.go @@ -269,6 +269,7 @@ func Init() { nvgpu.NVA06C_CTRL_CMD_GPFIFO_SCHEDULE: rmControlSimple, nvgpu.NVA06C_CTRL_CMD_SET_TIMESLICE: rmControlSimple, nvgpu.NVA06C_CTRL_CMD_PREEMPT: rmControlSimple, + nvgpu.NVA06F_CTRL_CMD_GPFIFO_SCHEDULE: rmControlSimple, nvgpu.NV0000_CTRL_CMD_SYSTEM_GET_BUILD_VERSION: ctrlClientSystemGetBuildVersion, nvgpu.NV0080_CTRL_CMD_FIFO_GET_CHANNELLIST: ctrlDevFIFOGetChannelList, nvgpu.NV0080_CTRL_CMD_GPU_GET_CLASSLIST: ctrlDevGpuGetClasslist,