Skip to content

Instantly share code, notes, and snippets.

@rizar
Last active January 9, 2022 18:54
Show Gist options
  • Save rizar/878155ab5d9a65beafaa96231cd8cec9 to your computer and use it in GitHub Desktop.
Save rizar/878155ab5d9a65beafaa96231cd8cec9 to your computer and use it in GitHub Desktop.
Throughput table
model precision throughput in teraWFLOPs
matmul 8192x1600x6400 bfloat16 237
linear MLP B=8192; 1600 -> 6400 -> 1600 -> 1 bfloat16 230
linear MLP B=8192; 1600 -> 6400 -> 1600 -> 1 mixed 207.6
residual ReLU MLP B=8192; 1600 -> 6400 -> 1600 -> 1 mixed 185.7
HuggingFace GPT2 B=32 L=256; d=1600 mixed 68
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment