detect

Documentation

Index

• func Detect(slotType, agentID, visibleGPUs string, artificialSlots int) ([]device.Device, error)
• type RocmDevice
  • func GetRocmDeviceByUUID(uuid string) *RocmDevice

Functions

func Detect

func Detect(slotType, agentID, visibleGPUs string, artificialSlots int) ([]device.Device, error)

Detect the devices available. If artificial devices are configured, prefers those, otherwise, we detect cuda, rocm, cpu (or no) devices based on the configured slot type.

Types

type RocmDevice

type RocmDevice struct {
	UUID       string `json:"Unique ID"`
	CardSKU    string `json:"Card SKU"`
	CardVendor string `json:"Card vendor"`
	CardModel  string `json:"Card model"`
	PCIBus     string `json:"PCI Bus"`
	Index      int
}

RocmDevice metadata.

func GetRocmDeviceByUUID

func GetRocmDeviceByUUID(uuid string) *RocmDevice

GetRocmDeviceByUUID gets a RocmDevice by UUID from the singleton discovered Rocm devices.