gpu

Documentation

Index

• type Client
  • func NewClient(t mockConstructorTestingTNewClient) *Client
  • func (_m *Client) GetAllocatableDevices(ctx context.Context) (gpu.DeviceList, gpu.Error)
  • func (_m *Client) GetDevices(ctx context.Context) (gpu.DeviceList, gpu.Error)
  • func (_m *Client) GetUsedDevices(ctx context.Context) (gpu.DeviceList, gpu.Error)

Types

type Client

type Client struct {
	mock.Mock
}

Client is an autogenerated mock type for the Client type

func NewClient

func NewClient(t mockConstructorTestingTNewClient) *Client

NewClient creates a new instance of Client. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.

func (*Client) GetAllocatableDevices

func (_m *Client) GetAllocatableDevices(ctx context.Context) (gpu.DeviceList, gpu.Error)

GetAllocatableDevices provides a mock function with given fields: ctx

func (*Client) GetDevices

func (_m *Client) GetDevices(ctx context.Context) (gpu.DeviceList, gpu.Error)

GetDevices provides a mock function with given fields: ctx

func (*Client) GetUsedDevices

func (_m *Client) GetUsedDevices(ctx context.Context) (gpu.DeviceList, gpu.Error)

GetUsedDevices provides a mock function with given fields: ctx