TPU vs. GPU

  • Thread starter Thread starter Hornbein
  • Start date Start date
Join the discussion
Registration is free. Ask a follow-up in this thread, or start your own.
4 replies · 146 views
Hornbein
Gold Member
Messages
4,233
Reaction score
3,359
What's the difference? They both mostly do matrix multiplication. I don't see what is tensor about a TPU.
 
Physics news on Phys.org
One aspect will be the supported floating point formats. Once upon a time, GPUs only supported float (32-bit), then came support for double (64-bit), but with "much" less speed than float, at least compared to the same difference for a CPU.

Todays GPUs support FP32, FP16, FP8, INT8 efficiently, + FP64, and other common CPU integer types, albeit less efficiently. Already this part is slightly annoying and challenging, when scientific computation software developped on CPU gets ported to GPU, because those details suddenly get much more important for overall performance than they were on CPU.

TPUs don't even bother to support FP64, and are most efficient for strange floating point formats like bfloat16. They have a wide dynamic range, but limited precision in the sense of "classical" numerics theory. Actually, because the numerics theory I learned at university is unsuitable for them, I probably won't use them for scientific computing. I simply I don't have the time to build-up new intuitions for that type of limited precision provided by bfloat16.
 
Thanks, TPUs are cheaper and lower precision and have nothing to do with tensors.
 
gentzen said:
I simply I don't have the time to build-up new intuitions for that type of limited precision provided by bfloat16.

Yes, working in a world where 256 + 1 = 258 would be disorientating (or is 512 + 1 = 512 worse?)