6 Commits
Author SHA1 Message Date
Lucas ManningandgVisor bot 129d4b63a7 Add support for more TPU devices.
PiperOrigin-RevId: 737696637
2025-03-17 11:53:29 -07:00
Jing ChenandgVisor bot 17edc7260e Support TPU v6e w/ gVisor.
PiperOrigin-RevId: 687514246
2024-10-18 20:43:34 -07:00
Jing Chen a093ad0450 Simplify and format gVisor codebase.
The changes are just output of `gofmt -s -w .`.
2024-10-13 00:50:32 -07:00
Lucas ManningandgVisor bot bbbecc35cc Add support for v5pod and fix TPU v5 bugs.
V5 support was broken in a few different ways:
- tpu device files (/dev/vfio/X) were created with incorrect minor device nums.
- PCI paths on bus' other than 0000:00 were not supported.
- VFIO unmap was broken and not properly added to the seccomp allowlist.
- The VFIO main device file (/dev/vfio/vfio) did not account for overlapping
  device address ranges that correspond to different VFIO container groups.

Previously TPU support was tested on machines with single TPUs, which masked
most of these issues.

All these issues should be fixed by this change. Tested manually on GKE.

PiperOrigin-RevId: 656037853
2024-07-25 12:12:30 -07:00
Jing ChenandgVisor bot e184849c0c Add function to extract TPU v5's minor device number.
PiperOrigin-RevId: 583486707
2023-11-17 14:26:59 -08:00
Lucas ManningandgVisor bot 44d987d984 Add sanitization checks for TPU ioctl parameters and rename TPU device methods.
The methods are renamed so that it's more clear they are explicitly for TPUV4,
not any arbitrary accel device.

PiperOrigin-RevId: 577317824
2023-10-27 15:26:45 -07:00