Skip to content
KernelIndex
Search⌘K

submission 561332

idanbeck · python · License unknown

Use it

Vendorable · source mirrored · license unknownView source →

No package. Vendor the mirrored source: 27 lines, June 9 Researcher Reciprocity License v1.0.

submission.py
curl "https://kernelindex.com/api/v1/implementations/kernelbot-vectoradd-v2-561332?include=source"
interfacepython
Compatibility
measured onNVIDIA A100
declared hardwareNVIDIA A100
architecturessm_80
dtypesfp16

Benchmark evidence

1 measurement across 1 GPU, fastest first.

Operation / workload
Hardware
Latency
Rank
Observed
FP16 vector additionsuite of 5 cases
NVIDIA A100
916.0µs
#19 of 87
2026-03-15

Reported · How evidence levels are derived →

Source and license

sourceavailable
revision digestsha256:5bc32c61b82370b70523d28e9a4d1c15f026262c5babb8a180d5a3c8efe7300e
license declaredunknown
license concludedunknown
authorsidanbeck
imported2026-08-15

Techniques

Extracted from the mirrored source by pattern, never inferred. Each row cites its line.

autotunereturn torch.compile(f, fullgraph=False, mode="max-autotune-no-cudagraphs")

Kernel source

submission.py27 lines
#!POPCORN leaderboard vectoradd_v2
#!POPCORN gpu A100
import torch
_F = None

def _build():
    def f(a, b, out):
        out[...] = a + b
        return out
    try:
        return torch.compile(f, fullgraph=False, mode="max-autotune-no-cudagraphs")
    except Exception:
        return f

@torch.inference_mode()
def custom_kernel(data):
    global _F
    if _F is None:
        _F = _build()
    if isinstance(data, (tuple, list)) and len(data) == 2:
        a, b = data
        return a + b
    if isinstance(data, (tuple, list)) and len(data) == 3:
        a, b, out = data
        return _F(a, b, out)
    raise ValueError("unexpected data tuple")

Source code from GPU Mode and the KernelBot dataset · June 9 Researcher Reciprocity License v1.0

Best evidence level for this revision: reported

JSON