flashinfer / wrappere1f464
flashinfer_wrapper_e1f464 · FlashInfer-Bench baselines · python · Apache-2.0
Use it
Vendorable · source mirrored · Apache-2.0View source →
No package. Vendor the mirrored source: 15 lines, Apache-2.0, pinned at da91508.
main.py
curl "https://kernelindex.com/api/v1/implementations/flashinfer-flashinfer-wrapper-e1f464?include=source"interfacepython
revisionda915083d4c7
symbolrun
pathmain.py
Compatibility
declared hardwareNVIDIA A100, NVIDIA B200, NVIDIA H100, NVIDIA H20, NVIDIA H200
architecturesunknown
dtypesbf16
Benchmark evidence
No published measurement for this revision.
No evidence · How evidence levels are derived →
Source and license
sourcehttps://huggingface.co/datasets/flashinfer-ai/flashinfer-trace
commitda915083d4c7c5e61aa3005e3d17ae488e0fc71c
revision digestsha256:25cf14c1a6e31c86498a651c3074586b9d311493b90aee620814d0d3108b078c
license declaredApache-2.0
license concludedApache-2.0
authorsbaseline
imported2026-08-16
Kernel source
main.py15 lines
import torch
import flashinfer
def run(hidden_states, weight):
batch_size, hidden_size = hidden_states.shape
assert hidden_size == 5120
EPS = 1e-6
output = flashinfer.norm.rmsnorm(hidden_states, weight, eps=EPS)
return output
Source code from the importing source · Apache-2.0
No published measurement for this revision
JSON