gce

package

v0.6.0 Latest Latest Go to latest Published: Jun 23, 2025 License: Apache-2.0 Imports: 10 Imported by: 0

Details

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

Repository

github.com/google/dranet

Links

Documentation ¶

Index ¶

Variables
func GetGCEAttributes(network, topology string) map[resourceapi.QualifiedName]resourceapi.DeviceAttribute
func GetInstance(ctx context.Context) (*cloudprovider.CloudInstance, error)
type GPUDirectSupport

Constants ¶

This section is empty.

Variables ¶

View Source

var (
	// https://cloud.google.com/compute/docs/accelerator-optimized-machines#network-protocol
	// machine types have a one to one mapping to a network protocol in google cloud
	NetworkProtocolMap = map[string]GPUDirectSupport{
		"a3-highgpu-1g":  GPUDirectTCPX,
		"a3-highgpu-2g":  GPUDirectTCPX,
		"a3-highgpu-4g":  GPUDirectTCPX,
		"a3-highgpu-8g":  GPUDirectTCPX,
		"a3-edgegpu-8g":  GPUDirectTCPX,
		"a3-megagpu-8g":  GPUDirectTCPXO,
		"a3-ultragpu-8g": GPUDirectRDMA,
		"a4-highgpu-8g":  GPUDirectRDMA,
	}
)

Functions ¶

func GetGCEAttributes ¶ added in v0.5.0

func GetGCEAttributes(network, topology string) map[resourceapi.QualifiedName]resourceapi.DeviceAttribute

GetGCEAttributes fetches all attributes related to the provided GCP network.

func GetInstance ¶

func GetInstance(ctx context.Context) (*cloudprovider.CloudInstance, error)

GetInstance retrieves GCE instance properties by querying the metadata server.

Types ¶

type GPUDirectSupport ¶

type GPUDirectSupport string

GPUDirectSupport represents the type of GPUDirect support for a given machine type.

const (
	GPUDirectTCPX  GPUDirectSupport = "GPUDirect-TCPX"
	GPUDirectTCPXO GPUDirectSupport = "GPUDirect-TCPXO"
	GPUDirectRDMA  GPUDirectSupport = "GPUDirect-RDMA"
)

Source Files ¶

View all Source files

gce.go

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