TPU v5p supported configurations¶
The slice shapes you can request for TPU v5p, and how many chips and hosts each uses.
Slice shapes¶
Note: A slice shape gives the number of chips along each dimension of the interconnect. A 2D shape is written AxB and a 3D shape AxBxC. Multiply the dimensions to get the number of chips in the slice.
Topology |
Chips |
Hosts |
Cubes |
Twisted supported |
|---|---|---|---|---|
2x2x1 |
4 |
1 |
N/A |
N/A |
2x2x2 |
8 |
2 |
N/A |
N/A |
2x4x4 |
32 |
8 |
N/A |
N/A |
4x4x4 |
64 |
16 |
1 |
N/A |
4x4x8 |
128 |
32 |
2 |
Yes |
4x8x8 |
256 |
64 |
4 |
Yes |
8x8x8 |
512 |
128 |
8 |
N/A |
8x8x16 |
1024 |
256 |
16 |
Yes |
8x16x16 |
2048 |
512 |
32 |
Yes |
16x16x16 |
4096 |
1024 |
64 |
N/A |
16x16x24 |
6144 |
1536 |
96 |
N/A |
Every v5p VM uses the ct5p-hightpu-4t machine type with 4 chips.
Parent topic:TPU v5p