device

Documentation

Index

• type Device
  • func (d *Device) Proto() *devicev1.Device
  • func (d *Device) String() string
• type ID
• type Type
  • func (t Type) Proto() devicev1.Type

Types

type Device

type Device struct {
	ID    ID     `json:"id"`
	Brand string `json:"brand"`
	UUID  string `json:"uuid"`
	Type  Type   `json:"type"`
}

Device represents a single computational device on an agent.

func (*Device) Proto

func (d *Device) Proto() *devicev1.Device

Proto returns the proto representation of the device.

func (*Device) String

func (d *Device) String() string

type ID

type ID int

ID the type of Device.ID.

type Type

type Type string

Type is a string holding the type of the Device.

const (
	// CPU represents a CPU device.
	CPU Type = "cpu"
	// CUDA represents a CUDA device.
	CUDA Type = "cuda"
	// ROCM represents an AMD GPU device.
	ROCM Type = "rocm"
	// ZeroSlot represents cpu devices on agents where only GPUs are modeled.
	ZeroSlot Type = ""
)

func (Type) Proto

func (t Type) Proto() devicev1.Type

Proto returns the proto representation of the device type.