Skip to content
KernelIndex
Search⌘K
Sign in

GEMM n16384 k2048

43 eligible runs
gemm

General matrix multiply (GEMM) C = A @ B.T. Captured from Llama-3.2-1B mlp.gate_up_proj (fused gate+up: 2 * 8192 = 16384).

Source baseline · unbeaten
20.9µsmean
torch / matmul0a4e73FlashInfer-Bench baselines · Apache-2.0 · python

Reported evidence · last observed 2026-04-20. The source's designated baseline implementation. Reported by source; not independently reproduced.

No package · vendor the source →View source →Run detail →

Current records

Workloadm = 80 · fp1643 cases
m
dtype
1fp162fp164fp167fp168fp1615fp1616fp1624fp1632fp1635fp1640fp1648fp1656fp1664fp1670fp1672fp1680fp1688fp1696fp16104fp16112fp16120fp16128fp16136fp16144fp16152fp16160fp16168fp16176fp16184fp16192fp16200fp16208fp16216fp16224fp16232fp16240fp16248fp16256fp16972fp162053fp162379fp168192fp16
scrolls · 43 cases total

Not measured on H100 for this workload. Challenges →

Source-native comparison · GPU NVIDIA B200 · Workload m = 80 · fp16 · CUDA 13.0 · Framework pytorch 2.11.0+cu130 · Protocol flashinfer-bench · mean · 1 results · last observed 2026-04-20Record history →
Estimated floor 8.43 µs · record 2.48× above itestimate, not evidence ›
DRAM 8.43 µs · compute 2.39 µs · bandwidth-bound on B200
every declared tensor crosses HBM exactly once (8,000 GB/s, B200 datasheet)
2·M·N·K with M=80, N=16384, K=2048 at the dense bf16 peak (2,250 TFLOP/s)
headroom-v1: a lower bound from declared tensors and datasheet peaks. A kernel can sit well above it for good reasons.
#
Implementation
Latency
vs #1
Trust
Observed
1
torch / matmul0a4e73baselineFlashInfer-Bench baselines
20.9µs
1.00×
Reported · Apache-2.0 · source
2026-04-20

Measured exactly what you asked. The source's designated baseline implementation. Reported by source; not independently reproduced.

source mirroredApache-2.0no install recipeView source →Run detail →

Scaling by m

100.0 µs200.0 µs300.0 µs400.0 µs14248825620538km →21.0 µs21.1 µs21.0 µs21.0 µs21.0 µs20.9 µs21.0 µs21.0 µs21.0 µs21.0 µs20.9 µs21.0 µs21.0 µs21.0 µs20.9 µs21.0 µs20.9 µs21.7 µs22.6 µs21.7 µs22.2 µs22.5 µs22.5 µs22.5 µs22.5 µs23.3 µs23.1 µs23.3 µs23.0 µs23.7 µs23.2 µs23.5 µs23.1 µs24.4 µs24.0 µs25.5 µs24.9 µs24.9 µs25.0 µs53.8 µs93.4 µs104.7 µs384.7 µstorch / matmul 0a4e73
torch / matmul 0a4e73best per workload · NVIDIA B200 · CUDA 13.0 · pytorch 2.11.0+cu130 · flashinfer-bench held constant

Implementations

Implementation
Runtime
Best latency
Evidence
Availability
torch / matmul0a4e73FlashInfer-Bench baselines
python
20.9µs
1.00×
Reported
Apache-2.0 · source

Semantics

Inputs and outputs
afp16 [m, k]
bfp16 [n, k]
cfp16 [m, n]
Axes and behavior
kconstant = 2048
mvariable
nconstant = 16384
determinismunspecified
constraintsNo mutation or aliasing
Identity
aliasgemm_n16384_k2048modelllama-3-2-1bsha256f0fc13590a06…
Sources: FlashInfer-Bench (2026-04-20) · Apache-2.0last observed 2026-04-20How records are decidedJSON