Google Cloud TPU v2
Fields
accelerators/google-tpu-v2Notes
Per-chip figures. TPU v2 is the first generation with HBM, bfloat16 and the ICI that makes a supercomputer out of chips; the v2 supercomputer is a 16x16 2D torus of 256 chips. The corpus references v2 repeatedly (supply pricing notes call it a legacy pod alongside v3; the v3 docs page has a 'Performance benefits of TPU v3 over v2' section) but held no record for it, and this is that record. WHAT GOOGLE DOES PUBLISH, per generation, for a multi-host deployment: HBM capacity 16 GiB per chip at 700 GB/s; ICI 4 links per chip at 496 Gbit/s PER DIRECTION, i.e. 992 Gbit/s bidirectional per chip = 124 GB/s bidirectional; pod scale 256 chips per supercomputer. Pod-scale published numbers from the same table: 11.8 PFLOPs peak and 15.9 Tbit/s bisection bandwidth (15.9 Tbit/s is 32 links x 496 Gbit/s, which is exactly the bisection of a 16x16 torus). Clock 700 MHz, TDP 280 W per chip, 124 kW for a full 256-chip supercomputer, die <611 mm2 on a process the paper states only as '>12nm' (so process_nm is null, not a guess). VERIFIED NEGATIVE - PROCESS NODE. process_nm is null and that is the correct value, not a gap. The paper states the foundry partner is under NDA and gives TPUv2's technology only as '>12nm', an upper bound. Recording any specific node would be inventing a number from a bound. VERIFIED NEGATIVE - MEMORY BUS WIDTH. memory_bus_bit is null. The paper describes thirty-two 128-bit buses to four HBM stacks, which is a wiring description of a multi-stack HBM interface and not a single published aggregate bus width; no aggregate figure is stated, so none was back-computed from the 700 GB/s. TWO TRAPS SPECIFIC TO THIS RECORD. (1) ON-CHIP vs OFF-CHIP MEMORY: Table 3 prints '32MiB/16GiB' for memory size. The 32 MiB is ON-CHIP memory inside the VPU and the 16 GiB is OFF-CHIP HBM. The paper separately describes the vector memory (Vmem) itself as 32K 128x32-bit elements, which is 16 MiB - i.e. HALF the 32 MiB on-chip figure in the table, and the paper does not state what accounts for the other half (it also lists 32 2D vector registers of 128x8 32-bit elements, i.e. 4 KiB each). That decomposition is not reconciled by the source and is NOT reconstructed here; only the table's 32MiB/16GiB split is recorded, as printed. vram_gb here is the OFF-CHIP HBM only, because that is the capacity that bounds a model; the 32 MiB on-chip figure is recorded in the notes rather than in onchip_cache, which is reserved for a cache/LLC/SRAM-tier hierarchy and whose 'kind' enum no member honestly describes a TPU vector scratchpad. Do not read vram_gb as total chip-attached memory. (2) TDP IS NOT A MEASURED POWER FIGURE on this part: 280 W is the paper's TDP row for TPUv2. By contrast v3 and v4 carry Google-MEASURED min/mean/max power rows. Comparing 280 W (v2, TDP) against 262 W (v3, measured max) as though they were the same kind of number is wrong even though they are close. WHY THE 32b AND 16b ROWS ARE BOTH PRESENT and the reason this part is a poor fit for the compute-heavy work in this corpus: a BF16 intensity of 46e12/700e9 = 66 FLOP/byte is roughly half of v5e's and a quarter of v6e's, and the FP32 path is 3 TFLOPS. TPU v2 is dominated by infeed and rematerialization - Google's v3 docs page lists the actual v2->v3 benefits as FLOPS per TensorCore for compute-bound models, less recomputation where data did not fit v2 memory, and new model/batch shapes that did not fit at all.