Source of nvml-mock.go - NVIDIA GPU

Unified diff
Side-by-side diff
Ignore whitespace
Show whitespace characters
Hide e-diff
Color accessibility
Source viewDiff to previous
 
    Temperature    int
    Name           string
    MemoryInfo     *nvml.MemoryInfo
    MemoryInfoV2   *nvml.MemoryInfoV2
    PcieThroughput uint
    FanSpeed       uint
    PowerUsage     uint
    WantedErr      error
}
​
func (m *MockRunner) InitNVML() error {
func (m *MockRunner) Init() error {
    if m.WantedErr != nil {
        return m.WantedErr
    }
​
    if m.IsInit {
        return nvml.ErrAlreadyInitialized
    }
​
    m.IsInit = true
    return nil
}
​
func (m *MockRunner) InitNVMLv2() error {
func (m *MockRunner) InitV2() error {
    if m.WantedErr != nil {
        return m.WantedErr
    }
​
    if m.IsInit {
        return nvml.ErrAlreadyInitialized
    }
​
    m.IsInit = true
    return nil