diff --git a/pkg/nvml/consts.go b/pkg/nvml/consts.go index d6a7ee3..cba00b9 100644 --- a/pkg/nvml/consts.go +++ b/pkg/nvml/consts.go @@ -152,3 +152,12 @@ const ( FEATURE_DISABLED = EnableState(nvml.FEATURE_DISABLED) FEATURE_ENABLED = EnableState(nvml.FEATURE_ENABLED) ) + +// Compute mode constants +const ( + COMPUTEMODE_DEFAULT = ComputeMode(nvml.COMPUTEMODE_DEFAULT) + COMPUTEMODE_EXCLUSIVE_THREAD = ComputeMode(nvml.COMPUTEMODE_EXCLUSIVE_THREAD) + COMPUTEMODE_PROHIBITED = ComputeMode(nvml.COMPUTEMODE_PROHIBITED) + COMPUTEMODE_EXCLUSIVE_PROCESS = ComputeMode(nvml.COMPUTEMODE_EXCLUSIVE_PROCESS) + COMPUTEMODE_COUNT = ComputeMode(nvml.COMPUTEMODE_COUNT) +)