deviceinfo

Documentation

Overview

Package deviceinfo provides the devices information for cos-gpu-installer

Index

• Variables
• type GPUType
  • func GetGPUTypeInfo() (GPUType, error)
  • func ParseGPUType(gpu string) (GPUType, error)
  • func (g GPUType) OpenSupported() bool
  • func (g GPUType) String() string

Variables

var AllGPUTypeStrings = []string{
	"NVIDIA_TESLA_K80",
	"NVIDIA_TESLA_P4",
	"NVIDIA_TESLA_P100",
	"NVIDIA_TESLA_V100",
	"NVIDIA_L4",
	"NVIDIA_H100_80GB",
	"NVIDIA_TESLA_A100",
	"NVIDIA_A100_80GB",
	"NVIDIA_TESLA_T4",
}

var AvailableGPUTypesList = []GPUType{
	P4,
	P100,
	V100,
	L4,
	T4,
	H100,
	A100_40GB,
	A100_80GB,
}

Types

type GPUType

type GPUType int

const (
	K80 GPUType = iota
	P4
	P100
	V100
	L4
	T4
	H100
	A100_40GB
	A100_80GB
	NO_GPU
	Others
)

func GetGPUTypeInfo

func GetGPUTypeInfo() (GPUType, error)

TODO(gshuoy): b/331317222 - add the pci id in the proto file.

func ParseGPUType

func ParseGPUType(gpu string) (GPUType, error)

ParseGPUType converts a string to a GPUType enum.

func (GPUType) OpenSupported

func (g GPUType) OpenSupported() bool

TODO(gshuoy): b/331317222 - Add the open source supported in the proto file.

func (GPUType) String

func (g GPUType) String() string