Dev finds 1x1 convs with constant weights best squeeze TFLOP/s from Pixel 9A's TPU

mgostIH · x · 2026-10-09

While debugging the Pixel 9A's G4 TPU compiler, mgostIH found matmul performance is highly sensitive to compiler preferences: 1x1 convolutions with constant parameters are the best way to squeeze out TFLOP/s, but you must find the compiler's happy path or compilation times out. First-hand on-device TPU tuning experience.

Related event: Pixel 9A TPU compiler slammed by developer, performance data hidden(3 posts)→

Original post →

More from Infra

Infra channel →