nvproxy

Documentation

Overview

Package nvproxy implements proxying for the Nvidia GPU Linux kernel driver: https://github.com/NVIDIA/open-gpu-kernel-modules.

Supported Nvidia GPUs: T4, L4, A100, A10G and H100.

Index

• func ExpectedDriverChecksum(version DriverVersion) (string, bool)
• func Filters() seccomp.SyscallRules
• func ForEachSupportDriver(f func(version DriverVersion, checksum string))
• func HostDriverVersion() (string, error)
• func Init()
• func Register(vfsObj *vfs.VirtualFilesystem, versionStr string, uvmDevMajor uint32) error
• func SupportedIoctls(version DriverVersion) (frontendIoctls map[uint32]struct{}, uvmIoctls map[uint32]struct{}, ...)
• type DriverVersion
  • func DriverVersionFrom(version string) (DriverVersion, error)
  • func LatestDriver() DriverVersion
  • func NewDriverVersion(major, minor, patch int) DriverVersion
  • func (v DriverVersion) Equals(other DriverVersion) bool
  • func (v DriverVersion) String() string

Functions

func ExpectedDriverChecksum

func ExpectedDriverChecksum(version DriverVersion) (string, bool)

ExpectedDriverChecksum returns the expected checksum for a given version. Precondition: Init() must have been called.

func Filters

func Filters() seccomp.SyscallRules

Filters returns seccomp-bpf filters for this package.

func ForEachSupportDriver

func ForEachSupportDriver(f func(version DriverVersion, checksum string))

ForEachSupportDriver calls f on all supported drivers. Precondition: Init() must have been called.

func HostDriverVersion

func HostDriverVersion() (string, error)

HostDriverVersion returns the version of the host Nvidia driver.

func Init

func Init()

Init initializes abis global map.

func Register

func Register(vfsObj *vfs.VirtualFilesystem, versionStr string, uvmDevMajor uint32) error

Register registers all devices implemented by this package in vfsObj.

func SupportedIoctls

func SupportedIoctls(version DriverVersion) (frontendIoctls map[uint32]struct{}, uvmIoctls map[uint32]struct{}, controlCmds map[uint32]struct{}, allocClasses map[uint32]struct{}, ok bool)

SupportedIoctls returns the ioctl numbers that are supported by nvproxy at a given version.

Types

type DriverVersion

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

DriverVersion represents a NVIDIA driver version patch release.

+stateify savable

func DriverVersionFrom

func DriverVersionFrom(version string) (DriverVersion, error)

DriverVersionFrom returns a DriverVersion from a string.

func LatestDriver

func LatestDriver() DriverVersion

LatestDriver returns the latest supported driver. Precondition: Init() must have been called.

func NewDriverVersion

func NewDriverVersion(major, minor, patch int) DriverVersion

NewDriverVersion returns a new driver version.

func (DriverVersion) Equals

func (v DriverVersion) Equals(other DriverVersion) bool

Equals returns true if the two driver versions are equal.

func (DriverVersion) String

func (v DriverVersion) String() string