Documentation
¶
Index ¶
- Constants
- Variables
- func CheckForGPUInfoFile() bool
- func GetGPUInfo() ([]byte, error)
- type GPUManager
- type NvidiaGPUManager
- func (n *NvidiaGPUManager) GetDevices() []types.PlatformDevice
- func (n *NvidiaGPUManager) GetDriverVersion() string
- func (n *NvidiaGPUManager) GetGPUIDsUnsafe() []string
- func (n *NvidiaGPUManager) GetGPUMemoryMiBUnsafe() map[string]uint64
- func (n *NvidiaGPUManager) GetHasVGPU() bool
- func (n *NvidiaGPUManager) GetMpsControlBinaryPresent() bool
- func (n *NvidiaGPUManager) GetMpsServiceEnabled() bool
- func (n *NvidiaGPUManager) Initialize() error
- func (n *NvidiaGPUManager) SetDevices()
- func (n *NvidiaGPUManager) SetDriverVersion(version string)
- func (n *NvidiaGPUManager) SetGPUIDs(gpuIDs []string)
- func (n *NvidiaGPUManager) SetGPUMemoryMiB(gpuMemoryMiB map[string]uint64)
Constants ¶
const ( // GPUInfoDirPath is the directory where gpus and driver info are saved GPUInfoDirPath = "/var/lib/ecs/gpu" // NvidiaGPUInfoFilePath is the file path where gpus and driver info are saved NvidiaGPUInfoFilePath = GPUInfoDirPath + "/nvidia-gpu-info.json" )
Variables ¶
var GPUInfoFileExists = CheckForGPUInfoFile
var GetGPUInfoJSON = GetGPUInfo
Functions ¶
func CheckForGPUInfoFile ¶
func CheckForGPUInfoFile() bool
func GetGPUInfo ¶
Types ¶
type GPUManager ¶
type GPUManager interface {
Initialize() error
SetGPUIDs([]string)
GetGPUIDsUnsafe() []string
GetGPUMemoryMiBUnsafe() map[string]uint64
SetDevices()
GetDevices() []types.PlatformDevice
SetDriverVersion(string)
GetDriverVersion() string
GetMpsControlBinaryPresent() bool
GetMpsServiceEnabled() bool
GetHasVGPU() bool
}
GPUManager encompasses methods to get information on GPUs and their driver
func NewNvidiaGPUManager ¶
func NewNvidiaGPUManager() GPUManager
NewNvidiaGPUManager is used to obtain NvidiaGPUManager handle
type NvidiaGPUManager ¶
type NvidiaGPUManager struct {
DriverVersion string `json:"DriverVersion"`
GPUIDs []string `json:"GPUIDs"`
GPUDevices []types.PlatformDevice `json:"-"`
GPUMemoryMiB map[string]uint64 `json:"GPUMemoryMiB,omitempty"`
MpsControlBinaryPresent bool `json:"MpsControlBinaryPresent"`
MpsServiceEnabled bool `json:"MpsServiceEnabled"`
HasVGPU bool `json:"HasVGPU"`
// contains filtered or unexported fields
}
NvidiaGPUManager is used as a wrapper for NVML APIs and implements GPUManager interface
func (*NvidiaGPUManager) GetDevices ¶
func (n *NvidiaGPUManager) GetDevices() []types.PlatformDevice
GetDevices returns the GPU devices as PlatformDevices
func (*NvidiaGPUManager) GetDriverVersion ¶
func (n *NvidiaGPUManager) GetDriverVersion() string
GetDriverVersion is a getter for nvidia driver version
func (*NvidiaGPUManager) GetGPUIDsUnsafe ¶
func (n *NvidiaGPUManager) GetGPUIDsUnsafe() []string
GetGPUIDs returns the GPUIDs
func (*NvidiaGPUManager) GetGPUMemoryMiBUnsafe ¶
func (n *NvidiaGPUManager) GetGPUMemoryMiBUnsafe() map[string]uint64
GetGPUMemoryMiBUnsafe returns the per-GPU usable memory map
func (*NvidiaGPUManager) GetHasVGPU ¶
func (n *NvidiaGPUManager) GetHasVGPU() bool
GetHasVGPU reports whether any device on the instance is an NVIDIA vGPU slice.
func (*NvidiaGPUManager) GetMpsControlBinaryPresent ¶
func (n *NvidiaGPUManager) GetMpsControlBinaryPresent() bool
GetMpsControlBinaryPresent reports whether ecs-init found the MPS control binary on the host.
func (*NvidiaGPUManager) GetMpsServiceEnabled ¶
func (n *NvidiaGPUManager) GetMpsServiceEnabled() bool
GetMpsServiceEnabled reports whether ecs-init found nvidia-mps.service enabled on the host.
func (*NvidiaGPUManager) Initialize ¶
func (n *NvidiaGPUManager) Initialize() error
Initialize sets the fields of Nvidia GPU Manager struct
func (*NvidiaGPUManager) SetDevices ¶
func (n *NvidiaGPUManager) SetDevices()
func (*NvidiaGPUManager) SetDriverVersion ¶
func (n *NvidiaGPUManager) SetDriverVersion(version string)
SetDriverVersion is a setter for nvidia driver version
func (*NvidiaGPUManager) SetGPUIDs ¶
func (n *NvidiaGPUManager) SetGPUIDs(gpuIDs []string)
SetGPUIDs sets the GPUIDs
func (*NvidiaGPUManager) SetGPUMemoryMiB ¶
func (n *NvidiaGPUManager) SetGPUMemoryMiB(gpuMemoryMiB map[string]uint64)
SetGPUMemoryMiB sets the per-GPU usable memory map
Source Files
Directories