Skip to content

Commit dc7222d

Browse files
ayushr2gvisor-bot
authored andcommitted
nvproxy: Add support for 575.57.08.
PiperOrigin-RevId: 770339157
1 parent 2e9ad00 commit dc7222d

File tree

2 files changed

+21
-1
lines changed

2 files changed

+21
-1
lines changed

pkg/abi/nvgpu/ctrl.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -582,6 +582,11 @@ const (
582582
NV2080_CTRL_CMD_TIMER_SET_GR_TICK_FREQ = 0x20800407
583583
)
584584

585+
// From src/common/sdk/nvidia/inc/ctrl/ctrl2080/ctrl2080thermal.h:
586+
const (
587+
NV2080_CTRL_CMD_THERMAL_SYSTEM_EXECUTE_V2 = 0x20800513
588+
)
589+
585590
// From src/common/sdk/nvidia/inc/ctrl/ctrl503c.h:
586591
const (
587592
NV503C_CTRL_CMD_REGISTER_VA_SPACE = 0x503c0102

pkg/sentry/devices/nvproxy/version.go

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -817,7 +817,22 @@ func Init() {
817817
})
818818

819819
v570_124_06 := addDriverABI(570, 124, 06, "1818c90657d17e510de9fa032385ff7e99063e848e901cb4636ee71c8b339313", v570_86_15)
820-
_ = addDriverABI(570, 133, 20, "1253d17b1528e8a24bf1f34a8ac6591c924b98ad7a32344bde253aa622ac1605", v570_124_06)
820+
v570_133_20 := addDriverABI(570, 133, 20, "1253d17b1528e8a24bf1f34a8ac6591c924b98ad7a32344bde253aa622ac1605", v570_124_06)
821+
822+
// 575.51.02 is an intermediate unqualified version from the main branch.
823+
v575_51_02 := func() *driverABI {
824+
abi := v570_133_20()
825+
abi.controlCmd[nvgpu.NV2080_CTRL_CMD_THERMAL_SYSTEM_EXECUTE_V2] = ctrlHandler(rmControlSimple, compUtil)
826+
prevStructs := abi.getStructs
827+
abi.getStructs = func() *driverABIStructs {
828+
structs := prevStructs()
829+
structs.controlStructs[nvgpu.NV2080_CTRL_CMD_THERMAL_SYSTEM_EXECUTE_V2] = simpleDriverStruct("NV2080_CTRL_THERMAL_SYSTEM_EXECUTE_V2_PARAMS")
830+
return structs
831+
}
832+
return abi
833+
}
834+
835+
_ = addDriverABI(575, 57, 8, "2aa701dac180a7b20a6e578cccd901ded8d44e57d60580f08f9d28dd1fffc6f2", v575_51_02)
821836
})
822837
}
823838

0 commit comments

Comments
 (0)