Skip to content
KernelIndex
Search⌘K

torch / matmul73342c

torch_matmul_73342c · FlashInfer-Bench baselines · python · Apache-2.0

Use it

Vendorable · source mirrored · Apache-2.0View source →

No package. Vendor the mirrored source: 7 lines, Apache-2.0, pinned at da91508.

main.py
curl "https://kernelindex.com/api/v1/implementations/flashinfer-torch-matmul-73342c?include=source"
interfacepython
revisionda915083d4c7
symbolrun
pathmain.py
Compatibility
measured onNVIDIA B200
declared hardwareNVIDIA B200
architecturesunknown
dtypesfp16

Benchmark evidence

43 measurements across 1 GPU, fastest first.

Operation / workload
Hardware
Latency
Rank
Observed
GEMM n3072 k2048fp16 · [15, 2048]
NVIDIA B200
10.8µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [16, 2048]
NVIDIA B200
10.8µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [40, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [32, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [24, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [7, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [35, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [56, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [48, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [64, 2048]
NVIDIA B200
10.9µs
#1 of 1
2026-04-20
Show all 43 measurements ›
GEMM n3072 k2048fp16 · [72, 2048]
NVIDIA B200
11.0µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [70, 2048]
NVIDIA B200
11.0µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [8, 2048]
NVIDIA B200
11.0µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [80, 2048]
NVIDIA B200
11.1µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [96, 2048]
NVIDIA B200
11.1µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [88, 2048]
NVIDIA B200
11.1µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [4, 2048]
NVIDIA B200
11.2µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [144, 2048]
NVIDIA B200
11.2µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [136, 2048]
NVIDIA B200
11.3µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [184, 2048]
NVIDIA B200
11.3µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [176, 2048]
NVIDIA B200
11.4µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [112, 2048]
NVIDIA B200
11.4µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [128, 2048]
NVIDIA B200
11.4µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [2, 2048]
NVIDIA B200
11.4µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [120, 2048]
NVIDIA B200
11.5µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [192, 2048]
NVIDIA B200
11.5µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [224, 2048]
NVIDIA B200
11.8µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [168, 2048]
NVIDIA B200
11.8µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [232, 2048]
NVIDIA B200
11.9µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [160, 2048]
NVIDIA B200
12.0µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [240, 2048]
NVIDIA B200
12.1µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [152, 2048]
NVIDIA B200
12.2µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [216, 2048]
NVIDIA B200
12.2µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [104, 2048]
NVIDIA B200
12.2µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [248, 2048]
NVIDIA B200
12.2µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [256, 2048]
NVIDIA B200
12.4µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [200, 2048]
NVIDIA B200
12.6µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [208, 2048]
NVIDIA B200
12.7µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [1, 2048]
NVIDIA B200
12.7µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [972, 2048]
NVIDIA B200
16.8µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [2379, 2048]
NVIDIA B200
27.1µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [2053, 2048]
NVIDIA B200
27.1µs
#1 of 1
2026-04-20
GEMM n3072 k2048fp16 · [8192, 2048]
NVIDIA B200
70.2µs
#1 of 1
2026-04-20

Reported · How evidence levels are derived →

Source and license

sourcehttps://huggingface.co/datasets/flashinfer-ai/flashinfer-trace
commitda915083d4c7c5e61aa3005e3d17ae488e0fc71c
revision digestsha256:e9a1ef45aaffced2ec1f7b24d92bc1350c7bcf2f58f978cba6d1288465564d4b
license declaredApache-2.0
license concludedApache-2.0
authorsbaseline
imported2026-08-16

Kernel source

main.py7 lines
import torch
import torch.nn.functional as F

def run(A: torch.Tensor, B: torch.Tensor):
    C = F.linear(A, B)
    return C

Source code from FlashInfer-Bench (flashinfer-ai/flashinfer-trace) · Apache-2.0

Best evidence level for this revision: reported

JSON