From fb0dc9d525d493383f9cceefc9157e96c3de63ad Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Tue, 12 Mar 2024 17:23:12 +0200 Subject: [PATCH] Add ComputeMode constants Signed-off-by: Evan Lezar --- pkg/nvml/consts.go | 9 +++++++++ 1 file changed, 9 insertions(+) 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) +)