assert

Documentation

Index

• func AssertMigConfig(c *Context) error
• func AssertMigMode(c *Context) error
• func BuildCommand() *cli.Command
• func CheckFlags(f *Flags) error
• func GetLogger() *logrus.Logger
• func GetSelectedMigConfig(f *Flags, spec *v1.Spec) (v1.MigConfigSpecSlice, error)
• func ParseConfigFile(f *Flags) (*v1.Spec, error)
• func WalkSelectedMigConfigForEachGPU(migConfig v1.MigConfigSpecSlice, ...) error
• type Context
• type Flags

Functions

func AssertMigConfig

func AssertMigConfig(c *Context) error

func AssertMigMode

func AssertMigMode(c *Context) error

func BuildCommand

func BuildCommand() *cli.Command

func CheckFlags

func CheckFlags(f *Flags) error

func GetLogger

func GetLogger() *logrus.Logger

func GetSelectedMigConfig

func GetSelectedMigConfig(f *Flags, spec *v1.Spec) (v1.MigConfigSpecSlice, error)

func ParseConfigFile

func ParseConfigFile(f *Flags) (*v1.Spec, error)

func WalkSelectedMigConfigForEachGPU

func WalkSelectedMigConfigForEachGPU(migConfig v1.MigConfigSpecSlice, f func(*v1.MigConfigSpec, int, types.DeviceID) error) error

Types

type Context

type Context struct {
	*cli.Context
	Flags     *Flags
	MigConfig v1.MigConfigSpecSlice
	Nvml      nvml.Interface
}

type Flags

type Flags struct {
	ConfigFile     string
	SelectedConfig string
	SkipReset      bool
	ModeOnly       bool
	ValidConfig    bool
}