netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Support TPU v6e w/ gVisor.

Browse Source 
PiperOrigin-RevId: 687514246
This commit is contained in:
Jing Chen
2024-10-18 20:43:34 -07:00
committed by gVisor bot
parent 0b59173cff
commit 17edc7260e
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/abi/tpu/tpu.go
+3
View File
@@ -44,6 +44,9 @@ const (
// TPUV5pDeviceID is the PCI device ID of TPU V5p hardware.
TPUV5pDeviceID = 0x0062
// TPUV6eDeviceID is the PCI device ID of TPU V6e hardware.
TPUV6eDeviceID = 0x006f
)
// TPUV4InterruptsMap maps BAR indices to valid register offsets.
runsc/cmd/chroot.go
+1
View File
@@ -146,6 +146,7 @@ func tpuProxyUpdateChroot(hostRoot, chroot string, spec *specs.Spec, conf *confi
}
allowedDeviceIDs[tpu.TPUV5pDeviceID] = struct{}{}
allowedDeviceIDs[tpu.TPUV5eDeviceID] = struct{}{}
allowedDeviceIDs[tpu.TPUV6eDeviceID] = struct{}{}
}
if len(allowedDeviceIDs) == 0 {
paths, err = filepath.Glob(path.Join(hostRoot, "dev/accel*"))