Skip to content

Commit 9a6b0f0

Browse files
authored
Add TPU version tpu7x to TPU.jl (#2796)
1 parent bd27253 commit 9a6b0f0

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/accelerators/TPU.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,7 @@ const _GOOGLE_PCI_VENDOR_ID = "0x1ae0"
146146
v5p
147147
v5e
148148
v6e
149+
tpu7x
149150
end
150151

151152
const _TPU_PCI_DEVICE_IDS = Dict(
@@ -155,6 +156,7 @@ const _TPU_PCI_DEVICE_IDS = Dict(
155156
"0x0062" => TPUVersion.v5p,
156157
"0x0063" => TPUVersion.v5e,
157158
"0x006f" => TPUVersion.v6e,
159+
"0x0076" => TPUVersion.tpu7x,
158160
)
159161

160162
has_tpu() = first(num_available_tpu_chips_and_device_id()) > 0

0 commit comments

Comments
 (0)