diff --git a/pkg/abi/tpu/tpu.go b/pkg/abi/tpu/tpu.go index 65a6e99e57..5b104e1d67 100644 --- a/pkg/abi/tpu/tpu.go +++ b/pkg/abi/tpu/tpu.go @@ -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. diff --git a/runsc/cmd/chroot.go b/runsc/cmd/chroot.go index 5912987a75..00f0cf0a58 100644 --- a/runsc/cmd/chroot.go +++ b/runsc/cmd/chroot.go @@ -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*"))