utils

package

v1.4.0 Latest Latest Go to latest Published: Oct 4, 2025 License: Apache-2.0 Imports: 14 Imported by: 0

Details

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

Repository

github.com/ROCm/gpu-operator

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
Variables
func GetDefaultDriversVersion(node v1.Node) (string, error)
func GetDriverTypeNodeLabel(devConfig *v1alpha1.DeviceConfig) string
func GetDriverTypeTag(devCfg *amdv1alpha1.DeviceConfig) string
func GetDriverVersion(node v1.Node, deviceConfig amdv1alpha1.DeviceConfig) (string, error)
func HasNodeLabelKey(node v1.Node, labelKey string) bool
func HasNodeLabelTemplateMatch(nodeLabels map[string]string, template string) (bool, string, string, string)
func IsOpenShift(logger logr.Logger) bool
func IsPrometheusServiceMonitorEnable(devConfig *amdv1alpha1.DeviceConfig) bool
func RemoveOldNodeLabels(node *v1.Node) bool
func ShouldUseKMM(devConfig *v1alpha1.DeviceConfig) bool
func UbuntuDefaultDriverVersionsMapper(fullImageStr string) (string, error)
func UpdateDriverTypeNodeLabel(ctx context.Context, cli client.Client, devConfig *v1alpha1.DeviceConfig, ...) error

Constants ¶

View Source

const (
	KindDeviceConfig = "DeviceConfig"

	NodeFeatureLabelAmdGpu  = "feature.node.kubernetes.io/amd-gpu"
	NodeFeatureLabelAmdVGpu = "feature.node.kubernetes.io/amd-vgpu"
	// device plugin
	ResourceNamingStrategyFlag = "resource_naming_strategy"
	SingleStrategy             = "single"
	MixedStrategy              = "mixed"

	// kubevirt
	DriverTypeFlag          = "driver_type"
	DriverTypeContainer     = "container"
	DriverTypeVFPassthrough = "vf-passthrough"
	DriverTypePFPassthrough = "pf-passthrough"
	DefaultUtilsImage       = "docker.io/rocm/gpu-operator-utils:latest"
	// workerMgr related labels
	LoadVFIOAction              = "loadVFIO"
	UnloadVFIOAction            = "unloadVFIO"
	WorkerActionLabelKey        = "gpu.operator.amd.com/worker-action"
	VFIOMountReadyLabelTemplate = "gpu.operator.amd.com/%v.%v.vfio.ready"
	DriverTypeNodeLabelTemplate = "gpu.operator.amd.com/%v.%v.driver"
	KMMModuleReadyLabelTemplate = "kmm.node.kubernetes.io/%v.%v.ready"
	// Operand metadata
	MetricsExporterNameSuffix = "-metrics-exporter"
	TestRunnerNameSuffix      = "-test-runner"
	DevicePluginNameSuffix    = "-device-plugin"
	NodeLabellerNameSuffix    = "-node-labeller"
)

Variables ¶

View Source

var (

	// kubevirt
	DefaultVFDeviceIDs = []string{
		"7410",
		"74b5",
		"74b9",
		"7461",
		"73ae",
	}
	DefaultPFDeviceIDs = []string{
		"74a5",
		"74a2",
		"74b6",
		"74a8",
		"74a0",
		"74a1",
		"74a9",
		"74bd",
		"740f",
		"7408",
		"740c",
	}
)

Functions ¶

func GetDefaultDriversVersion ¶

func GetDefaultDriversVersion(node v1.Node) (string, error)

func GetDriverTypeNodeLabel ¶ added in v1.4.0

func GetDriverTypeNodeLabel(devConfig *v1alpha1.DeviceConfig) string

func GetDriverTypeTag ¶ added in v1.4.0

func GetDriverTypeTag(devCfg *amdv1alpha1.DeviceConfig) string

func GetDriverVersion ¶

func GetDriverVersion(node v1.Node, deviceConfig amdv1alpha1.DeviceConfig) (string, error)

func HasNodeLabelKey ¶ added in v1.2.2

func HasNodeLabelKey(node v1.Node, labelKey string) bool

func HasNodeLabelTemplateMatch ¶ added in v1.4.0

func HasNodeLabelTemplateMatch(nodeLabels map[string]string, template string) (bool, string, string, string)

func IsOpenShift ¶ added in v1.2.2

func IsOpenShift(logger logr.Logger) bool

func IsPrometheusServiceMonitorEnable ¶ added in v1.2.2

func IsPrometheusServiceMonitorEnable(devConfig *amdv1alpha1.DeviceConfig) bool

IsPrometheusServiceMonitorEnable checks if the Prometheus ServiceMonitor is enabled in the DeviceConfig

func RemoveOldNodeLabels ¶ added in v1.2.2

func RemoveOldNodeLabels(node *v1.Node) bool

func ShouldUseKMM ¶ added in v1.4.0

func ShouldUseKMM(devConfig *v1alpha1.DeviceConfig) bool

ShouldUseKMM return true if KMM needs to be triggered otherwise return false

func UbuntuDefaultDriverVersionsMapper ¶

func UbuntuDefaultDriverVersionsMapper(fullImageStr string) (string, error)

func UpdateDriverTypeNodeLabel ¶ added in v1.4.0

func UpdateDriverTypeNodeLabel(ctx context.Context, cli client.Client, devConfig *v1alpha1.DeviceConfig, nodes *v1.NodeList, cleanup bool) error

Types ¶

This section is empty.

Source Files ¶

View all Source files

utils.go

Directories ¶

Path	Synopsis
client Code generated by MockGen.	Code generated by MockGen.
cmd
conditions
config
configmanager Code generated by MockGen.	Code generated by MockGen.
controllers Code generated by MockGen.	Code generated by MockGen.
watchers Code generated by MockGen.	Code generated by MockGen.
workermgr Code generated by MockGen.	Code generated by MockGen.
kmmmodule Code generated by MockGen.	Code generated by MockGen.
metricsexporter Code generated by MockGen.	Code generated by MockGen.
nodelabeller Code generated by MockGen.	Code generated by MockGen.
test
testrunner Code generated by MockGen.	Code generated by MockGen.
validator Code generated by MockGen.	Code generated by MockGen.

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