detect

package

v0.0.0-...-3511abf Latest Latest Go to latest Published: Nov 2, 2023 License: Apache-2.0 Imports: 20 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/determined-ai/determined

Documentation ¶

Index ¶

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

Constants ¶

This section is empty.

Variables ¶

This section is empty.

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.

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL