nvkind

Documentation

Index

• func GetClusterNames() (sets.Set[string], error)
• type Cluster
  • func NewCluster(opts ...ClusterOption) (*Cluster, error)
  • func (c *Cluster) Create(opts ...ClusterCreateOption) error
  • func (c *Cluster) Delete() error
  • func (c *Cluster) GetNodes() ([]Node, error)
  • func (c *Cluster) PrintClusterInfo() error
  • func (c *Cluster) RegisterNvidiaRuntimeClass() error
• type ClusterCreateOption
  • func WithRetain() ClusterCreateOption
  • func WithWait(wait time.Duration) ClusterCreateOption
• type ClusterCreateOptions
• type ClusterOption
  • func WithConfig(config *Config) ClusterOption
  • func WithKubeConfig(kubeconfig string) ClusterOption
  • func WithName(name string) ClusterOption
• type ClusterOptions
• type Config
  • func NewConfig(opts ...ConfigOption) (*Config, error)
• type ConfigOption
  • func WithConfigTemplate[T string | []byte](arg T) ConfigOption
  • func WithConfigValues[T string | []byte](arg T) ConfigOption
  • func WithDefaultName(name string) ConfigOption
  • func WithFuncMap(funcMap template.FuncMap) ConfigOption
  • func WithImage(image string) ConfigOption
  • func WithNvml(nvml nvml.Interface) ConfigOption
  • func WithOutput(stdout, stderr io.Writer) ConfigOption
• type ConfigOptions
• type GPUInfo
• type Node
  • func (n *Node) ConfigureContainerRuntime() error
  • func (n *Node) GetGPUInfo() ([]GPUInfo, error)
  • func (n *Node) HasGPUs() bool
  • func (n *Node) InstallContainerToolkit() error
  • func (n *Node) PatchProcDriverNvidia() error

Functions

func GetClusterNames

func GetClusterNames() (sets.Set[string], error)

Types

type Cluster

type Cluster struct {
	Name string
	// contains filtered or unexported fields
}

func NewCluster

func NewCluster(opts ...ClusterOption) (*Cluster, error)

func (*Cluster) Create

func (c *Cluster) Create(opts ...ClusterCreateOption) error

func (*Cluster) Delete

func (c *Cluster) Delete() error

Note(JP): not exposed via CLI, we may want to remove this for now.

func (*Cluster) GetNodes

func (c *Cluster) GetNodes() ([]Node, error)

func (*Cluster) PrintClusterInfo

func (c *Cluster) PrintClusterInfo() error

func (*Cluster) RegisterNvidiaRuntimeClass

func (c *Cluster) RegisterNvidiaRuntimeClass() error

type ClusterCreateOption

type ClusterCreateOption func(*ClusterCreateOptions)

func WithRetain

func WithRetain() ClusterCreateOption

func WithWait

func WithWait(wait time.Duration) ClusterCreateOption

type ClusterCreateOptions

type ClusterCreateOptions struct {
	// contains filtered or unexported fields
}

type ClusterOption

type ClusterOption func(*ClusterOptions)

func WithConfig

func WithConfig(config *Config) ClusterOption

func WithKubeConfig

func WithKubeConfig(kubeconfig string) ClusterOption

func WithName

func WithName(name string) ClusterOption

type ClusterOptions

type ClusterOptions struct {
	// contains filtered or unexported fields
}

type Config

type Config struct {
	*kind.Cluster
	// contains filtered or unexported fields
}

func NewConfig

func NewConfig(opts ...ConfigOption) (*Config, error)

type ConfigOption

type ConfigOption func(*ConfigOptions)

func WithConfigTemplate

func WithConfigTemplate[T string | []byte](arg T) ConfigOption

func WithConfigValues

func WithConfigValues[T string | []byte](arg T) ConfigOption

func WithDefaultName

func WithDefaultName(name string) ConfigOption

func WithFuncMap

func WithFuncMap(funcMap template.FuncMap) ConfigOption

func WithImage

func WithImage(image string) ConfigOption

func WithNvml

func WithNvml(nvml nvml.Interface) ConfigOption

func WithOutput

func WithOutput(stdout, stderr io.Writer) ConfigOption

type ConfigOptions

type ConfigOptions struct {
	// contains filtered or unexported fields
}

type GPUInfo

type GPUInfo struct {
	Index string
	Name  string
	UUID  string
}

type Node

type Node struct {
	Name string
	// contains filtered or unexported fields
}

func (*Node) ConfigureContainerRuntime

func (n *Node) ConfigureContainerRuntime() error

func (*Node) GetGPUInfo

func (n *Node) GetGPUInfo() ([]GPUInfo, error)

func (*Node) HasGPUs

func (n *Node) HasGPUs() bool

func (*Node) InstallContainerToolkit

func (n *Node) InstallContainerToolkit() error

func (*Node) PatchProcDriverNvidia

func (n *Node) PatchProcDriverNvidia() error