deviceinfo

package

v0.0.0-...-43f56ce Latest Latest Go to latest Published: Sep 19, 2025 License: Apache-2.0 Imports: 5 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

cos.googlesource.com/cos/tools

Links

Open Source Insights

Documentation ¶

Overview ¶

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

Constants ¶

This section is empty.

Variables ¶

View Source

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

Functions ¶

This section is empty.

Types ¶

type GPUType ¶

type GPUType int

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

func AvailableGPUTypesList ¶

func AvailableGPUTypesList(milestone string) ([]GPUType, error)

AvailableGPUTypesList returns a list of GPU devices supported based on the milestone

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

func (GPUType) SupportedArches ¶

func (g GPUType) SupportedArches() []string

TODO(gshuoy): b/331317222 - Add the arch support in the proto file. SupportedArches returns the list of supported architectures for the GPU type. Typically either "x86_64", "aarch64", or both (if future GPUs support both).

func (GPUType) SupportsArch ¶

func (g GPUType) SupportsArch(arch string) bool

Source Files ¶

View all Source files

gpu_devices.go

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