gpu

type CheckGpuStatus struct {
	common.KubeAction
}

type CheckWslGPU struct {
}

type ConfigureContainerdRuntime struct {
	common.KubeAction
}

type ContainerdInstalled struct {
	common.KubePrepare
}

type CudaInstalled struct {
	common.KubePrepare
}

type CudaNotInstalled struct {
	common.KubePrepare
	CudaInstalled
}

type CurrentNodeInK8s struct {
	common.KubePrepare
}

type DisableNouveauModule struct {
	common.KubeModule
}

type GPUEnablePrepare struct {
	common.KubePrepare
}

type GpuDevicePluginInstalled struct {
	common.KubePrepare
}

type InstallContainerToolkitModule struct {
	common.KubeModule
	manifest.ManifestModule
	Skip          bool // enableGPU && ubuntuVersionSupport
	SkipCudaCheck bool
}

type InstallCudaDriver struct {
	common.KubeAction
	manifest.ManifestAction
}

type InstallDriversModule struct {
	common.KubeModule
	manifest.ManifestModule
	Skip bool // enableGPU && ubuntuVersionSupport

	// log a failure message and then exit
	// instead of silently skip the jobs when:
	// 1. no card is found (which skips the driver installation)
	// 2. no driver is found (which skips the container toolkit installation)
	FailOnNoInstallation bool
}

type InstallNvidiaContainerToolkit struct {
	common.KubeAction
}

type InstallPlugin struct {
	common.KubeAction
}

type InstallPluginModule struct {
	common.KubeModule
	Skip bool // enableGPU && ubuntuVersionSupport
}

type NodeLabelingModule struct {
	common.KubeModule
}

type NodeUnlabelingModule struct {
	common.KubeModule
}

type NvidiaGraphicsCard struct {
	common.KubePrepare
	ExitOnNotFound bool
}

type PatchK3sDriver struct {
	common.KubeAction
}

type PrintGpuStatus struct {
	common.KubeAction
}

type PrintPluginsStatus struct {
	common.KubeAction
}

type RemoveContainerRuntimeConfig struct {
	common.KubeAction
}

type RemoveNodeLabels struct {
	common.KubeAction
}

type RestartContainerdModule struct {
	common.KubeModule
	Skip bool // enableGPU && ubuntuVersionSupport
}

type RestartK3sServiceModule struct {
	common.KubeModule
	Skip bool // enableGPU && ubuntuVersionSupport
}

type RestartPlugin struct {
	common.KubeAction
}

type UninstallCudaModule struct {
	common.KubeModule
}

type UninstallNvidiaDrivers struct {
	common.KubeAction
}

type UpdateNodeGPUInfo struct {
	common.KubeAction
}

type UpdateNvidiaContainerToolkitSource struct {
	common.KubeAction
	manifest.ManifestAction
}

type WriteNouveauBlacklist struct {
	common.KubeAction
}