diff --git a/pkg/abi/tpu/tpu.go b/pkg/abi/tpu/tpu.go index e31c647331..bc96a6c620 100644 --- a/pkg/abi/tpu/tpu.go +++ b/pkg/abi/tpu/tpu.go @@ -51,11 +51,11 @@ const ( // TPUV6ePFDeviceID is PCI device ID of TPU V6e hardware PF. TPUV6ePFDeviceID = 0x006e - // TPUV6pDeviceID is the PCI device ID of TPU V6p hardware. - TPUV6pDeviceID = 0x0075 + // TPU7xDeviceID is the PCI device ID of TPU 7x hardware. + TPU7xDeviceID = 0x0075 - // TPUV6pPFDeviceID is the PCI device ID of TPU V6p hardware PF. - TPUV6pPFDeviceID = 0x0076 + // TPU7xPFDeviceID is the PCI device ID of TPU 7x hardware PF. + TPU7xPFDeviceID = 0x0076 ) // TPUV4InterruptsMap maps BAR indices to valid register offsets. diff --git a/runsc/cmd/util/tpu.go b/runsc/cmd/util/tpu.go index 8b4804dfa6..8ee85f67ce 100644 --- a/runsc/cmd/util/tpu.go +++ b/runsc/cmd/util/tpu.go @@ -41,7 +41,7 @@ var ( tpuV4DeviceIDs = map[uint64]struct{}{tpu.TPUV4DeviceID: {}, tpu.TPUV4liteDeviceID: {}} tpuVFIODeviceIDs = map[uint64]struct{}{ tpu.TPUV4DeviceID: {}, tpu.TPUV4liteDeviceID: {}, tpu.TPUV5eDeviceID: {}, tpu.TPUV5pDeviceID: {}, - tpu.TPUV6eDeviceID: {}, tpu.TPUV6pDeviceID: {}, tpu.TPUV6ePFDeviceID: {}, tpu.TPUV6pPFDeviceID: {}, + tpu.TPUV6eDeviceID: {}, tpu.TPU7xDeviceID: {}, tpu.TPUV6ePFDeviceID: {}, tpu.TPU7xPFDeviceID: {}, } pciDeviceRegex = regexp.MustCompile(`0000:([[:xdigit:]]{2}|[[:xdigit:]]{4}):[[:xdigit:]]{2}\.[[:xdigit:]]{1,2}`) )