Add missing control ioctls used by NCCL-tests.

Fixes #10647.

PiperOrigin-RevId: 651491183
This commit is contained in:
Anthony Cui
2024-07-11 12:10:27 -07:00
committed by gVisor bot
parent 87c3b6f741
commit 32ed2f7987
2 changed files with 9 additions and 1 deletions
+7 -1
View File
@@ -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:
+2
View File
@@ -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,