gce

package

v0.3.1 Latest Latest Go to latest Published: Apr 2, 2025 License: Apache-2.0 Imports: 7 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 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 GetInstance ¶

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

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