Skip to content

Commit 7fc1ab7

Browse files
milantracygvisor-bot
authored andcommitted
Add support for TPU7x Ghostfish.
PiperOrigin-RevId: 840451792
1 parent 9d75d56 commit 7fc1ab7

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

pkg/abi/tpu/tpu.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,11 +51,11 @@ const (
5151
// TPUV6ePFDeviceID is PCI device ID of TPU V6e hardware PF.
5252
TPUV6ePFDeviceID = 0x006e
5353

54-
// TPUV6pDeviceID is the PCI device ID of TPU V6p hardware.
55-
TPUV6pDeviceID = 0x0075
54+
// TPU7xDeviceID is the PCI device ID of TPU 7x hardware.
55+
TPU7xDeviceID = 0x0075
5656

57-
// TPUV6pPFDeviceID is the PCI device ID of TPU V6p hardware PF.
58-
TPUV6pPFDeviceID = 0x0076
57+
// TPU7xPFDeviceID is the PCI device ID of TPU 7x hardware PF.
58+
TPU7xPFDeviceID = 0x0076
5959
)
6060

6161
// TPUV4InterruptsMap maps BAR indices to valid register offsets.

runsc/cmd/util/tpu.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ var (
4141
tpuV4DeviceIDs = map[uint64]struct{}{tpu.TPUV4DeviceID: {}, tpu.TPUV4liteDeviceID: {}}
4242
tpuVFIODeviceIDs = map[uint64]struct{}{
4343
tpu.TPUV4DeviceID: {}, tpu.TPUV4liteDeviceID: {}, tpu.TPUV5eDeviceID: {}, tpu.TPUV5pDeviceID: {},
44-
tpu.TPUV6eDeviceID: {}, tpu.TPUV6pDeviceID: {}, tpu.TPUV6ePFDeviceID: {}, tpu.TPUV6pPFDeviceID: {},
44+
tpu.TPUV6eDeviceID: {}, tpu.TPU7xDeviceID: {}, tpu.TPUV6ePFDeviceID: {}, tpu.TPU7xPFDeviceID: {},
4545
}
4646
pciDeviceRegex = regexp.MustCompile(`0000:([[:xdigit:]]{2}|[[:xdigit:]]{4}):[[:xdigit:]]{2}\.[[:xdigit:]]{1,2}`)
4747
)

0 commit comments

Comments
 (0)