We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bd27253 commit 9a6b0f0Copy full SHA for 9a6b0f0
1 file changed
src/accelerators/TPU.jl
@@ -146,6 +146,7 @@ const _GOOGLE_PCI_VENDOR_ID = "0x1ae0"
146
v5p
147
v5e
148
v6e
149
+ tpu7x
150
end
151
152
const _TPU_PCI_DEVICE_IDS = Dict(
@@ -155,6 +156,7 @@ const _TPU_PCI_DEVICE_IDS = Dict(
155
156
"0x0062" => TPUVersion.v5p,
157
"0x0063" => TPUVersion.v5e,
158
"0x006f" => TPUVersion.v6e,
159
+ "0x0076" => TPUVersion.tpu7x,
160
)
161
162
has_tpu() = first(num_available_tpu_chips_and_device_id()) > 0
0 commit comments