kubernetes/test/e2e/node/gpu.go

var _ …

var _ …

var _ …

func createAndValidatePod(ctx context.Context, f *framework.Framework, podClient *e2epod.PodClient, pod *v1.Pod) { … }

func testNvidiaCLIPod() *v1.Pod { … }

func testMatrixMultiplicationPod() *v1.Pod { … }

func SetupEnvironmentAndSkipIfNeeded(ctx context.Context, f *framework.Framework, clientSet clientset.Interface) { … }

func areGPUsAvailableOnAllSchedulableNodes(ctx context.Context, clientSet clientset.Interface) bool { … }

func logOSImages(ctx context.Context, f *framework.Framework) { … }

const driverInstallTimeout …

// SetupNVIDIAGPUNode install Nvidia Drivers and wait for Nvidia GPUs to be available on nodes
func SetupNVIDIAGPUNode(ctx context.Context, f *framework.Framework) { … }

func waitForGPUs(ctx context.Context, f *framework.Framework, namespace, name string) { … }

// StartJob starts a simple CUDA job that requests gpu and the specified number of completions
func StartJob(ctx context.Context, f *framework.Framework, completions int32) { … }

func podNames(pods []v1.Pod) []string { … }

// VerifyJobNCompletions verifies that the job has completions number of successful pods
func VerifyJobNCompletions(ctx context.Context, f *framework.Framework, completions int32) { … }