gpu

Documentation

Overview

Package gpu implements GPU plugin.

Index

• type Config
• type Tester
  • func New(cfg Config) Tester

Types

type Config

type Config struct {
	Logger    *zap.Logger
	LogWriter io.Writer
	Stopc     chan struct{}
	EKSConfig *eksconfig.Config
	K8SClient k8s_client.EKS
}

Config defines GPU configuration.

type Tester

type Tester interface {
	// Name returns the name of the tester.
	Name() string
	// InstallNvidiaDriver installs the Nvidia device plugin for Kubernetes.
	// After GPU worker nodes join the cluster, one must apply the Nvidia
	// device plugin for Kubernetes as a DaemonSet.
	// ref. https://docs.aws.amazon.com/eks/latest/userguide/create-managed-node-group.html
	// ref. https://docs.aws.amazon.com/eks/latest/userguide/gpu-ami.html
	// ref. https://github.com/NVIDIA/k8s-device-plugin
	InstallNvidiaDriver() error
	// CreateNvidiaSMI launches a pod manifest that launches a Cuda container that
	// runs "nvidia-smi" on a GPU worker node.
	// ref. https://docs.aws.amazon.com/eks/latest/userguide/gpu-ami.html
	CreateNvidiaSMI() error
}

Tester defines GPU tester.

func New

func New(cfg Config) Tester

New creates a new Job tester.