common

package

v0.2.0 Latest Latest Go to latest Published: Mar 23, 2025 License: Apache-2.0 Imports: 8 Imported by: 0

Details

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

Repository

github.com/NVIDIA/KAI-scheduler

Links

Documentation ¶

Index ¶

Constants
func AddDirectEnvVarsConfigMapSource(container *v1.Container, directEnvVarsMapName string)
func AddEnvVarToContainer(container *v1.Container, envVar v1.EnvVar)
func AddVisibleDevicesEnvVars(container *v1.Container, sharedGpuConfigMapName string)
func GetConfigMapName(pod *v1.Pod, container *v1.Container) (string, error)
func GetConfigVolumeName(configMapName string) string
func GetDirectEnvVarConfigMapName(pod *v1.Pod, container *v1.Container) (string, error)
func IsSharedGPUAllocation(bindRequest *v1alpha2.BindRequest) bool
func SetConfigMapVolume(pod *v1.Pod, configMapName string)
func SetNvidiaVisibleDevices(ctx context.Context, kubeClient client.Client, pod *v1.Pod, ...) error
func SetRunaiNumOfGPUDevices(ctx context.Context, kubeClient client.Client, pod *v1.Pod, ...) error
func UpdateConfigMapEnvironmentVariable(ctx context.Context, kubeclient client.Client, task *v1.Pod, ...) error

Constants ¶

View Source

const (
	NvidiaVisibleDevices = "NVIDIA_VISIBLE_DEVICES"
	RunaiVisibleDevices  = "RUNAI-VISIBLE-DEVICES"
	RunaiNumOfGpus       = "RUNAI_NUM_OF_GPUS"
	ReceivedTypeFraction = "Fraction"
	ReceivedTypeRegular  = "Regular"
)

Variables ¶

This section is empty.

Functions ¶

func AddDirectEnvVarsConfigMapSource ¶

func AddDirectEnvVarsConfigMapSource(container *v1.Container, directEnvVarsMapName string)

func AddEnvVarToContainer ¶

func AddEnvVarToContainer(container *v1.Container, envVar v1.EnvVar)

func AddVisibleDevicesEnvVars ¶

func AddVisibleDevicesEnvVars(container *v1.Container, sharedGpuConfigMapName string)

func GetConfigMapName ¶

func GetConfigMapName(pod *v1.Pod, container *v1.Container) (string, error)

func GetConfigVolumeName ¶

func GetConfigVolumeName(configMapName string) string

func GetDirectEnvVarConfigMapName ¶

func GetDirectEnvVarConfigMapName(pod *v1.Pod, container *v1.Container) (string, error)

func IsSharedGPUAllocation ¶

func IsSharedGPUAllocation(bindRequest *v1alpha2.BindRequest) bool

func SetConfigMapVolume ¶

func SetConfigMapVolume(pod *v1.Pod, configMapName string)

func SetNvidiaVisibleDevices ¶

func SetNvidiaVisibleDevices(ctx context.Context, kubeClient client.Client, pod *v1.Pod,
	container *v1.Container, visibleDevicesValue string) error

func SetRunaiNumOfGPUDevices ¶

func SetRunaiNumOfGPUDevices(
	ctx context.Context, kubeClient client.Client, pod *v1.Pod, container *v1.Container, numOfGPUs string,
) error

func UpdateConfigMapEnvironmentVariable ¶

func UpdateConfigMapEnvironmentVariable(
	ctx context.Context, kubeclient client.Client, task *v1.Pod,
	configMapName string, changesFunc func(map[string]string) error,
) error

Types ¶

This section is empty.

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
gpusharingconfigmap

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