mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2024-11-28 06:52:34 +00:00
d3997eceb2
Signed-off-by: Evan Lezar <elezar@nvidia.com>
80 lines
3.4 KiB
Go
80 lines
3.4 KiB
Go
/*
|
|
* Copyright (c) 2021, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package nvml
|
|
|
|
import (
|
|
"github.com/NVIDIA/go-nvml/pkg/nvml"
|
|
)
|
|
|
|
const (
|
|
SUCCESS = nvml.SUCCESS
|
|
ERROR_UNINITIALIZED = nvml.ERROR_UNINITIALIZED
|
|
ERROR_INVALID_ARGUMENT = nvml.ERROR_INVALID_ARGUMENT
|
|
ERROR_NOT_SUPPORTED = nvml.ERROR_NOT_SUPPORTED
|
|
ERROR_NO_PERMISSION = nvml.ERROR_NO_PERMISSION
|
|
ERROR_ALREADY_INITIALIZED = nvml.ERROR_ALREADY_INITIALIZED
|
|
ERROR_NOT_FOUND = nvml.ERROR_NOT_FOUND
|
|
ERROR_INSUFFICIENT_SIZE = nvml.ERROR_INSUFFICIENT_SIZE
|
|
ERROR_INSUFFICIENT_POWER = nvml.ERROR_INSUFFICIENT_POWER
|
|
ERROR_DRIVER_NOT_LOADED = nvml.ERROR_DRIVER_NOT_LOADED
|
|
ERROR_TIMEOUT = nvml.ERROR_TIMEOUT
|
|
ERROR_IRQ_ISSUE = nvml.ERROR_IRQ_ISSUE
|
|
ERROR_LIBRARY_NOT_FOUND = nvml.ERROR_LIBRARY_NOT_FOUND
|
|
ERROR_FUNCTION_NOT_FOUND = nvml.ERROR_FUNCTION_NOT_FOUND
|
|
ERROR_CORRUPTED_INFOROM = nvml.ERROR_CORRUPTED_INFOROM
|
|
ERROR_GPU_IS_LOST = nvml.ERROR_GPU_IS_LOST
|
|
ERROR_RESET_REQUIRED = nvml.ERROR_RESET_REQUIRED
|
|
ERROR_OPERATING_SYSTEM = nvml.ERROR_OPERATING_SYSTEM
|
|
ERROR_LIB_RM_VERSION_MISMATCH = nvml.ERROR_LIB_RM_VERSION_MISMATCH
|
|
ERROR_IN_USE = nvml.ERROR_IN_USE
|
|
ERROR_MEMORY = nvml.ERROR_MEMORY
|
|
ERROR_NO_DATA = nvml.ERROR_NO_DATA
|
|
ERROR_VGPU_ECC_NOT_SUPPORTED = nvml.ERROR_VGPU_ECC_NOT_SUPPORTED
|
|
ERROR_INSUFFICIENT_RESOURCES = nvml.ERROR_INSUFFICIENT_RESOURCES
|
|
ERROR_UNKNOWN = nvml.ERROR_UNKNOWN
|
|
)
|
|
|
|
const (
|
|
DEVICE_MIG_ENABLE = nvml.DEVICE_MIG_ENABLE
|
|
DEVICE_MIG_DISABLE = nvml.DEVICE_MIG_DISABLE
|
|
)
|
|
|
|
const (
|
|
GPU_INSTANCE_PROFILE_1_SLICE = nvml.GPU_INSTANCE_PROFILE_1_SLICE
|
|
GPU_INSTANCE_PROFILE_2_SLICE = nvml.GPU_INSTANCE_PROFILE_2_SLICE
|
|
GPU_INSTANCE_PROFILE_3_SLICE = nvml.GPU_INSTANCE_PROFILE_3_SLICE
|
|
GPU_INSTANCE_PROFILE_4_SLICE = nvml.GPU_INSTANCE_PROFILE_4_SLICE
|
|
GPU_INSTANCE_PROFILE_7_SLICE = nvml.GPU_INSTANCE_PROFILE_7_SLICE
|
|
GPU_INSTANCE_PROFILE_8_SLICE = nvml.GPU_INSTANCE_PROFILE_8_SLICE
|
|
GPU_INSTANCE_PROFILE_COUNT = nvml.GPU_INSTANCE_PROFILE_COUNT
|
|
)
|
|
|
|
const (
|
|
COMPUTE_INSTANCE_PROFILE_1_SLICE = nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE
|
|
COMPUTE_INSTANCE_PROFILE_2_SLICE = nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE
|
|
COMPUTE_INSTANCE_PROFILE_3_SLICE = nvml.COMPUTE_INSTANCE_PROFILE_3_SLICE
|
|
COMPUTE_INSTANCE_PROFILE_4_SLICE = nvml.COMPUTE_INSTANCE_PROFILE_4_SLICE
|
|
COMPUTE_INSTANCE_PROFILE_7_SLICE = nvml.COMPUTE_INSTANCE_PROFILE_7_SLICE
|
|
COMPUTE_INSTANCE_PROFILE_8_SLICE = nvml.COMPUTE_INSTANCE_PROFILE_8_SLICE
|
|
COMPUTE_INSTANCE_PROFILE_COUNT = nvml.COMPUTE_INSTANCE_PROFILE_COUNT
|
|
)
|
|
|
|
const (
|
|
COMPUTE_INSTANCE_ENGINE_PROFILE_SHARED = nvml.COMPUTE_INSTANCE_ENGINE_PROFILE_SHARED
|
|
COMPUTE_INSTANCE_ENGINE_PROFILE_COUNT = nvml.COMPUTE_INSTANCE_ENGINE_PROFILE_COUNT
|
|
)
|