Source of runner.go - NVIDIA GPU

Unified diff
Side-by-side diff
Ignore whitespace
Show whitespace characters
Hide e-diff
Color accessibility
Source viewDiff to previous
 
    count, ok := res.out[0].(uint)
    if !ok {
        m.t.Fatalf("expected uint in GetDeviceCountV2, got %T", res.out[0])
    }
​
    return count, err
}
​
// GetDeviceByIndexV2 is mock function.
//
//nolint:ireturn
//nolint:ireturn,nolintlint
func (m *MockRunner) GetDeviceByIndexV2(index uint) (nvml.Device, error) {
    m.t.Helper()
    res, err := m.handleFunctionCall("GetDeviceByIndexV2", index)
​
    if res.out[0] == nil {
        return nil, err
    }
​
    device, ok := res.out[0].(*MockDevice)
    if !ok {
        m.t.Fatalf("expected *MockRunner in GetDeviceByIndexV2, got %T", res.out[0])
    }
​
    device.t = m.t
​
    return device, err
}
​
// GetDeviceByUUID is mock function.
//
//nolint:ireturn
//nolint:ireturn,nolintlint
func (m *MockRunner) GetDeviceByUUID(uuid string) (nvml.Device, error) {
    m.t.Helper()
    res, err := m.handleFunctionCall("GetDeviceByUUID", uuid)
​
    if res.out[0] == nil {
        return nil, err
    }
​
    mockDevice, ok := res.out[0].(*MockDevice)
    if ok {