TPUv2 Benchmarks (Yup, they’re fast)

RiseML benchmarked Google’s TPUv2, and — surprise! — they’re fast. Also — surprise! — they’re cost efficient too.
I know, shocking, right? After all, if you consider selection-bias, of course they wouldn’t release something that wasn’t fast, and of course they’d price it so that it would be cost efficient. Then again, this wouldn’t be the first time a company has done something stupid, so, hey, we should probably keep that in mind .
Anyhow, it turns out that the TPUv2 is around 7.3x faster than an Nvidia P100, and around 2.8x faster than an Nvidia V100. Yes, it’s more expensive, but you actually end up saving on a $ per hour basis — it’s that much faster.
It’s important to note that the TPUv2 is not just a speeded up version of the TPUv1 — it’s quite a different beast. Unlike the TPUv1, which was focused on inference, the TPUv2 is actually focused on speeding up learning. Inside, it has systolic arrays (•) which do 180TFLOPS of matrix math, with 64GB of memory available — against the Nvidia V100’s 125 TFLOPS / 16GB.
Mind you, the whole thing is “cloud-based”. You don’t get a TPUv2 enabled instance, instead, you are assigned a TPUv2 that exists somewhere in the “cloud”.
(•) Systolic Array: Parallel input data flows through a network of hard-wired processor nodes, which combine, process, merge or sort the input data into a derived result. Because the wave-like propagation of data through a systolic array resembles the pulse of the human circulatory system, the name systolic was coined from medical terminology. The name is derived from systole as an analogy to the regular pumping of blood by the heart.

Comments

Popular posts from this blog

Erlang, Binaries, and Garbage Collection (Sigh)

Spot The Difference — Deep Learning Edition