tpuproxy

Documentation

Overview

Package tpuproxy contains tpu backend driver proxy implementations and helper functions.

Index

• Variables
• func Filters() seccomp.SyscallRules
• func RegisterTPUv4Device(ctx context.Context, creds *auth.Credentials, root vfs.VirtualDentry, ...) error
• func RegisterTPUv5Device(vfsObj *vfs.VirtualFilesystem, devPath string, minorNum uint32) error

Variables

var (
	// TPUv4DeviceRegex is the regex for detecting TPUv4 device paths.
	TPUv4DeviceRegex = regexp.MustCompile(`/dev/accel(\d+)`)

	// TPUv5DeviceRegex is the regex for detecting TPUv5 device paths.
	TPUv5DeviceRegex = regexp.MustCompile(`/dev/vfio/(\d+)`)
)

Functions

func Filters

func Filters() seccomp.SyscallRules

Filters returns seccomp-bpf filters for this package.

func RegisterTPUv4Device

func RegisterTPUv4Device(ctx context.Context, creds *auth.Credentials, root vfs.VirtualDentry, vfsObj *vfs.VirtualFilesystem, devPath string, minorNum uint32) error

RegisterTPUv4Device registers the TPUv4 device with the provided minor number where the corresponding PCI device is located at pciPath. Accel devices always have their device file number set to their minor number.

func RegisterTPUv5Device

func RegisterTPUv5Device(vfsObj *vfs.VirtualFilesystem, devPath string, minorNum uint32) error

RegisterTPUv5Device registers the TPUv5 device with the provided device path and minor number.