Skip to content
KernelIndex
Search⌘K

MLP

4 eligible runs
model

KernelBench level3 problem 1: MLP. The computation is the reference PyTorch module's forward pass; the output shape follows the module (mirrored as each implementation's source).

Source baseline · unbeaten
3.79ms±0.00 · mean of 100
PyTorch eagerPyTorch · MIT · python

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

Current records

Not measured on B200 for this workload. Challenges →

Source-native comparison · GPU NVIDIA H100 · Workload batch_size = 128 · input_size = 16384 · fp32 · Protocol KernelBench timing scripts · mean · 2 results · last observed 2026-03-05Record history →
Estimated floor 2.50 µs · record 1513.54× above itestimate, not evidence ›
DRAM 2.50 µs · bandwidth-bound on H100 SXM
every declared tensor crosses HBM exactly once (3,350 GB/s, H100 SXM datasheet)
no arithmetic formula for this family: bandwidth floor only
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
PyTorch eagerbaselinePyTorch
3.79ms±0.00
1.00×
Reported · MIT · source
2026-03-05stale

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

source mirroredMITno install recipeView source →Run detail →
1=
3.79ms±0.00
1.00×
Reported · MIT · source
2026-03-05stale

Measured exactly what you asked; tied ranks share a number. Reported by source; not independently reproduced.

source mirroredMITno install recipeView source →Run detail →

Implementations

Implementation
Runtime
Best latency
Evidence
Availability
python · torch_eager
3.79ms
1.00×
Reported
MIT · source
python · torch_compile_inductor
3.79ms
1.00×
Reported
MIT · source

Semantics

Inputs and outputs
input_1fp32 [batch_size, input_size]
outfloat [out]
Axes and behavior
outvariable
batch_sizevariable
input_sizevariable
determinismunspecified
constraintsNo mutation or aliasing
Identity
sha256557652a0c715…
Sources: KernelBench baseline timings (2026-03-05) · MITlast observed 2026-03-05How records are decidedJSON