utils

package

v1.4.1 Latest Latest Go to latest Published: Jun 16, 2025 License: Apache-2.0, Apache-2.0 Imports: 15 Imported by: 0

Documentation ¶

Index ¶

Constants
Variables
func GetAKSGPUImageSHA(size string) string
func GetGPUDriverType(size string) string
func GetGPUDriverVersion(size string) string
func GetMaxPods(nodeClass *v1beta1.AKSNodeClass, networkPlugin, networkPluginMode string) int32
func GetSubnetResourceID(subscriptionID, resourceGroupName, virtualNetworkName, subnetName string) string
func GetVMName(providerID string) (string, error)
func GetVnetSubnetIDComponents(vnetSubnetID string) (vnetSubnetResource, error)
func GetZone(vm *armcompute.VirtualMachine) (string, error)
func ImageReferenceToString(imageRef *armcompute.ImageReference) string
func IsMarinerEnabledGPUSKU(vmSize string) bool
func IsNvidiaEnabledSKU(vmSize string) bool
func IsVMDeleting(vm armcompute.VirtualMachine) bool
func MakeVMZone(zone string) []*string
func MakeZone(location string, zoneID string) string
func PrettySlice[T any](s []T, maxItems int) string
func ResourceIDToProviderID(ctx context.Context, id string) string
func StringMap(list v1.ResourceList) map[string]string
func UseGridDrivers(size string) bool
func WithDefaultFloat64(key string, def float64) float64
type NvidiaSKUConfig

Constants ¶

const (
	Nvidia470CudaDriverVersion = "470.82.01"

	// https://github.com/Azure/AgentBaker/blob/c0e684e5cecebcf61554cc7d2e2d2191972d35ed/parts/common/components.json#L797-L811
	NvidiaCudaDriverVersion = "550.144.03"
	AKSGPUCudaVersionSuffix = "20250328201547"

	NvidiaGridDriverVersion = "550.144.06"
	AKSGPUGridVersionSuffix = "20250512225043"
)

TODO: Get these from agentbaker

Variables ¶

View Source

var ConvergedGPUDriverSizes = map[string]bool{
	"standard_nv6ads_a10_v5":   true,
	"standard_nv12ads_a10_v5":  true,
	"standard_nv18ads_a10_v5":  true,
	"standard_nv36ads_a10_v5":  true,
	"standard_nv72ads_a10_v5":  true,
	"standard_nv36adms_a10_v5": true,
	"standard_nc8ads_a10_v4":   true,
	"standard_nc16ads_a10_v4":  true,
	"standard_nc32ads_a10_v4":  true,
}

ConvergedGPUDriverSizes : these sizes use a "converged" driver to support both cuda/grid workloads.

how do you figure this out? ask HPC or find out by trial and error. installing vanilla cuda drivers will fail to install with opaque errors. see https://github.com/Azure/azhpc-extensions/blob/daaefd78df6f27012caf30f3b54c3bd6dc437652/NvidiaGPU/resources.json

Functions ¶

func GetAKSGPUImageSHA ¶

func GetAKSGPUImageSHA(size string) string

func GetGPUDriverType ¶ added in v0.5.5

func GetGPUDriverType(size string) string

GetGPUDriverType returns the type of GPU driver for given VM SKU ("grid" or "cuda")

func GetGPUDriverVersion ¶

func GetGPUDriverVersion(size string) string

NV series GPUs target graphics workloads vs NC which targets compute. they typically use GRID, not CUDA drivers, and will fail to install CUDA drivers. NVv1 seems to run with CUDA, NVv5 requires GRID. NVv3 is untested on AKS, NVv4 is AMD so n/a, and NVv2 no longer seems to exist (?).

func GetMaxPods ¶ added in v0.7.5

func GetMaxPods(nodeClass *v1beta1.AKSNodeClass, networkPlugin, networkPluginMode string) int32

GetMaxPods resolves what we should set max pods to for a given nodeclass. If not specified, defaults based on network-plugin. 30 for "azure", 110 for "kubenet", or 250 for "none" and network plugin mode overlay.

func GetSubnetResourceID ¶ added in v0.4.0

func GetSubnetResourceID(subscriptionID, resourceGroupName, virtualNetworkName, subnetName string) string

GetSubnetResourceID constructs the subnet resource id

func GetVMName ¶

func GetVMName(providerID string) (string, error)

GetVMName parses the provider ID stored on the node to get the vmName associated with a node

func GetVnetSubnetIDComponents ¶ added in v0.4.0

func GetVnetSubnetIDComponents(vnetSubnetID string) (vnetSubnetResource, error)

func GetZone ¶ added in v0.6.1

func GetZone(vm *armcompute.VirtualMachine) (string, error)

GetZone returns the zone for the given virtual machine, or an empty string if there is no zone specified

func ImageReferenceToString ¶ added in v0.7.0

func ImageReferenceToString(imageRef *armcompute.ImageReference) string

func IsMarinerEnabledGPUSKU ¶

func IsMarinerEnabledGPUSKU(vmSize string) bool

IsNvidiaEnabledSKU determines if an VM SKU has nvidia driver support

func IsNvidiaEnabledSKU ¶

func IsNvidiaEnabledSKU(vmSize string) bool

IsNvidiaEnabledSKU determines if an VM SKU has nvidia driver support

func IsVMDeleting ¶ added in v0.7.0

func IsVMDeleting(vm armcompute.VirtualMachine) bool

func MakeVMZone ¶ added in v0.6.1

func MakeVMZone(zone string) []*string

VM Zones field expects just the zone number, without region

func MakeZone ¶ added in v0.6.1

func MakeZone(location string, zoneID string) string

MakeZone returns the zone value in format of <region>-<zone-id>.

func PrettySlice ¶ added in v0.7.0

func PrettySlice[T any](s []T, maxItems int) string

PrettySlice truncates a slice after a certain number of max items to ensure that the Slice isn't too long

func ResourceIDToProviderID ¶

func ResourceIDToProviderID(ctx context.Context, id string) string

func StringMap ¶ added in v0.7.0

func StringMap(list v1.ResourceList) map[string]string

StringMap returns the string map representation of the resource list

func UseGridDrivers ¶ added in v0.5.5

func UseGridDrivers(size string) bool

func WithDefaultFloat64 ¶ added in v0.7.0

func WithDefaultFloat64(key string, def float64) float64

WithDefaultFloat64 returns the float64 value of the supplied environment variable or, if not present, the supplied default value. If the float64 conversion fails, returns the default

Types ¶

type NvidiaSKUConfig ¶ added in v0.5.5

type NvidiaSKUConfig struct {
	NvidiaEnabledSKUFamilies        map[string][]string `yaml:"nvidiaEnabledSKUs"`
	MarinerNvidiaEnabledSKUFamilies map[string][]string `yaml:"marinerNvidiaEnabledSKUs"`
}

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
opts
project

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